Wed, 26 Nov 2008 10:01:03 +0000
Cleanup and 'assert' error handling.
1 | 1 | #include "dbd_sqlite3.h" |
2 | ||
3 | int dbd_sqlite3_statement_create(lua_State *L, connection_t *conn, const char *sql_query); | |
4 | ||
2 | 5 | /* |
3 | 6 | * connection,err = DBD.SQLite3.New(dbfile) |
2 | 7 | */ |
1 | 8 | static int connection_new(lua_State *L) { |
9 | int n = lua_gettop(L); | |
10 | ||
11 | const char *db = NULL; | |
12 | connection_t *conn = NULL; | |
13 | ||
14 | /* db */ | |
15 | switch (n) { | |
3 | 16 | case 1: |
1 | 17 | if (lua_isnil(L, 1) == 0) |
18 | db = luaL_checkstring(L, 1); | |
19 | } | |
20 | ||
21 | conn = (connection_t *)lua_newuserdata(L, sizeof(connection_t)); | |
22 | ||
3 | 23 | if (sqlite3_open(db, &conn->sqlite) != SQLITE_OK) { |
1 | 24 | lua_pushnil(L); |
3 | 25 | lua_pushfstring(L, "Failed to connect to database: %s", sqlite3_errmsg(conn->sqlite)); |
26 | return 2; | |
1 | 27 | } |
28 | ||
3 | 29 | luaL_getmetatable(L, DBD_SQLITE_CONNECTION); |
30 | lua_setmetatable(L, -2); | |
31 | ||
1 | 32 | return 1; |
33 | } | |
34 | ||
2 | 35 | /* |
36 | * success = connection:close() | |
37 | */ | |
1 | 38 | static int connection_close(lua_State *L) { |
39 | connection_t *conn = (connection_t *)luaL_checkudata(L, 1, DBD_SQLITE_CONNECTION); | |
40 | int disconnect = 0; | |
41 | ||
42 | if (conn->sqlite) { | |
3 | 43 | sqlite3_close(conn->sqlite); |
44 | disconnect = 1; | |
45 | conn->sqlite = NULL; | |
1 | 46 | } |
47 | ||
48 | lua_pushboolean(L, disconnect); | |
49 | return 1; | |
50 | } | |
51 | ||
3 | 52 | |
2 | 53 | /* |
54 | * ok = connection:ping() | |
55 | */ | |
1 | 56 | static int connection_ping(lua_State *L) { |
57 | connection_t *conn = (connection_t *)luaL_checkudata(L, 1, DBD_SQLITE_CONNECTION); | |
58 | int ok = 0; | |
59 | ||
60 | if (conn->sqlite) { | |
61 | ok = 1; | |
62 | } | |
63 | ||
64 | lua_pushboolean(L, ok); | |
65 | return 1; | |
66 | } | |
67 | ||
2 | 68 | /* |
3 | 69 | * statement,err = connection:prepare(sql_str) |
2 | 70 | */ |
1 | 71 | static int connection_prepare(lua_State *L) { |
72 | connection_t *conn = (connection_t *)luaL_checkudata(L, 1, DBD_SQLITE_CONNECTION); | |
73 | ||
74 | if (conn->sqlite) { | |
75 | return dbd_sqlite3_statement_create(L, conn, luaL_checkstring(L, 2)); | |
76 | } | |
77 | ||
78 | lua_pushnil(L); | |
3 | 79 | lua_pushstring(L, "Connection not available"); |
80 | return 2; | |
1 | 81 | } |
82 | ||
2 | 83 | /* |
84 | * __gc | |
85 | */ | |
1 | 86 | static int connection_gc(lua_State *L) { |
87 | /* always close the connection */ | |
88 | connection_close(L); | |
89 | ||
90 | return 0; | |
91 | } | |
92 | ||
2 | 93 | int dbd_sqlite3_connection(lua_State *L) { |
94 | /* | |
95 | * instance methods | |
96 | */ | |
97 | static const luaL_Reg connection_methods[] = { | |
98 | {"close", connection_close}, | |
99 | {"ping", connection_ping}, | |
100 | {"prepare", connection_prepare}, | |
101 | {NULL, NULL} | |
102 | }; | |
1 | 103 | |
2 | 104 | /* |
105 | * class methods | |
106 | */ | |
107 | static const luaL_Reg connection_class_methods[] = { | |
108 | {"New", connection_new}, | |
109 | {NULL, NULL} | |
110 | }; | |
1 | 111 | |
112 | luaL_newmetatable(L, DBD_SQLITE_CONNECTION); | |
113 | luaL_register(L, 0, connection_methods); | |
114 | lua_pushvalue(L,-1); | |
115 | lua_setfield(L, -2, "__index"); | |
116 | ||
117 | lua_pushcfunction(L, connection_gc); | |
118 | lua_setfield(L, -2, "__gc"); | |
119 | ||
120 | luaL_register(L, DBD_SQLITE_CONNECTION, connection_class_methods); | |
121 | ||
122 | return 1; | |
123 | } |