Sun, 23 Nov 2008 04:12:04 +0000
Cleanup and commenting
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 | /* |
6 | * connection = DBD.MySQl.New(dbname, user, password, host, port) | |
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 | ||
45 | if (mysql_real_connect(conn->mysql, host, user, password, db, port, unix_socket, client_flag)) { | |
46 | luaL_getmetatable(L, DBD_MYSQL_CONNECTION); | |
47 | lua_setmetatable(L, -2); | |
48 | } else { | |
49 | luaL_error(L, "Failed to connect to database: %s", mysql_error(conn->mysql)); | |
50 | lua_pushnil(L); | |
51 | } | |
52 | ||
53 | return 1; | |
54 | } | |
55 | ||
2 | 56 | /* |
57 | * success = connection:close() | |
58 | */ | |
1 | 59 | static int connection_close(lua_State *L) { |
60 | connection_t *conn = (connection_t *)luaL_checkudata(L, 1, DBD_MYSQL_CONNECTION); | |
61 | int disconnect = 0; | |
62 | ||
63 | if (conn->mysql) { | |
64 | mysql_close(conn->mysql); | |
65 | disconnect = 1; | |
66 | } | |
67 | ||
68 | lua_pushboolean(L, disconnect); | |
69 | return 1; | |
70 | } | |
71 | ||
2 | 72 | /* |
73 | * ok = connection:ping() | |
74 | */ | |
1 | 75 | static int connection_ping(lua_State *L) { |
76 | connection_t *conn = (connection_t *)luaL_checkudata(L, 1, DBD_MYSQL_CONNECTION); | |
77 | int err = 1; | |
78 | ||
79 | if (conn->mysql) { | |
80 | err = mysql_ping(conn->mysql); | |
81 | } | |
82 | ||
83 | lua_pushboolean(L, !err); | |
84 | return 1; | |
85 | } | |
86 | ||
2 | 87 | /* |
88 | * statement = connection:prepare(sql_string) | |
89 | */ | |
1 | 90 | static int connection_prepare(lua_State *L) { |
91 | connection_t *conn = (connection_t *)luaL_checkudata(L, 1, DBD_MYSQL_CONNECTION); | |
92 | ||
93 | if (conn->mysql) { | |
94 | return dbd_mysql_statement_create(L, conn, luaL_checkstring(L, 2)); | |
95 | } | |
96 | ||
97 | lua_pushnil(L); | |
98 | return 1; | |
99 | } | |
100 | ||
2 | 101 | /* |
102 | * __gc | |
103 | */ | |
1 | 104 | static int connection_gc(lua_State *L) { |
105 | /* always close the connection */ | |
106 | connection_close(L); | |
107 | ||
108 | return 0; | |
109 | } | |
110 | ||
2 | 111 | int dbd_mysql_connection(lua_State *L) { |
112 | static const luaL_Reg connection_methods[] = { | |
113 | {"close", connection_close}, | |
114 | {"ping", connection_ping}, | |
115 | {"prepare", connection_prepare}, | |
116 | {NULL, NULL} | |
117 | }; | |
1 | 118 | |
2 | 119 | static const luaL_Reg connection_class_methods[] = { |
120 | {"New", connection_new}, | |
121 | {NULL, NULL} | |
122 | }; | |
1 | 123 | |
124 | luaL_newmetatable(L, DBD_MYSQL_CONNECTION); | |
125 | luaL_register(L, 0, connection_methods); | |
126 | lua_pushvalue(L,-1); | |
127 | lua_setfield(L, -2, "__index"); | |
128 | ||
129 | lua_pushcfunction(L, connection_gc); | |
130 | lua_setfield(L, -2, "__gc"); | |
131 | ||
132 | luaL_register(L, DBD_MYSQL_CONNECTION, connection_class_methods); | |
133 | ||
134 | return 1; | |
135 | } | |
136 |