-
Notifications
You must be signed in to change notification settings - Fork 1
/
pqxx_extensions.h
69 lines (55 loc) · 1.89 KB
/
pqxx_extensions.h
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
#pragma once
#include <pqxx/pqxx>
struct placeholder {};
// stupid library doesnt support UUIDs
struct UUID {
unsigned char bytes[16];
};
// TODO: replace binarystring completely?
// binarystring isn't default constructible which trolls the shitty stream code
struct binary : pqxx::binarystring {
using pqxx::binarystring::binarystring;
binary() : pqxx::binarystring(std::string_view{}) {}
binary(const pqxx::binarystring& data) : pqxx::binarystring(data) {}
};
inline unsigned hex_to_digit(char hex) {
auto x = static_cast<unsigned char>(hex);
if (x >= '0' and x <= '9')
return x - '0';
else if (x >= 'a' and x <= 'f')
return 10 + x - 'a';
else if (x >= 'A' and x <= 'F')
return 10 + x - 'A';
else
throw std::runtime_error{"Invalid hex in bytea."};
}
namespace pqxx {
template<>
struct string_traits<binary> : string_traits<pqxx::binarystring> {};
template<> struct nullness<binary> : no_null<pqxx::binarystring> {};
template<>
struct string_traits<placeholder> {
static placeholder from_string(std::string_view) {
return {};
}
};
template<> struct nullness<placeholder> : no_null<placeholder> {};
template<>
struct string_traits<UUID> {
static UUID from_string(std::string_view text) {
namespace views = std::ranges::views;
std::string hex;
for (char c : text) {
if (c != '-') hex.push_back(c);
}
if (hex.size() > 32) throw std::runtime_error("bad uuid string");
UUID uuid{};
for (int i = 0; i < 16; i += 2) {
auto hi = hex_to_digit(hex[i]), lo = hex_to_digit(hex[i + 1]);
uuid.bytes[i] = static_cast<unsigned char>((hi << 4) | lo);
}
return uuid;
}
};
template<> struct nullness<UUID> : no_null<UUID> {};
}