Files
scylladb/utils/uuid.cc
Asias He 2888c3086c utils: Add uuid_xor_to_uint32 helper
Convert the uuid to a uint32_t using xor.
It is useful to get a uint32_t number from the uuid.

Refs: #16927

Closes scylladb/scylladb#17049
2024-02-01 10:27:55 +02:00

71 lines
1.9 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 "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));
}
}
uint32_t uuid_xor_to_uint32(const UUID& uuid) {
size_t h = std::hash<utils::UUID>{}(uuid);
return uint32_t(h);
}
}