-
Notifications
You must be signed in to change notification settings - Fork 0
/
lua_db.c
90 lines (81 loc) · 1.82 KB
/
lua_db.c
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
// lua_db.c
//
// Lua postgresql stored procedure bindings
//
// © 2010 David J. Goehrig
//
#include "include.h"
#include "defines.h"
#include "memory.h"
#include "log.h"
#include "headers.h"
#include "database.h"
#include <lua.h>
#include <lauxlib.h>
#include <lualib.h>
extern lua_State* lins;
extern str lins_buffer;
extern Database lins_database;
extern Database db;
str lua2str(int index);
void str2lua(str s);
Headers stored_procedures;
static int DBClosureLua(lua_State* l)
{
if (! lins_database) lins_database = new_database();
int n = lua_gettop(l);
str qry = lua2str(lua_upvalueindex(1));
for (int i = 1; i <= n; ++i)
qry = append(qry,_("'%s'%s", lua2str(i), i < n ? _(","): _("")));
lua_pop(l,n);
qry = append(qry,_(")"));
int res = query(qry);
if (res < 0) {
str2lua(db_error());
return 1;
}
lua_newtable(l);
n = lua_gettop(l);
for (int i = 0; i < res; ++i) {
lua_pushnumber(l,i+1);
lua_newtable(l);
for (int j = 0; j < fields(); ++j) {
str2lua(field(j));
str2lua(fetch(i,j));
lua_settable(l,n+2);
}
lua_settable(l,n);
}
return 1;
}
int PrepareProcedures(char* schema)
{
new_database();
str sql = _("SELECT * from functions_in('%c');", schema);
int res = query(sql);
if (res < 0) {
dblog("%s",db_error());
close_database(db);
return 1;
}
stored_procedures = new_headers();
for (int i = 0; i < res && i < MAX_HEADERS; ++i) {
str proc = fetch(i,0);
append_header(stored_procedures,proc,_("SELECT * FROM %s(",proc));
}
close_database(db);
return 0;
}
int ProceduresLua(lua_State* l)
{
int i;
int n = lua_gettop(l);
lua_pop(l,n);
over (stored_procedures,i) {
str2lua(Key(stored_procedures,i)); // once to bind it
str2lua(Value(stored_procedures,i)); // once to store base query
lua_pushcclosure(l,DBClosureLua,1);
lua_settable(l,LUA_GLOBALSINDEX);
}
return 0;
}