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
|
#include "generator/ugc_db.hpp"
#include "base/logging.hpp"
#include "base/macros.hpp"
#include <sqlite3.h>
#include <sstream>
namespace
{
struct Results
{
bool empty = true;
std::ostringstream values;
};
} // namespace
namespace generator
{
static int callback(void * results_ptr, int argc, char ** argv, char ** azColName)
{
Results & results = *reinterpret_cast<Results *>(results_ptr);
for (int i = 0; i < argc; i++)
{
if (results.empty)
results.empty = false;
else
results.values << ",";
results.values << (argv[i] ? argv[i] : "{}");
}
return 0;
}
UGCDB::UGCDB(std::string const & path)
{
int rc = sqlite3_open(path.c_str(), &m_db);
if (rc)
{
LOG(LERROR, ("Can't open database:", sqlite3_errmsg(m_db)));
sqlite3_close(m_db);
m_db = nullptr;
}
}
UGCDB::~UGCDB()
{
if (m_db)
sqlite3_close(m_db);
}
bool UGCDB::Get(base::GeoObjectId const & id, std::vector<uint8_t> & blob)
{
if (!m_db)
return false;
Results results;
results.values << "[";
std::ostringstream cmd;
cmd << "SELECT value FROM ratings WHERE key=" << id.GetEncodedId() << ";";
char * zErrMsg = nullptr;
auto rc = sqlite3_exec(m_db, cmd.str().c_str(), callback, &results, &zErrMsg);
if (rc != SQLITE_OK)
{
LOG(LERROR, ("SQL error:", zErrMsg));
sqlite3_free(zErrMsg);
return false;
}
results.values << "]";
return ValueToBlob(results.values.str(), blob);
}
bool UGCDB::Exec(std::string const & statement)
{
if (!m_db)
return false;
char * zErrMsg = nullptr;
auto rc = sqlite3_exec(m_db, statement.c_str(), nullptr, nullptr, &zErrMsg);
if (rc != SQLITE_OK)
{
LOG(LERROR, ("SQL error:", zErrMsg));
sqlite3_free(zErrMsg);
return false;
}
return true;
}
bool UGCDB::ValueToBlob(std::string const & src, std::vector<uint8_t> & blob)
{
blob.assign(src.cbegin(), src.cend());
return true;
}
} // namespace generator
|
