git.m6w6.name Git - awesomized/libmemcached/blob - testing/lib/random.hpp

   1 #pragma once
   2
   3 #include <cstddef>
   4 #include <chrono>
   5 #include <random>
   6 #include <string>
   7 #include <type_traits>
   8 #include <utility>
   9
  10 using namespace std;
  11
  12 using kv_pair = pair<string, string>;
  13
  14 template<typename T>
  15 enable_if_t<is_integral_v<T>, T> random_num(T min, T max) {
  16   using namespace chrono;
  17   using rnd = mt19937;
  18   using dst = uniform_int_distribution<T>;
  19
  20   auto time = duration_cast<nanoseconds>(system_clock::now().time_since_epoch());
  21   auto seed = static_cast<rnd::result_type>(time.count() % numeric_limits<T>::max());
  22   auto rgen = rnd{seed};
  23   return dst(min, max)(rgen);
  24 }
  25
  26 unsigned random_port();
  27 string random_port_string(const string &);
  28
  29 char random_ascii(char min = '!', char max = '~');
  30 string random_ascii_string(size_t len, char min = '!', char max = '~');
  31 kv_pair random_ascii_pair(size_t minlen = 1<<2, size_t maxlen = 1<<10);
  32
  33 template<template <typename> class Container>
  34 auto random_ascii_pairs(size_t count, size_t minlen = 1<<2, size_t maxlen = 1<<10) {
  35   Container<kv_pair> v;
  36
  37   v.reserve(count);
  38   for (size_t i = 0; i < count; ++i) {
  39     v.emplace_back(random_ascii_pair(minlen, maxlen));
  40   }
  41
  42   return v;
  43 }
  44
  45 string random_socket(const string &prefix = "/tmp/libmc.");
  46 string random_socket_or_port_string(const string &what);
  47 string random_socket_or_port_flag(const string &binary);
  48
  49 inline auto random_socket_or_port_arg() {
  50   return make_pair(&random_socket_or_port_flag, &random_socket_or_port_string);
  51 }