mirrors/cosmopolitan
1
0
Fork 0
mirror of https://github.com/jart/cosmopolitan.git synced 2025-08-08 19:00:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add config options to redbean SQLite

Browse source
This commit is contained in:
Paul Kulchenko 2023-01-16 10:41:20 -08:00
parent 59f14c7046
commit c5d8d915b3
1 changed files with 62 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

63
tool/net/lsqlite3.c
View file

@ -131,6 +131,9 @@ static int sqlite_ses_meta_ref;
static int sqlite_reb_meta_ref; static int sqlite_reb_meta_ref;
static int sqlite_itr_meta_ref; static int sqlite_itr_meta_ref;
#endif #endif
/* global config configuration */
static int log_cb = LUA_NOREF; /* log callback */
static int log_udata;
/* /*
** ======================================================= ** =======================================================
@ -613,7 +616,7 @@ static sdb *newdb (lua_State *L) {
db->commit_hook_udata = db->commit_hook_udata =
db->rollback_hook_cb = db->rollback_hook_cb =
db->rollback_hook_udata = db->rollback_hook_udata =
LUA_NOREF; LUA_NOREF;
luaL_getmetatable(L, sqlite_meta); luaL_getmetatable(L, sqlite_meta);
lua_setmetatable(L, -2); /* set metatable */ lua_setmetatable(L, -2); /* set metatable */
@ -2443,6 +2446,58 @@ static int lsqlite_open_memory(lua_State *L) {
return lsqlite_do_open(L, ":memory:", SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE); return lsqlite_do_open(L, ":memory:", SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE);
} }
/*
** Log callback:
** Params: user, result code, log message
** Returns: none
*/
static void log_callback(void* user, int rc, const char *msg) {
if (log_cb != LUA_NOREF) {
lua_State *L = (lua_State*)user;
/* setup lua callback call */
int top = lua_gettop(L);
lua_rawgeti(L, LUA_REGISTRYINDEX, log_cb); /* get callback */
lua_rawgeti(L, LUA_REGISTRYINDEX, log_udata); /* get callback user data */
lua_pushinteger(L, rc);
lua_pushstring(L, msg);
if (lua_pcall(L, 3, 0, 0) != LUA_OK) lua_error(L);
lua_settop(L, top);
}
}
static int lsqlite_config(lua_State *L) {
int config = luaL_checkint(L, 1);
int result = SQLITE_OK;
switch (config) {
case SQLITE_CONFIG_LOG:
luaL_unref(L, LUA_REGISTRYINDEX, log_cb);
luaL_unref(L, LUA_REGISTRYINDEX, log_udata);
if (lua_gettop(L) < 2 || lua_isnil(L, 2)) {
log_cb =
log_udata = LUA_NOREF;
}
else {
luaL_checktype(L, 2, LUA_TFUNCTION);
/* make sure we have an userdata field (even if nil) */
lua_settop(L, 3);
log_udata = luaL_ref(L, LUA_REGISTRYINDEX);
log_cb = luaL_ref(L, LUA_REGISTRYINDEX);
}
break;
default:
result = SQLITE_MISUSE;
break;
}
lua_pushinteger(L, result);
return 1;
}
static int lsqlite_newindex(lua_State *L) { static int lsqlite_newindex(lua_State *L) {
lua_pushliteral(L, "attempt to change readonly table"); lua_pushliteral(L, "attempt to change readonly table");
lua_error(L); lua_error(L);
@ -2530,6 +2585,9 @@ static const struct {
SC(OPEN_SHAREDCACHE) SC(OPEN_SHAREDCACHE)
SC(OPEN_PRIVATECACHE) SC(OPEN_PRIVATECACHE)
/* config flags */
SC(CONFIG_LOG)
/* checkpoint flags */ /* checkpoint flags */
SC(CHECKPOINT_PASSIVE) SC(CHECKPOINT_PASSIVE)
SC(CHECKPOINT_FULL) SC(CHECKPOINT_FULL)
@ -2724,6 +2782,7 @@ static const luaL_Reg sqlitelib[] = {
{"version", lsqlite_version }, {"version", lsqlite_version },
{"open", lsqlite_open }, {"open", lsqlite_open },
{"open_memory", lsqlite_open_memory }, {"open_memory", lsqlite_open_memory },
{"config", lsqlite_config },
{"__newindex", lsqlite_newindex }, {"__newindex", lsqlite_newindex },
{NULL, NULL} {NULL, NULL}
@ -2743,6 +2802,8 @@ static void create_meta(lua_State *L, const char *name, const luaL_Reg *lib) {
} }
LUALIB_API int luaopen_lsqlite3(lua_State *L) { LUALIB_API int luaopen_lsqlite3(lua_State *L) {
/* call config before calling initialize */
sqlite3_config(SQLITE_CONFIG_LOG, log_callback, L);
sqlite3_initialize(); sqlite3_initialize();
create_meta(L, sqlite_meta, dblib); create_meta(L, sqlite_meta, dblib);