|
|
|
// Copyright (c) 2011-present, Facebook, Inc. All rights reserved.
|
|
|
|
// 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).
|
|
|
|
//
|
|
|
|
|
|
|
|
#pragma once
|
|
|
|
|
|
|
|
#include <sstream>
|
|
|
|
#include <string>
|
|
|
|
#include <unordered_map>
|
|
|
|
#include <vector>
|
|
|
|
|
|
|
|
#include "rocksdb/rocksdb_namespace.h"
|
|
|
|
|
|
|
|
namespace ROCKSDB_NAMESPACE {
|
|
|
|
|
|
|
|
class Slice;
|
|
|
|
|
|
|
|
extern std::vector<std::string> StringSplit(const std::string& arg, char delim);
|
|
|
|
|
|
|
|
template <typename T>
|
|
|
|
inline std::string ToString(T value) {
|
|
|
|
#if !(defined OS_ANDROID) && !(defined CYGWIN) && !(defined OS_FREEBSD)
|
|
|
|
return std::to_string(value);
|
|
|
|
#else
|
|
|
|
// Andorid or cygwin doesn't support all of C++11, std::to_string() being
|
|
|
|
// one of the not supported features.
|
|
|
|
std::ostringstream os;
|
|
|
|
os << value;
|
|
|
|
return os.str();
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
// Append a human-readable printout of "num" to *str
|
|
|
|
extern void AppendNumberTo(std::string* str, uint64_t num);
|
|
|
|
|
|
|
|
// Append a human-readable printout of "value" to *str.
|
|
|
|
// Escapes any non-printable characters found in "value".
|
|
|
|
extern void AppendEscapedStringTo(std::string* str, const Slice& value);
|
|
|
|
|
|
|
|
// Return a string printout of "num"
|
|
|
|
extern std::string NumberToString(uint64_t num);
|
|
|
|
|
Built-in support for generating unique IDs, bug fix (#8708)
Summary:
Env::GenerateUniqueId() works fine on Windows and on POSIX
where /proc/sys/kernel/random/uuid exists. Our other implementation is
flawed and easily produces collision in a new multi-threaded test.
As we rely more heavily on DB session ID uniqueness, this becomes a
serious issue.
This change combines several individually suitable entropy sources
for reliable generation of random unique IDs, with goal of uniqueness
and portability, not cryptographic strength nor maximum speed.
Specifically:
* Moves code for getting UUIDs from the OS to port::GenerateRfcUuid
rather than in Env implementation details. Callers are now told whether
the operation fails or succeeds.
* Adds an internal API GenerateRawUniqueId for generating high-quality
128-bit unique identifiers, by combining entropy from three "tracks":
* Lots of info from default Env like time, process id, and hostname.
* std::random_device
* port::GenerateRfcUuid (when working)
* Built-in implementations of Env::GenerateUniqueId() will now always
produce an RFC 4122 UUID string, either from platform-specific API or
by converting the output of GenerateRawUniqueId.
DB session IDs now use GenerateRawUniqueId while DB IDs (not as
critical) try to use port::GenerateRfcUuid but fall back on
GenerateRawUniqueId with conversion to an RFC 4122 UUID.
GenerateRawUniqueId is declared and defined under env/ rather than util/
or even port/ because of the Env dependency.
Likely follow-up: enhance GenerateRawUniqueId to be faster after the
first call and to guarantee uniqueness within the lifetime of a single
process (imparting the same property onto DB session IDs).
Pull Request resolved: https://github.com/facebook/rocksdb/pull/8708
Test Plan:
A new mini-stress test in env_test checks the various public
and internal APIs for uniqueness, including each track of
GenerateRawUniqueId individually. We can't hope to verify anywhere close
to 128 bits of entropy, but it can at least detect flaws as bad as the
old code. Serial execution of the new tests takes about 350 ms on
my machine.
Reviewed By: zhichao-cao, mrambacher
Differential Revision: D30563780
Pulled By: pdillinger
fbshipit-source-id: de4c9ff4b2f581cf784fcedb5f39f16e5185c364
3 years ago
|
|
|
// Put n digits from v in base kBase to (*buf)[0] to (*buf)[n-1] and
|
|
|
|
// advance *buf to the position after what was written.
|
|
|
|
template <size_t kBase>
|
|
|
|
inline void PutBaseChars(char** buf, size_t n, uint64_t v, bool uppercase) {
|
|
|
|
const char* digitChars = uppercase ? "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
|
|
|
: "0123456789abcdefghijklmnopqrstuvwxyz";
|
|
|
|
for (size_t i = n; i > 0; --i) {
|
|
|
|
(*buf)[i - 1] = digitChars[v % kBase];
|
|
|
|
v /= kBase;
|
|
|
|
}
|
|
|
|
*buf += n;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Return a human-readable version of num.
|
|
|
|
// for num >= 10.000, prints "xxK"
|
|
|
|
// for num >= 10.000.000, prints "xxM"
|
|
|
|
// for num >= 10.000.000.000, prints "xxG"
|
|
|
|
extern std::string NumberToHumanString(int64_t num);
|
|
|
|
|
|
|
|
// Return a human-readable version of bytes
|
|
|
|
// ex: 1048576 -> 1.00 GB
|
|
|
|
extern std::string BytesToHumanString(uint64_t bytes);
|
|
|
|
|
|
|
|
// Return a human-readable version of unix time
|
|
|
|
// ex: 1562116015 -> "Tue Jul 2 18:06:55 2019"
|
|
|
|
extern std::string TimeToHumanString(int unixtime);
|
|
|
|
|
|
|
|
// Append a human-readable time in micros.
|
|
|
|
int AppendHumanMicros(uint64_t micros, char* output, int len,
|
|
|
|
bool fixed_format);
|
|
|
|
|
|
|
|
// Append a human-readable size in bytes
|
|
|
|
int AppendHumanBytes(uint64_t bytes, char* output, int len);
|
|
|
|
|
|
|
|
// Return a human-readable version of "value".
|
|
|
|
// Escapes any non-printable characters found in "value".
|
|
|
|
extern std::string EscapeString(const Slice& value);
|
|
|
|
|
|
|
|
// Parse a human-readable number from "*in" into *value. On success,
|
|
|
|
// advances "*in" past the consumed number and sets "*val" to the
|
|
|
|
// numeric value. Otherwise, returns false and leaves *in in an
|
|
|
|
// unspecified state.
|
|
|
|
extern bool ConsumeDecimalNumber(Slice* in, uint64_t* val);
|
|
|
|
|
|
|
|
// Returns true if the input char "c" is considered as a special character
|
|
|
|
// that will be escaped when EscapeOptionString() is called.
|
|
|
|
//
|
|
|
|
// @param c the input char
|
|
|
|
// @return true if the input char "c" is considered as a special character.
|
|
|
|
// @see EscapeOptionString
|
|
|
|
bool isSpecialChar(const char c);
|
|
|
|
|
|
|
|
// If the input char is an escaped char, it will return the its
|
|
|
|
// associated raw-char. Otherwise, the function will simply return
|
|
|
|
// the original input char.
|
|
|
|
char UnescapeChar(const char c);
|
|
|
|
|
|
|
|
// If the input char is a control char, it will return the its
|
|
|
|
// associated escaped char. Otherwise, the function will simply return
|
|
|
|
// the original input char.
|
|
|
|
char EscapeChar(const char c);
|
|
|
|
|
|
|
|
// Converts a raw string to an escaped string. Escaped-characters are
|
|
|
|
// defined via the isSpecialChar() function. When a char in the input
|
|
|
|
// string "raw_string" is classified as a special characters, then it
|
|
|
|
// will be prefixed by '\' in the output.
|
|
|
|
//
|
|
|
|
// It's inverse function is UnescapeOptionString().
|
|
|
|
// @param raw_string the input string
|
|
|
|
// @return the '\' escaped string of the input "raw_string"
|
|
|
|
// @see isSpecialChar, UnescapeOptionString
|
|
|
|
std::string EscapeOptionString(const std::string& raw_string);
|
|
|
|
|
|
|
|
// The inverse function of EscapeOptionString. It converts
|
|
|
|
// an '\' escaped string back to a raw string.
|
|
|
|
//
|
|
|
|
// @param escaped_string the input '\' escaped string
|
|
|
|
// @return the raw string of the input "escaped_string"
|
|
|
|
std::string UnescapeOptionString(const std::string& escaped_string);
|
|
|
|
|
|
|
|
std::string trim(const std::string& str);
|
|
|
|
|
|
|
|
// Returns true if "string" ends with "pattern"
|
|
|
|
bool EndsWith(const std::string& string, const std::string& pattern);
|
|
|
|
|
|
|
|
// Returns true if "string" starts with "pattern"
|
|
|
|
bool StartsWith(const std::string& string, const std::string& pattern);
|
|
|
|
|
|
|
|
#ifndef ROCKSDB_LITE
|
|
|
|
bool ParseBoolean(const std::string& type, const std::string& value);
|
|
|
|
|
|
|
|
uint8_t ParseUint8(const std::string& value);
|
|
|
|
|
|
|
|
uint32_t ParseUint32(const std::string& value);
|
|
|
|
|
|
|
|
int32_t ParseInt32(const std::string& value);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
uint64_t ParseUint64(const std::string& value);
|
|
|
|
|
|
|
|
int ParseInt(const std::string& value);
|
|
|
|
|
|
|
|
int64_t ParseInt64(const std::string& value);
|
|
|
|
|
|
|
|
double ParseDouble(const std::string& value);
|
|
|
|
|
|
|
|
size_t ParseSizeT(const std::string& value);
|
|
|
|
|
|
|
|
std::vector<int> ParseVectorInt(const std::string& value);
|
|
|
|
|
|
|
|
bool SerializeIntVector(const std::vector<int>& vec, std::string* value);
|
|
|
|
|
|
|
|
extern const std::string kNullptrString;
|
|
|
|
|
|
|
|
// errnoStr() function returns a string that describes the error code passed in
|
|
|
|
// the argument err
|
|
|
|
extern std::string errnoStr(int err);
|
|
|
|
|
|
|
|
} // namespace ROCKSDB_NAMESPACE
|