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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
#include "postgresHandler.h"
#include <cctype>
#include <cstddef>
#include <iostream>
#include <regex>
#include <stdexcept>
#define KEY_INDEX 1
#define INT_INDEX 3
#define STRING_INDEX 4
#define LIST_OF_INT_INDEX 5
#define NULL_INDEX 6
const pqxx::result execute_query(pqxx::work &t_transaction,
std::string_view t_query) {
return t_transaction.exec(t_query);
}
void display_query_result(const pqxx::result &t_r) {
const int nb_columns = t_r.columns();
for (const auto &row : t_r) {
for (int i = 0; i < nb_columns; i++) {
std::cout << row[i].c_str() << " | ";
}
std::cout << std::endl;
}
}
static const unsigned int
get_type_of_regex_iterator(const std::sregex_iterator &t_sregex_it) {
std::array<int, 4> types{INT_INDEX, STRING_INDEX, LIST_OF_INT_INDEX,
NULL_INDEX};
for (const int &type : types) {
if ((*t_sregex_it)[type] != "") {
return type;
}
}
throw std::invalid_argument("regex iterator input is invalid");
}
const std::string format_json_value_to_sql(const unsigned int &t_type_index,
const std::string &t_json_value) {
switch (t_type_index) {
case INT_INDEX:
return t_json_value;
case STRING_INDEX: {
std::string json_copy = t_json_value;
*json_copy.begin() = '\'';
*std::prev(json_copy.end()) = '\'';
return json_copy;
}
case NULL_INDEX:
return "NULL";
case LIST_OF_INT_INDEX: {
std::string json_copy = t_json_value;
json_copy.append("\'");
json_copy.insert(json_copy.begin(), '\'');
return json_copy;
}
default:
throw std::invalid_argument("type index is invalid: ");
}
}
std::string format_key(const std::string &key) {
std::string key_copy(key);
std::stack<size_t> uppercase_indexes;
for (size_t i = 0; i < key.length(); i++) {
if (std::isupper(key[i])) {
uppercase_indexes.push(i);
}
}
while (!uppercase_indexes.empty()) {
size_t top_index = uppercase_indexes.top();
uppercase_indexes.pop();
key_copy[top_index] = std::tolower(key_copy[top_index]);
key_copy.insert(std::next(key_copy.begin(), top_index), '_');
}
return key_copy;
}
void add_cards_to_database(const std::string &t_cards, pqxx::work &t_tx) {
std::regex key_regex(
"\"([^\"?]+)\":(([0-9]+)|(\"[^\"?]*\")|(\\[[0-9,]*\\])|(null))");
auto words_begin =
std::sregex_iterator(t_cards.begin(), t_cards.end(), key_regex);
auto words_end = std::sregex_iterator();
if (words_begin != words_end) {
std::string insert_columns("INSERT INTO cards (");
std::string values(" VALUES (");
for (auto it = words_begin; it != words_end; ++it) {
insert_columns.append(format_key((*it)[KEY_INDEX].str()) +
","); // key
const unsigned int type_index = get_type_of_regex_iterator(it);
values.append(
format_json_value_to_sql(type_index, (*it)[type_index].str()) +
",");
}
*std::prev(insert_columns.end()) = ')';
*std::prev(values.end()) = ')';
values.append(" ON CONFLICT (id) DO NOTHING;");
execute_query(t_tx, insert_columns + values);
}
}
|
