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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
|
#include "connection.h"
#include "error.h"
#include "selectcommand.h"
#include "modifycommand.h"
#include "reflection.h"
class Opts {
public:
Opts() { port = 3306; }
std::string server;
std::string user;
std::string password;
std::string database;
unsigned int port;
std::string unix_socket;
static Reflector<Opts>::Vars vars;
};
Reflector<Opts>::Vars Opts::vars = {
Map(Opts, server),
Map(Opts, user),
Map(Opts, password),
Map(Opts, database),
Map(Opts, unix_socket),
Map(Opts, port),
};
MySQL::Connection::Connection(const std::string & str) :
txDepth(0),
rolledback(false)
{
Opts o(Reflector<Opts>::NameValueNew(str));
mysql_init(&conn);
if (mysql_real_connect(&conn, o.server.c_str(), o.user.c_str(), o.password.c_str(), o.database.c_str(),
o.port, o.unix_socket.c_str(), CLIENT_LOCAL_FILES | CLIENT_MULTI_STATEMENTS) == NULL) {
throw ConnectionError();
}
if (mysql_set_character_set(&conn, "utf8")) {
throw ConnectionError();
}
}
MySQL::Connection::~Connection()
{
mysql_close(&conn);
}
void
MySQL::Connection::finish() const
{
if (txDepth != 0) {
rollbackTx();
throw Error("Transaction still open");
}
}
int
MySQL::Connection::beginTx() const
{
if (txDepth == 0) {
checkResult(mysql_autocommit(&conn, 0), true);
rolledback = false;
}
return ++txDepth;
}
int
MySQL::Connection::commitTx() const
{
if (rolledback) {
return rollbackTx();
}
if (--txDepth == 0) {
checkResult(mysql_commit(&conn), true);
}
return txDepth;
}
int
MySQL::Connection::rollbackTx() const
{
if (--txDepth == 0) {
checkResult(mysql_rollback(&conn), true);
}
else {
rolledback = true;
}
return txDepth;
}
bool
MySQL::Connection::inTx() const
{
return txDepth;
}
DB::BulkDeleteStyle
MySQL::Connection::bulkDeleteStyle() const
{
return DB::BulkDeleteUsingUsingAlias;
}
DB::BulkUpdateStyle
MySQL::Connection::bulkUpdateStyle() const
{
return DB::BulkUpdateUsingJoin;
}
void
MySQL::Connection::ping() const
{
checkResult(mysql_ping(&conn), true);
}
DB::SelectCommand *
MySQL::Connection::newSelectCommand(const std::string & sql) const
{
return new SelectCommand(this, sql);
}
DB::ModifyCommand *
MySQL::Connection::newModifyCommand(const std::string & sql) const
{
return new ModifyCommand(this, sql);
}
void
MySQL::Connection::checkResult(my_bool actual, my_bool expected) const
{
if (actual != expected) {
throw Error(mysql_error(&conn));
}
}
void
MySQL::Connection::beginBulkUpload(const char * table, const char * extra) const
{
(void)table;
(void)extra;
}
void
MySQL::Connection::endBulkUpload(const char * msg) const
{
(void)msg;
}
size_t
MySQL::Connection::bulkUploadData(const char * data, size_t len) const
{
(void)data;
return len;
}
|
