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
|
#include "dbUtils.h"
#include <atomic>
#include <boost/algorithm/string/join.hpp>
#include <boost/lexical_cast.hpp>
#include <buffer.h>
#include <tablepatch.h>
namespace Gentoo {
namespace Utils {
namespace Database {
std::string
emptyClone(DB::Connection * db, const std::string & orig)
{
auto tempTable = orig;
auto dot = tempTable.rfind('.');
if (dot != std::string::npos) {
tempTable = tempTable.substr(dot + 1);
}
tempTable += "_clone_" + boost::lexical_cast<std::string>(db);
db->execute("CREATE TEMPORARY TABLE " + tempTable + " AS SELECT * FROM " + orig + " WHERE false");
return tempTable;
}
std::pair<std::string, DB::ModifyCommandPtr>
namedTemp(DB::Connection * db, const std::string & tempTable,
const std::map<std::string, const std::string> & cols)
{
std::set<std::string> keys;
std::set<std::string> defs;
for (auto c : cols) {
keys.insert(c.first);
defs.insert(c.first + " " + c.second);
}
db->execute("CREATE TEMPORARY TABLE " + tempTable + "(" + boost::join(defs, ",") + ")");
return {tempTable, tablePatchInserter(db, tempTable, keys)};
}
void
drop(DB::Connection * db, const std::string & table)
{
db->execute("DROP TABLE " + table);
}
DB::ModifyCommandPtr
tablePatchInserter(DB::Connection * dbc, const DB::TablePatch & p)
{
return tablePatchInserter(dbc, p.src, p.cols);
}
DB::ModifyCommandPtr
tablePatchInserter(DB::Connection * dbc, const std::string & t, const std::set<std::string> & c)
{
return dbc->modify("INSERT INTO " + t + "(" + boost::algorithm::join(c, ", ") + ") VALUES("
+ boost::algorithm::join(std::vector<std::string>(c.size(), "?"), ", ") + ")");
}
}
}
}
|
