写个Sqlite3到Lua的library

前言：上次自己给编写Lua调用的dll留了个记号，这次正好写了个小的程序来试下。因为接到个任务是从sqlite3数据库里拿数据并且做解析分组之类的工作。当时自然最简单的想法就是，直接在Lua中能操作sqlite3数据库就可以了嘛。就写了个小的dll，封装了3个最简单的函数给Lua：openDB、closeDB哈exec。好吧，这个就算是上次写的一个小补充吧。中间还有碰到给Lua返回整张表数据的问题，不过我没有铺开，我对Lua也是一知半解，有兴趣的自己研究下。

 

#include "..//comm.h" #include <string> #include <sstream> #include "sqlite3.h" static sqlite3 *pDB = NULL; // open database int luacf_openDB(lua_State* L) { std::string dbName = lua_tostring(L, 1); lua_Integer rc = sqlite3_open(dbName.c_str(), &pDB); lua_pushinteger(L, rc); return 1; } // close database int lucacf_closeDB(lua_State* L) { if (pDB) { lua_Integer rc = sqlite3_close(pDB); lua_pushinteger(L, rc); return 1; } lua_pushboolean(L, 0); lua_pushstring(L, "数据库对象为空，不能关闭！"); return 2; } // exec int luacf_exec(lua_State* L) { bool bRet = false; if (pDB) { sqlite3_stmt* pStmt = NULL; lua_Integer rc = sqlite3_prepare(pDB, lua_tostring(L, 1), lua_strlen(L, 1), &pStmt, NULL); lua_pushinteger(L, rc); while (rc == SQLITE_OK) { int nColumn = sqlite3_column_count(pStmt); if (nColumn <= 0) { sqlite3_finalize(pStmt); bRet = true; lua_pushboolean(L, 1); lua_pushnil(L); break; } // create result_data_table lua_newtable(L); int nTop = lua_gettop(L); int nIndex = 0; while (SQLITE_ROW == sqlite3_step(pStmt)) { // create row_data_table lua_newtable(L); int nRowTop = lua_gettop(L); for (int i = 1; i <= nColumn; i ++) { const unsigned char* pszData = sqlite3_column_text(pStmt, i - 1); // lua_pushinteger(L, i); if (pszData) lua_pushstring(L, (const char*)pszData); else lua_pushstring(L, ""); // set data to row_data_table lua_rawseti(L, nRowTop, i); // lua_settable(L, nRowTop); } // set row_data_table to result_data_table lua_rawseti(L, nTop, ++nIndex); // lua_pushinteger(L, nIndex++); // lua_insert(L, nRowTop); // lua_settable(L, nTop); } sqlite3_finalize(pStmt); lua_pushinteger(L, nIndex); break; } // end while (rc == SQLITE_OK) return 3; } // end if (pDB) lua_pushboolean(L, bRet); return 1; } // library luaL_reg libs[] = { {"openDB", luacf_openDB}, {"closeDB", lucacf_closeDB}, {"exec", luacf_exec}, {NULL, NULL} }; LUALIB_API int luaopen_luaSqlite3(lua_State* L) { luaL_register(L, "sqlite3lib", libs); return 1; } 

 

程序没什么好讲的，路上就遇到一个问题，返回二维表。说白了就是往表里面塞表。在Lua脚本中这是很简单的事情，但是跑到C中，就有点晕了。不过Lua还是提供了对应的操作，只是没仔细看仔细搜。先来看下C往Lua中如何插入表格吧。

int luacf_gettable(lua_State* L) { lua_newtable(L); // 新建table int nTop = lua_gettop(L); // 得到这个table在栈中的位置（不是简单的栈顶） CString str1 = "str"; const char* prefix = "tail "; for(int i = 1; i <= 10; i++, str += prefix) { lua_pushnumber(L, i); // 设置key lua_pushstring(L, str); // 设置value lua_settable(L, nTop); // 将键值对设置入对应的表中 } return 1; } 

这就是一个字符串数组table，有10条记录。但是插入value是通过lua_pushxxx设置的，但是并没有提供lua_pushtable这样的函数，那怎么办呢？lua还提供了一个lua_insert的函数，呵呵，这个可以将堆栈中的值放到你指定的位置。还有个函数更牛逼了，lua_rawseti，就像我开始代码中写的那样，文档中说是这个函数效率高。看来在插入二维表的时候，不仅仅是内部效率高，连代码行都省了好多，呵呵。