c14c70b89d
this is a part of a series to migrating from `operator<<(ostream&, ..)` based formatting to fmtlib based formatting. the goal here is to enable fmtlib to print UUID without using ostream<<. also, this change reimplements some formatting helpers using fmtlib for better performance and less dependencies on operator<<(), but we cannot drop it at this moment, as quite a few caller sites are still using operator<<(ostream&, const UUID&) and operator<<(ostream&, tagged_uuid<T>&). we will address them separately. * add fmt::formatter<UUID> * add fmt::formatter<tagged_uuid<T>> * implement UUID::to_string() using fmt::to_string() * implement operator<<(std::ostream&, const UUID&) with fmt::print(), this should help to improve the performance when printing uuid, as fmt::print() does not materialize a string when printing the uuid. Refs #13245 Signed-off-by: Kefu Chai <kefu.chai@scylladb.com>
67 lines
1.8 KiB
C++
67 lines
1.8 KiB
C++
/*
|
|
* Copyright (C) 2015-present ScyllaDB
|
|
*/
|
|
|
|
/*
|
|
* SPDX-License-Identifier: AGPL-3.0-or-later
|
|
*/
|
|
|
|
|
|
#include "UUID.hh"
|
|
#include <seastar/net/byteorder.hh>
|
|
#include <random>
|
|
#include <boost/iterator/function_input_iterator.hpp>
|
|
#include <boost/algorithm/string.hpp>
|
|
#include <string>
|
|
#include <seastar/core/sstring.hh>
|
|
#include "utils/serialization.hh"
|
|
#include "marshal_exception.hh"
|
|
|
|
namespace utils {
|
|
|
|
UUID
|
|
make_random_uuid() noexcept {
|
|
static thread_local std::mt19937_64 engine(std::random_device().operator()());
|
|
static thread_local std::uniform_int_distribution<uint64_t> dist;
|
|
uint64_t msb, lsb;
|
|
msb = dist(engine);
|
|
lsb = dist(engine);
|
|
msb &= ~uint64_t(0x0f << 12);
|
|
msb |= 0x4 << 12; // version 4
|
|
lsb &= ~(uint64_t(0x3) << 62);
|
|
lsb |= uint64_t(0x2) << 62; // IETF variant
|
|
return UUID(msb, lsb);
|
|
}
|
|
|
|
std::ostream& operator<<(std::ostream& out, const UUID& uuid) {
|
|
fmt::print(out, "{}", uuid);
|
|
return out;
|
|
}
|
|
|
|
UUID::UUID(sstring_view uuid) {
|
|
sstring uuid_string(uuid.begin(), uuid.end());
|
|
boost::erase_all(uuid_string, "-");
|
|
auto size = uuid_string.size() / 2;
|
|
if (size != 16) {
|
|
throw marshal_exception(format("UUID string size mismatch: '{}'", uuid));
|
|
}
|
|
sstring most = sstring(uuid_string.begin(), uuid_string.begin() + size);
|
|
sstring least = sstring(uuid_string.begin() + size, uuid_string.end());
|
|
int base = 16;
|
|
try {
|
|
std::size_t pos = 0;
|
|
this->most_sig_bits = std::stoull(most, &pos, base);
|
|
if (pos != most.size()) {
|
|
throw std::invalid_argument("");
|
|
}
|
|
this->least_sig_bits = std::stoull(least, &pos, base);
|
|
if (pos != least.size()) {
|
|
throw std::invalid_argument("");
|
|
}
|
|
} catch (const std::logic_error&) {
|
|
throw marshal_exception(format("invalid UUID: '{}'", uuid));
|
|
}
|
|
}
|
|
|
|
}
|