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