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

   1 /*
   2     +--------------------------------------------------------------------+
   3     | libmemcached - C/C++ Client Library for memcached                  |
   4     +--------------------------------------------------------------------+
   5     | Redistribution and use in source and binary forms, with or without |
   6     | modification, are permitted under the terms of the BSD license.    |
   7     | You should have received a copy of the license in a bundled file   |
   8     | named LICENSE; in case you did not receive a copy you can review   |
   9     | the terms online at: https://opensource.org/licenses/BSD-3-Clause  |
  10     +--------------------------------------------------------------------+
  11     | Copyright (c) 2006-2014 Brian Aker   https://datadifferential.com/ |
  12     | Copyright (c) 2020 Michael Wallner   <mike@php.net>                |
  13     +--------------------------------------------------------------------+
  14 */
  15
  16 #pragma once
  17
  18 #include <cstddef>
  19 #include <chrono>
  20 #include <iostream>
  21 #include <random>
  22 #include <string>
  23 #include <type_traits>
  24 #include <utility>
  25 #include <mutex>
  26
  27 using namespace std;
  28
  29 using kv_pair = pair<string, string>;
  30
  31 extern mt19937_64 rnd_eng;
  32 extern mutex rnd_mtx;
  33
  34 void random_setup();
  35
  36 template<typename T>
  37 enable_if_t<is_integral_v<T>, T> random_num(T min, T max) {
  38   lock_guard m{rnd_mtx};
  39   return uniform_int_distribution<T>(min, max)(rnd_eng);
  40 }
  41
  42 unsigned random_port();
  43 string random_port_string(const string &);
  44
  45 char random_binary();
  46 string random_binary_string(size_t len);
  47 char random_ascii(char min = '!', char max = '~');
  48 string random_ascii_string(size_t len, char min = '!', char max = '~');
  49 kv_pair random_ascii_pair(size_t minlen = 1 << 2, size_t maxlen = 1 << 10);
  50
  51 template<template<typename> class Container>
  52 auto random_ascii_pairs(size_t count, size_t minlen = 1 << 2, size_t maxlen = 1 << 10) {
  53   Container<kv_pair> v;
  54
  55   v.reserve(count);
  56   for (size_t i = 0; i < count; ++i) {
  57     v.emplace_back(random_ascii_pair(minlen, maxlen));
  58   }
  59
  60   return v;
  61 }
  62
  63 string random_socket(const string &prefix = "/tmp/libmc.");
  64 string random_socket_or_port_string(const string &what);
  65 string random_socket_or_port_flag(const string &binary);
  66
  67 inline auto random_socket_or_port_arg() {
  68   return make_pair(&random_socket_or_port_flag, &random_socket_or_port_string);
  69 }