Wed, 03 Dec 2008 08:44:40 +0000
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
1 | 1 | #include "dbd_mysql.h" |
2 | ||
3 | int dbd_mysql_statement_create(lua_State *L, connection_t *conn, const char *sql_query); | |
4 | ||
2 | 5 | /* |
3 | 6 | * connection,err = DBD.MySQl.New(dbname, user, password, host, port) |
2 | 7 | */ |
1 | 8 | static int connection_new(lua_State *L) { |
9 | int n = lua_gettop(L); | |
10 | ||
11 | connection_t *conn = NULL; | |
12 | ||
13 | const char *host = NULL; | |
14 | const char *user = NULL; | |
15 | const char *password = NULL; | |
16 | const char *db = NULL; | |
17 | int port = 0; | |
18 | ||
19 | const char *unix_socket = NULL; /* TODO always NULL */ | |
20 | int client_flag = 0; /* TODO always 0, set flags from options table */ | |
21 | ||
22 | /* db, user, password, host, port */ | |
23 | switch (n) { | |
24 | case 5: | |
25 | if (lua_isnil(L, 5) == 0) | |
26 | port = luaL_checkint(L, 5); | |
27 | case 4: | |
28 | if (lua_isnil(L, 4) == 0) | |
29 | host = luaL_checkstring(L, 4); | |
30 | case 3: | |
31 | if (lua_isnil(L, 3) == 0) | |
32 | password = luaL_checkstring(L, 3); | |
33 | case 2: | |
34 | if (lua_isnil(L, 2) == 0) | |
35 | user = luaL_checkstring(L, 2); | |
36 | case 1: | |
37 | if (lua_isnil(L, 1) == 0) | |
38 | db = luaL_checkstring(L, 1); | |
39 | } | |
40 | ||
41 | conn = (connection_t *)lua_newuserdata(L, sizeof(connection_t)); | |
42 | ||
43 | conn->mysql = mysql_init(NULL); | |
44 | ||
3 | 45 | if (!mysql_real_connect(conn->mysql, host, user, password, db, port, unix_socket, client_flag)) { |
1 | 46 | lua_pushnil(L); |
4 | 47 | lua_pushfstring(L, DBI_ERR_CONNECTION_FAILED, mysql_error(conn->mysql)); |
3 | 48 | return 2; |
1 | 49 | } |
50 | ||
13
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
51 | /* |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
52 | * by default turn off autocommit |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
53 | */ |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
54 | mysql_autocommit(conn->mysql, 0); |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
55 | |
3 | 56 | luaL_getmetatable(L, DBD_MYSQL_CONNECTION); |
57 | lua_setmetatable(L, -2); | |
58 | ||
1 | 59 | return 1; |
60 | } | |
61 | ||
2 | 62 | /* |
13
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
63 | * success = connection:autocommit(on) |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
64 | */ |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
65 | static int connection_autocommit(lua_State *L) { |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
66 | connection_t *conn = (connection_t *)luaL_checkudata(L, 1, DBD_MYSQL_CONNECTION); |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
67 | int on = lua_toboolean(L, 2); |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
68 | int err = 0; |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
69 | |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
70 | if (conn->mysql) { |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
71 | err = mysql_autocommit(conn->mysql, on); |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
72 | } |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
73 | |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
74 | lua_pushboolean(L, !err); |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
75 | return 1; |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
76 | } |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
77 | |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
78 | /* |
2 | 79 | * success = connection:close() |
80 | */ | |
1 | 81 | static int connection_close(lua_State *L) { |
82 | connection_t *conn = (connection_t *)luaL_checkudata(L, 1, DBD_MYSQL_CONNECTION); | |
83 | int disconnect = 0; | |
84 | ||
85 | if (conn->mysql) { | |
86 | mysql_close(conn->mysql); | |
87 | disconnect = 1; | |
3 | 88 | conn->mysql = NULL; |
1 | 89 | } |
90 | ||
91 | lua_pushboolean(L, disconnect); | |
92 | return 1; | |
93 | } | |
94 | ||
13
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
95 | /* |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
96 | * success = connection:commit() |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
97 | */ |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
98 | static int connection_commit(lua_State *L) { |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
99 | connection_t *conn = (connection_t *)luaL_checkudata(L, 1, DBD_MYSQL_CONNECTION); |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
100 | int err = 0; |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
101 | |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
102 | if (conn->mysql) { |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
103 | err = mysql_commit(conn->mysql); |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
104 | } |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
105 | |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
106 | lua_pushboolean(L, !err); |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
107 | return 1; |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
108 | } |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
109 | |
2 | 110 | /* |
111 | * ok = connection:ping() | |
112 | */ | |
1 | 113 | static int connection_ping(lua_State *L) { |
114 | connection_t *conn = (connection_t *)luaL_checkudata(L, 1, DBD_MYSQL_CONNECTION); | |
115 | int err = 1; | |
116 | ||
117 | if (conn->mysql) { | |
118 | err = mysql_ping(conn->mysql); | |
119 | } | |
120 | ||
121 | lua_pushboolean(L, !err); | |
122 | return 1; | |
123 | } | |
124 | ||
2 | 125 | /* |
3 | 126 | * statement,err = connection:prepare(sql_string) |
2 | 127 | */ |
1 | 128 | static int connection_prepare(lua_State *L) { |
129 | connection_t *conn = (connection_t *)luaL_checkudata(L, 1, DBD_MYSQL_CONNECTION); | |
130 | ||
131 | if (conn->mysql) { | |
132 | return dbd_mysql_statement_create(L, conn, luaL_checkstring(L, 2)); | |
133 | } | |
134 | ||
135 | lua_pushnil(L); | |
4 | 136 | lua_pushstring(L, DBI_ERR_DB_UNAVAILABLE); |
3 | 137 | |
138 | return 2; | |
1 | 139 | } |
140 | ||
2 | 141 | /* |
13
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
142 | * success = connection:rollback() |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
143 | */ |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
144 | static int connection_rollback(lua_State *L) { |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
145 | connection_t *conn = (connection_t *)luaL_checkudata(L, 1, DBD_MYSQL_CONNECTION); |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
146 | int err = 0; |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
147 | |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
148 | if (conn->mysql) { |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
149 | err = mysql_rollback(conn->mysql); |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
150 | } |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
151 | |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
152 | lua_pushboolean(L, !err); |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
153 | return 1; |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
154 | } |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
155 | |
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
156 | /* |
2 | 157 | * __gc |
158 | */ | |
1 | 159 | static int connection_gc(lua_State *L) { |
160 | /* always close the connection */ | |
161 | connection_close(L); | |
162 | ||
163 | return 0; | |
164 | } | |
165 | ||
2 | 166 | int dbd_mysql_connection(lua_State *L) { |
167 | static const luaL_Reg connection_methods[] = { | |
13
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
168 | {"autocommit", connection_autocommit}, |
2 | 169 | {"close", connection_close}, |
13
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
170 | {"commit", connection_commit}, |
2 | 171 | {"ping", connection_ping}, |
172 | {"prepare", connection_prepare}, | |
13
10c8c6f0da14
Added connection:autocommit(), connection:commit(), and connection:rollback() to control transactions.
nrich@ii.net
parents:
4
diff
changeset
|
173 | {"rollback", connection_rollback}, |
2 | 174 | {NULL, NULL} |
175 | }; | |
1 | 176 | |
2 | 177 | static const luaL_Reg connection_class_methods[] = { |
178 | {"New", connection_new}, | |
179 | {NULL, NULL} | |
180 | }; | |
1 | 181 | |
182 | luaL_newmetatable(L, DBD_MYSQL_CONNECTION); | |
183 | luaL_register(L, 0, connection_methods); | |
184 | lua_pushvalue(L,-1); | |
185 | lua_setfield(L, -2, "__index"); | |
186 | ||
187 | lua_pushcfunction(L, connection_gc); | |
188 | lua_setfield(L, -2, "__gc"); | |
189 | ||
190 | luaL_register(L, DBD_MYSQL_CONNECTION, connection_class_methods); | |
191 | ||
192 | return 1; | |
193 | } | |
194 |