2016-02-09 23:12:00 +00:00
|
|
|
// Copyright (c) 2011-present, Facebook, Inc. All rights reserved.
|
2017-07-15 23:03:42 +00:00
|
|
|
// This source code is licensed under both the GPLv2 (found in the
|
|
|
|
// COPYING file in the root directory) and Apache 2.0 License
|
|
|
|
// (found in the LICENSE.Apache file in the root directory).
|
2015-11-06 16:07:08 +00:00
|
|
|
//
|
|
|
|
|
|
|
|
#include "util/random.h"
|
|
|
|
|
|
|
|
#include <stdint.h>
|
|
|
|
#include <string.h>
|
2015-11-10 20:50:09 +00:00
|
|
|
#include <thread>
|
|
|
|
#include <utility>
|
2015-11-06 16:07:08 +00:00
|
|
|
|
|
|
|
#include "port/likely.h"
|
|
|
|
#include "util/thread_local.h"
|
|
|
|
|
2022-05-18 22:25:19 +00:00
|
|
|
#define STORAGE_DECL static thread_local
|
2015-11-06 16:07:08 +00:00
|
|
|
|
2020-02-20 20:07:53 +00:00
|
|
|
namespace ROCKSDB_NAMESPACE {
|
2015-11-06 16:07:08 +00:00
|
|
|
|
|
|
|
Random* Random::GetTLSInstance() {
|
|
|
|
STORAGE_DECL Random* tls_instance;
|
|
|
|
STORAGE_DECL std::aligned_storage<sizeof(Random)>::type tls_instance_bytes;
|
|
|
|
|
|
|
|
auto rv = tls_instance;
|
|
|
|
if (UNLIKELY(rv == nullptr)) {
|
2015-11-10 20:50:09 +00:00
|
|
|
size_t seed = std::hash<std::thread::id>()(std::this_thread::get_id());
|
|
|
|
rv = new (&tls_instance_bytes) Random((uint32_t)seed);
|
2015-11-06 16:07:08 +00:00
|
|
|
tls_instance = rv;
|
|
|
|
}
|
|
|
|
return rv;
|
|
|
|
}
|
|
|
|
|
2020-07-09 21:33:42 +00:00
|
|
|
std::string Random::HumanReadableString(int len) {
|
|
|
|
std::string ret;
|
|
|
|
ret.resize(len);
|
|
|
|
for (int i = 0; i < len; ++i) {
|
|
|
|
ret[i] = static_cast<char>('a' + Uniform(26));
|
|
|
|
}
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
std::string Random::RandomString(int len) {
|
|
|
|
std::string ret;
|
|
|
|
ret.resize(len);
|
|
|
|
for (int i = 0; i < len; i++) {
|
|
|
|
ret[i] = static_cast<char>(' ' + Uniform(95)); // ' ' .. '~'
|
|
|
|
}
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
2021-05-12 19:34:22 +00:00
|
|
|
std::string Random::RandomBinaryString(int len) {
|
|
|
|
std::string ret;
|
|
|
|
ret.resize(len);
|
|
|
|
for (int i = 0; i < len; i++) {
|
|
|
|
ret[i] = static_cast<char>(Uniform(CHAR_MAX));
|
|
|
|
}
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
2020-02-20 20:07:53 +00:00
|
|
|
} // namespace ROCKSDB_NAMESPACE
|