blob: 5828b2632368ff7ce032e1f60669e909c1562807 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
#pragma once
#include <cstddef>
#include <type_traits>
#include <util/generic/vector.h>
#include <util/generic/string.h>
#include <util/generic/maybe.h>
namespace NSQLComplete {
template <class T>
struct TByteSize;
template <class T>
requires std::is_fundamental_v<T>
struct TByteSize<T> {
size_t operator()(const T& x) const noexcept {
return sizeof(x);
}
};
template <class T>
struct TByteSize<TVector<T>> {
size_t operator()(const TVector<T>& x) const noexcept {
size_t bytes = sizeof(x);
bytes = Accumulate(x, bytes, [](size_t acc, const T& x) {
return acc + TByteSize<T>()(x);
});
bytes += x.capacity() * sizeof(T);
return bytes;
}
};
template <>
struct TByteSize<TString> {
size_t operator()(const TString& x) const noexcept {
return std::max(sizeof(x), sizeof(x) + x.capacity());
}
};
template <class T>
struct TByteSize<TMaybe<T>> {
size_t operator()(const TMaybe<T>& x) const noexcept {
return x.Transform([](const T& x) { return TByteSize<T>()(x) - sizeof(T); }).GetOrElse(0) +
sizeof(TMaybe<T>);
}
};
template <class T>
concept CByteSized = requires(const T& x) {
{ TByteSize<T>()(x) } -> std::convertible_to<std::size_t>;
};
} // namespace NSQLComplete
|
