sq_lmdb.cpp 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709
  1. //#ifdef USE_LMDB
  2. #ifdef __cplusplus
  3. extern "C" {
  4. #endif
  5. #include "squirrel.h"
  6. #include <string.h>
  7. #include <stdio.h>
  8. #include <stdlib.h> /* for malloc */
  9. #include <assert.h> /* for a few sanity tests */
  10. SQ_OPT_STRING_STRLEN();
  11. #include "lmdb.h"
  12. static const SQChar SQ_LMDB_ENV[] = _SC("LmDB_Env");
  13. static const SQChar LMDB_ENV_Tag[] = _SC("sq_LmDB_Env_ctx");
  14. #define GET_LmDB_Env_INSTANCE() SQ_GET_INSTANCE(v, 1, MDB_env, LMDB_ENV_Tag) \
  15. if(self == NULL) return sq_throwerror(v, _SC("LmDB_Env object already closed"));
  16. static SQRESULT LmDB_Env_release_hook(SQUserPointer p, SQInteger size, void*)
  17. {
  18. MDB_env *self = (MDB_env*)p;
  19. if(self) mdb_env_close(self);
  20. return 0;
  21. }
  22. static SQRESULT LmDB_Env_constructor(HSQUIRRELVM v, MDB_env *LmDB_Env, int free_on_gc)
  23. {
  24. if(!LmDB_Env)
  25. return sq_throwerror(v, _SC("Could'nt create an LmDB_Env object."));
  26. sq_pushstring(v, SQ_LMDB_ENV, -1);
  27. if(sq_getonroottable(v) == SQ_OK){
  28. sq_pushstring(v, LMDB_ENV_Tag, -1);
  29. if(sq_get(v, -2) == SQ_OK){
  30. if(sq_createinstance(v, -1) == SQ_OK){
  31. sq_setinstanceup(v, -1, LmDB_Env);
  32. if(free_on_gc) sq_setreleasehook(v,-1, LmDB_Env_release_hook);
  33. return 1;
  34. }
  35. }
  36. }
  37. return SQ_ERROR;
  38. }
  39. static SQRESULT sq_LmDB_Env_close(HSQUIRRELVM v){
  40. SQ_FUNC_VARS_NO_TOP(v);
  41. GET_LmDB_Env_INSTANCE();
  42. LmDB_Env_release_hook(self, 0, v);
  43. sq_setinstanceup(v, 1, 0);
  44. return 0;
  45. }
  46. static SQRESULT sq_LmDB_Env_copy(HSQUIRRELVM v){
  47. SQ_FUNC_VARS_NO_TOP(v);
  48. GET_LmDB_Env_INSTANCE();
  49. SQ_GET_STRING(v, 2, path);
  50. // int mdb_env_copy(MDB_env *env, const char *path)
  51. sq_pushinteger(v, mdb_env_copy(self, path));
  52. return 1;
  53. }
  54. static SQRESULT sq_LmDB_Env_dbi_close(HSQUIRRELVM v){
  55. SQ_FUNC_VARS_NO_TOP(v);
  56. GET_LmDB_Env_INSTANCE();
  57. // void mdb_dbi_close(MDB_env *env, MDB_dbi dbi)
  58. //mdb_dbi_close(self, MDB_dbi dbi);
  59. return 0;
  60. }
  61. static SQRESULT sq_LmDB_Env_get_flags(HSQUIRRELVM v){
  62. SQ_FUNC_VARS_NO_TOP(v);
  63. GET_LmDB_Env_INSTANCE();
  64. unsigned int int_flags;
  65. // int mdb_env_get_flags(MDB_env *env, unsigned int *flags)
  66. mdb_env_get_flags(self, &int_flags);
  67. sq_pushinteger(v, (SQInteger)int_flags);
  68. return 1;
  69. }
  70. static SQRESULT sq_LmDB_Env_get_path(HSQUIRRELVM v){
  71. SQ_FUNC_VARS_NO_TOP(v);
  72. GET_LmDB_Env_INSTANCE();
  73. // int mdb_env_get_path(MDB_env *env, const char **path)
  74. const char *path;
  75. int err = mdb_env_get_path(self, &path);
  76. if(err) sq_pushinteger(v, err);
  77. else sq_pushstring(v, path, -1);
  78. return 1;
  79. }
  80. static SQRESULT sq_LmDB_Env_info(HSQUIRRELVM v){
  81. SQ_FUNC_VARS_NO_TOP(v);
  82. GET_LmDB_Env_INSTANCE();
  83. // int mdb_env_info(MDB_env *env, MDB_envinfo *stat)
  84. MDB_envinfo info;
  85. mdb_env_info(self,&info);
  86. sq_newtable(v);
  87. sq_pushliteral(v, _SC("me_mapsize"));
  88. sq_pushinteger(v,info.me_mapsize);
  89. sq_rawset(v, -3);
  90. sq_pushliteral(v, _SC("ms_last_pgno"));
  91. sq_pushinteger(v,info.me_last_pgno);
  92. sq_rawset(v, -3);
  93. sq_pushliteral(v, _SC("me_last_txnid"));
  94. sq_pushinteger(v,info.me_last_txnid);
  95. sq_rawset(v, -3);
  96. sq_pushliteral(v, _SC("me_maxreaders"));
  97. sq_pushinteger(v,info.me_maxreaders);
  98. sq_rawset(v, -3);
  99. sq_pushliteral(v, _SC("me_numreaders"));
  100. sq_pushinteger(v,info.me_numreaders);
  101. sq_rawset(v, -3);
  102. return 1;
  103. }
  104. static SQRESULT sq_LmDB_Env_open(HSQUIRRELVM v){
  105. SQ_FUNC_VARS_NO_TOP(v);
  106. GET_LmDB_Env_INSTANCE();
  107. SQ_GET_STRING(v, 2, path);
  108. SQ_GET_INTEGER(v, 3, int_flags);
  109. SQ_GET_INTEGER(v, 4, int_mode);
  110. // int mdb_env_open(MDB_env *env, const char *path, unsigned int flags, mdb_mode_t mode)
  111. sq_pushinteger(v, mdb_env_open(self, path, int_flags, int_mode));
  112. return 1;
  113. }
  114. static SQRESULT sq_LmDB_Env_set_flags(HSQUIRRELVM v){
  115. SQ_FUNC_VARS_NO_TOP(v);
  116. GET_LmDB_Env_INSTANCE();
  117. SQ_GET_INTEGER(v, 2, int_flags);
  118. SQ_GET_INTEGER(v, 3, int_onOff);
  119. // int mdb_env_set_flags(MDB_env *env, unsigned int flags, int onoff)
  120. sq_pushinteger(v, mdb_env_set_flags(self, int_flags, int_onOff));
  121. return 1;
  122. }
  123. static SQRESULT sq_LmDB_Env_set_mapsize(HSQUIRRELVM v){
  124. SQ_FUNC_VARS_NO_TOP(v);
  125. GET_LmDB_Env_INSTANCE();
  126. SQ_GET_INTEGER(v, 2, int_size);
  127. // int mdb_env_set_mapsize(MDB_env *env, mdb_size_t size)
  128. sq_pushinteger(v, mdb_env_set_mapsize(self, int_size));
  129. return 1;
  130. }
  131. static SQRESULT sq_LmDB_Env_set_maxdbs(HSQUIRRELVM v){
  132. SQ_FUNC_VARS_NO_TOP(v);
  133. GET_LmDB_Env_INSTANCE();
  134. SQ_GET_INTEGER(v, 2, int_dbs);
  135. // int mdb_env_set_maxdbs(MDB_env *env, MDB_dbi dbs)
  136. sq_pushinteger(v, mdb_env_set_maxdbs(self, int_dbs));
  137. return 1;
  138. }
  139. static SQRESULT sq_LmDB_Env_set_maxreaders(HSQUIRRELVM v){
  140. SQ_FUNC_VARS_NO_TOP(v);
  141. GET_LmDB_Env_INSTANCE();
  142. SQ_GET_INTEGER(v, 2, int_readers);
  143. // int mdb_env_set_maxreaders(MDB_env *env, unsigned int readers)
  144. sq_pushinteger(v, mdb_env_set_maxreaders(self, int_readers));
  145. return 1;
  146. }
  147. static SQRESULT stat_to_table(HSQUIRRELVM v, MDB_stat *stat){
  148. sq_newtable(v);
  149. sq_pushliteral(v, _SC("ms_psize"));
  150. sq_pushinteger(v,stat->ms_psize);
  151. sq_rawset(v, -3);
  152. sq_pushliteral(v, _SC("ms_depth"));
  153. sq_pushinteger(v,stat->ms_depth);
  154. sq_rawset(v, -3);
  155. sq_pushliteral(v, _SC("ms_branch_pages"));
  156. sq_pushinteger(v,stat->ms_branch_pages);
  157. sq_rawset(v, -3);
  158. sq_pushliteral(v, _SC("ms_leaf_pages"));
  159. sq_pushinteger(v,stat->ms_leaf_pages);
  160. sq_rawset(v, -3);
  161. sq_pushliteral(v, _SC("ms_overflow_pages"));
  162. sq_pushinteger(v,stat->ms_overflow_pages);
  163. sq_rawset(v, -3);
  164. sq_pushliteral(v, _SC("ms_entries"));
  165. sq_pushinteger(v,stat->ms_entries);
  166. sq_rawset(v, -3);
  167. return 1;
  168. }
  169. static SQRESULT sq_LmDB_Env_stat(HSQUIRRELVM v){
  170. SQ_FUNC_VARS_NO_TOP(v);
  171. GET_LmDB_Env_INSTANCE();
  172. // int mdb_env_stat(MDB_env *env, MDB_stat *stat)
  173. MDB_stat stat;
  174. mdb_env_stat(self, &stat);
  175. return stat_to_table(v, &stat);
  176. }
  177. static SQRESULT sq_LmDB_Env_sync(HSQUIRRELVM v){
  178. SQ_FUNC_VARS_NO_TOP(v);
  179. GET_LmDB_Env_INSTANCE();
  180. SQ_GET_INTEGER(v, 2, int_force);
  181. // int mdb_env_sync(MDB_env *env, int force)
  182. sq_pushinteger(v, mdb_env_sync(self, int_force));
  183. return 1;
  184. }
  185. static SQRESULT sq_LmDB_Env_txn_begin(HSQUIRRELVM v){
  186. SQ_FUNC_VARS_NO_TOP(v);
  187. GET_LmDB_Env_INSTANCE();
  188. // int mdb_txn_begin(MDB_env *env, MDB_txn *parent, unsigned int flags, MDB_txn **txn)
  189. return 0;
  190. }
  191. static SQRESULT sq_LmDB_strerror(HSQUIRRELVM v){
  192. SQ_FUNC_VARS_NO_TOP(v);
  193. GET_LmDB_INSTANCE();
  194. SQ_GET_INTEGER(v, 2, int_error);
  195. // char *mdb_strerror(int err)
  196. sq_pushstring(v, mdb_strerror(int_error), -1);
  197. return 1;
  198. }
  199. static SQRESULT sq_LmDB_version(HSQUIRRELVM v){
  200. SQ_FUNC_VARS_NO_TOP(v);
  201. GET_LmDB_INSTANCE();
  202. // char *mdb_version(int *major, int *minor, int *patch)
  203. int major, minor, patch;
  204. sq_pushstring(v, mdb_version(&major, &minot, &patch), -1);
  205. return 1;
  206. }
  207. #define _DECL_LMDB_ENV_FUNC(name,nparams,pmask) {_SC(#name),sq_LmDB_Env_##name,nparams,pmask}
  208. static SQRegFunction LmDB_Env_obj_funcs[]={
  209. _DECL_LMDB_ENV_FUNC(close, 1, _SC("x")),
  210. _DECL_LMDB_ENV_FUNC(copy, 2, _SC("x.")),
  211. _DECL_LMDB_ENV_FUNC(dbi_close, 1, _SC("x")),
  212. _DECL_LMDB_ENV_FUNC(get_flags, 2, _SC("xi")),
  213. _DECL_LMDB_ENV_FUNC(get_path, 1, _SC("x")),
  214. _DECL_LMDB_ENV_FUNC(info, 1, _SC("x")),
  215. _DECL_LMDB_ENV_FUNC(open, 4, _SC("xsii")),
  216. _DECL_LMDB_ENV_FUNC(set_flags, 3, _SC("xii")),
  217. _DECL_LMDB_ENV_FUNC(set_mapsize, 2, _SC("xi")),
  218. _DECL_LMDB_ENV_FUNC(set_maxdbs, 2, _SC("xi")),
  219. _DECL_LMDB_ENV_FUNC(set_maxreaders, 2, _SC("xi")),
  220. _DECL_LMDB_ENV_FUNC(stat, 1, _SC("x")),
  221. _DECL_LMDB_ENV_FUNC(sync, 2, _SC("xi")),
  222. _DECL_LMDB_ENV_FUNC(txn_begin, 1, _SC("x")),
  223. _DECL_LMDB_FUNC(strerror, 2, _SC(".i")),
  224. _DECL_LMDB_FUNC(version, 1, _SC(".")),
  225. {0,0}
  226. };
  227. #undef _DECL_LMDB_ENV_FUNC
  228. static const SQChar SQ_LMDB_TRANSACTION[] = _SC("LmDB_Transaction");
  229. static const SQChar LMDB_TRANSACTION_Tag[] = _SC("sq_LmDB_Transaction_ctx");
  230. #define GET_LmDB_Transaction_INSTANCE() SQ_GET_INSTANCE(v, 1, SQ_LMDB_TRANSACTION, LMDB_TRANSACTION_Tag) \
  231. if(self == NULL) return sq_throwerror(v, _SC("LmDB_Transaction object already closed"));
  232. static SQRESULT LmDB_Transaction_release_hook(SQUserPointer p, SQInteger size, void*)
  233. {
  234. LMDB_TRANSACTION *self = (LMDB_TRANSACTION*)p;
  235. if(self) LmDB_Transaction_free(self);
  236. return 0;
  237. }
  238. static SQRESULT LmDB_Transaction_free(HSQUIRRELVM v)
  239. {
  240. SQ_FUNC_VARS_NO_TOP(v);
  241. GET_LmDB_Transaction_INSTANCE();
  242. LmDB_Transaction_release_hook(self, 0, v);
  243. sq_setinstanceup(v, 1, 0);
  244. return 0;
  245. }
  246. static SQRESULT LmDB_Transaction_constructor(HSQUIRRELVM v, LMDB_TRANSACTION *LmDB_Transaction, int free_on_gc)
  247. {
  248. if(!LmDB_Transaction)
  249. return sq_throwerror(v, _SC("Could'nt create an LmDB_Transaction object."));
  250. sq_pushstring(v, SQ_LMDB_TRANSACTION, -1);
  251. if(sq_getonroottable(v) == SQ_OK){
  252. sq_pushstring(v, LMDB_TRANSACTION_Tag, -1);
  253. if(sq_get(v, -2) == SQ_OK){
  254. if(sq_createinstance(v, -1) == SQ_OK){
  255. sq_setinstanceup(v, -1, LmDB_Transaction);
  256. if(free_on_gc) sq_setreleasehook(v,-1, LmDB_Transaction_release_hook);
  257. return 1;
  258. }
  259. }
  260. }
  261. return SQ_ERROR;
  262. }
  263. static SQRESULT sq_LmDB_Transaction_abort(HSQUIRRELVM v){
  264. SQ_FUNC_VARS_NO_TOP(v);
  265. GET_LmDB_Transaction_INSTANCE();
  266. // void mdb_txn_abort(MDB_txn *txn)
  267. mdb_txn_abort(self);
  268. return 0;
  269. }
  270. static MDB_val* pop_val(HSQUIRRELVM v,int index, MDB_val* val) {
  271. if ( sq_gettype(v,index) == OT_NULL ) {
  272. return NULL;
  273. }
  274. SQInteger sz;
  275. sq_getstr_and_size(v, index, (const SQChar**)&val->mv_data, &sz);
  276. val->mv_size = sz;
  277. return val;
  278. }
  279. static SQRESULT txn_cmp_helper(HSQUIRRELVM v,int mode) {
  280. SQ_FUNC_VARS_NO_TOP(v);
  281. GET_LmDB_Transaction_INSTANCE();
  282. SQ_GET_INTEGER(v, 2, dbi);
  283. MDB_val key, value;
  284. pop_val(v, 3, &key);
  285. pop_val(v, 4, &value);
  286. if ( mode==MDB_DUPSORT )
  287. return sq_pushinteger(v, mdb_dcmp(self,dbi, &key, &value));
  288. return sq_pushinteger(v, mdb_cmp(self,dbi, &key, &value));
  289. }
  290. static SQRESULT sq_LmDB_Transaction_cmp(HSQUIRRELVM v){
  291. // int mdb_cmp(MDB_txn *txn, MDB_dbi dbi, const MDB_val *a, const MDB_val *b)
  292. return txn_cmp_helper(v, 0);
  293. }
  294. static SQRESULT sq_LmDB_Transaction_cursor_open(HSQUIRRELVM v){
  295. // int mdb_cursor_open(MDB_txn *txn, MDB_dbi dbi, MDB_cursor **cursor)
  296. return txn_cmp_helper(v, 1);
  297. }
  298. static SQRESULT sq_LmDB_Transaction_cursor_renew(HSQUIRRELVM v){
  299. SQ_FUNC_VARS_NO_TOP(v);
  300. GET_LmDB_Transaction_INSTANCE();
  301. // int mdb_cursor_renew(MDB_txn *txn, MDB_cursor *cursor)
  302. //sq_pushinteger(v, mdb_cursor_renew(self, ));
  303. return 1;
  304. }
  305. static SQRESULT sq_LmDB_Transaction_dbi_drop(HSQUIRRELVM v){
  306. SQ_FUNC_VARS_NO_TOP(v);
  307. GET_LmDB_Transaction_INSTANCE();
  308. SQ_GET_INTEGER(v, 2, dbi);
  309. SQ_GET_INTEGER(v, 3, del);
  310. // int mdb_drop(MDB_txn *txn, MDB_dbi dbi, int del)
  311. sq_pushinteger(v, mdb_drop(self, dbi, del));
  312. return 1;
  313. }
  314. static SQRESULT sq_LmDB_Transaction_dbi_open(HSQUIRRELVM v){
  315. SQ_FUNC_VARS_NO_TOP(v);
  316. GET_LmDB_Transaction_INSTANCE();
  317. SQ_GET_STRING(v, 2, name);
  318. SQ_GET_INTEGER(v, 3, int_flags);
  319. // int mdb_dbi_open(MDB_txn *txn, const char *name, unsigned int flags, MDB_dbi *dbi)
  320. return 0;
  321. }
  322. static SQRESULT sq_LmDB_Transaction_dcmp(HSQUIRRELVM v){
  323. SQ_FUNC_VARS_NO_TOP(v);
  324. GET_LmDB_Transaction_INSTANCE();
  325. // int mdb_dcmp(MDB_txn *txn, MDB_dbi dbi, const MDB_val *a, const MDB_val *b)
  326. return 0;
  327. }
  328. static SQRESULT sq_LmDB_Transaction_del(HSQUIRRELVM v){
  329. SQ_FUNC_VARS_NO_TOP(v);
  330. GET_LmDB_Transaction_INSTANCE();
  331. SQ_GET_STRING(v, 2, key);
  332. // int mdb_del(MDB_txn *txn, MDB_dbi dbi, MDB_val *key, MDB_val *data)
  333. sq_pushinteger(v, mdb_del(self, dbi, del));
  334. return 1;
  335. }
  336. static SQRESULT sq_LmDB_Transaction_get(HSQUIRRELVM v){
  337. SQ_FUNC_VARS_NO_TOP(v);
  338. GET_LmDB_Transaction_INSTANCE();
  339. SQ_GET_STRING(v, 2, key);
  340. // int mdb_get(MDB_txn *txn, MDB_dbi dbi, MDB_val *key, MDB_val *data)
  341. return 0;
  342. }
  343. static SQRESULT sq_LmDB_Transaction_int_commit(HSQUIRRELVM v){
  344. SQ_FUNC_VARS_NO_TOP(v);
  345. GET_LmDB_Transaction_INSTANCE();
  346. // int mdb_txn_commit(MDB_txn *txn)
  347. sq_pushinteger(v, mdb_txn_commit(self));
  348. return 1;
  349. }
  350. static SQRESULT sq_LmDB_Transaction_put(HSQUIRRELVM v){
  351. SQ_FUNC_VARS_NO_TOP(v);
  352. GET_LmDB_Transaction_INSTANCE();
  353. MDB_val key, value;
  354. pop_val(v, 2, &key);
  355. pop_val(v, 3, &value);
  356. SQ_GET_INTEGER(v, 4, int_flags);
  357. sq_pushinteger(v, mdb_put(self, &key, &value, int_flags));
  358. return 1;
  359. SQ_GET_STRING(v, 2, key);
  360. SQ_GET_STRING(v, 3, value);
  361. SQ_GET_INTEGER(v, 4, int_flags);
  362. // int mdb_put(MDB_txn *txn, MDB_dbi dbi, MDB_val *key, MDB_val *data, unsigned int flags)
  363. return 0;
  364. }
  365. static SQRESULT sq_LmDB_Transaction_renew(HSQUIRRELVM v){
  366. SQ_FUNC_VARS_NO_TOP(v);
  367. GET_LmDB_Transaction_INSTANCE();
  368. // int mdb_txn_renew(MDB_txn *txn)
  369. sq_pushinteger(v, mdb_txn_renew(self));
  370. return 1;
  371. }
  372. static SQRESULT sq_LmDB_Transaction_reset(HSQUIRRELVM v){
  373. SQ_FUNC_VARS_NO_TOP(v);
  374. GET_LmDB_Transaction_INSTANCE();
  375. // void mdb_txn_reset(MDB_txn *txn)
  376. sq_pushinteger(v, mdb_txn_reset(self));
  377. return 1;
  378. }
  379. static SQRESULT sq_LmDB_Transaction_stat(HSQUIRRELVM v){
  380. SQ_FUNC_VARS_NO_TOP(v);
  381. GET_LmDB_Transaction_INSTANCE();
  382. // int mdb_stat(MDB_txn *txn, MDB_dbi dbi, MDB_stat *stat)
  383. return 0;
  384. }
  385. #define _DECL_LMDB_TRANSACTION_FUNC(name,nparams,pmask) {_SC(#name),sq_LmDB_Transaction_##name,nparams,pmask}
  386. static SQRegFunction LmDB_Transaction_obj_funcs[]={
  387. _DECL_LMDB_TRANSACTION_FUNC(abort, 1, _SC("x")),
  388. _DECL_LMDB_TRANSACTION_FUNC(cmp, 1, _SC("x")),
  389. _DECL_LMDB_TRANSACTION_FUNC(cursor_open, 1, _SC("x")),
  390. _DECL_LMDB_TRANSACTION_FUNC(cursor_renew, 1, _SC("x")),
  391. _DECL_LMDB_TRANSACTION_FUNC(dbi_drop, 1, _SC("x")),
  392. _DECL_LMDB_TRANSACTION_FUNC(dbi_open, 3, _SC("x.i")),
  393. _DECL_LMDB_TRANSACTION_FUNC(dcmp, 1, _SC("x")),
  394. _DECL_LMDB_TRANSACTION_FUNC(del, 2, _SC("x.")),
  395. _DECL_LMDB_TRANSACTION_FUNC(get, 2, _SC("x.")),
  396. _DECL_LMDB_TRANSACTION_FUNC(int_commit, 1, _SC("x")),
  397. _DECL_LMDB_TRANSACTION_FUNC(put, 4, _SC("x..i")),
  398. _DECL_LMDB_TRANSACTION_FUNC(renew, 1, _SC("x")),
  399. _DECL_LMDB_TRANSACTION_FUNC(reset, 1, _SC("x")),
  400. _DECL_LMDB_TRANSACTION_FUNC(stat, 1, _SC("x")),
  401. {0,0}
  402. };
  403. #undef _DECL_LMDB_TRANSACTION_FUNC
  404. static const SQChar LMDB_CURSOR[] = _SC("LmDB_Cursor");
  405. static const SQChar LMDB_CURSOR_Tag[] = _SC("sq_LmDB_Cursor_ctx");
  406. #define GET_LmDB_Cursor_INSTANCE() SQ_GET_INSTANCE(v, 1, LMDB_CURSOR, LMDB_CURSOR_Tag) \
  407. if(self == NULL) return sq_throwerror(v, _SC("LmDB_Cursor object already closed"));
  408. static SQRESULT LmDB_Cursor_release_hook(SQUserPointer p, SQInteger size, void*)
  409. {
  410. MDB_cursor *self = (MDB_cursor*)p;
  411. if(self) mdb_cursor_close(self);
  412. return 0;
  413. }
  414. static SQRESULT LmDB_Cursor_constructor(HSQUIRRELVM v, MDB_cursor *lmdb_Cursor, int free_on_gc)
  415. {
  416. if(!lmdb_Cursor)
  417. return sq_throwerror(v, _SC("Could'nt create an LmDB_Cursor object."));
  418. sq_pushstring(v, LMDB_CURSOR, -1);
  419. if(sq_getonroottable(v) == SQ_OK){
  420. sq_pushstring(v, LMDB_CURSOR_Tag, -1);
  421. if(sq_get(v, -2) == SQ_OK){
  422. if(sq_createinstance(v, -1) == SQ_OK){
  423. sq_setinstanceup(v, -1, lmdb_Cursor);
  424. if(free_on_gc) sq_setreleasehook(v,-1, LmDB_Cursor_release_hook);
  425. return 1;
  426. }
  427. }
  428. }
  429. return SQ_ERROR;
  430. }
  431. static SQRESULT sq_LmDB_Cursor_constructor(HSQUIRRELVM v){
  432. // if(!load_library()) return sq_throwerror(v, _SC("Failed to load libcurl !"));
  433. //int mdb_cursor_open(MDB_txn *txn, MDB_dbi dbi, MDB_cursor **cursor);
  434. MDB_cursor *lmdb_Cursor = NULL;
  435. int rc = mdb_cursor_open(txn, dbi, &lmdb_Cursor);
  436. return LmDB_Cursor_constructor(v, 1, lmdb_Cursor, 1);
  437. }
  438. static SQRESULT sq_LmDB_Cursor_close(HSQUIRRELVM v){
  439. SQ_FUNC_VARS_NO_TOP(v);
  440. GET_ssl_INSTANCE();
  441. LmDB_Cursor_release_hook(self, 0, v);
  442. sq_setinstanceup(v, 1, 0);
  443. return 0;
  444. }
  445. static SQRESULT sq_LmDB_Cursor_count(HSQUIRRELVM v){
  446. SQ_FUNC_VARS_NO_TOP(v);
  447. GET_LmDB_Cursor_INSTANCE();
  448. mdb_size_t count;
  449. mdb_cursor_count(self, &count);
  450. sq_pushinteger(v, (SQInteger)count);
  451. return 1;
  452. }
  453. static SQRESULT sq_LmDB_Cursor_del(HSQUIRRELVM v){
  454. SQ_FUNC_VARS_NO_TOP(v);
  455. GET_LmDB_Cursor_INSTANCE();
  456. MDB_val key, value;
  457. pop_val(v, 2, &key);
  458. pop_val(v, 3, &value);
  459. SQ_GET_INTEGER(v, 4, int_flags);
  460. sq_pushinteger(v, mdb_cursor_del(self, &key, &value, int_flags));
  461. return 1;
  462. }
  463. static SQRESULT sq_LmDB_Cursor_get(HSQUIRRELVM v){
  464. SQ_FUNC_VARS_NO_TOP(v);
  465. GET_LmDB_Cursor_INSTANCE();
  466. MDB_val key, value;
  467. pop_val(v, 2, &key);
  468. pop_val(v, 3, &value);
  469. SQ_GET_INTEGER(v, 4, op);
  470. sq_pushinteger(v, mdb_cursor_get(self, &key, &value, op));
  471. return 1;
  472. }
  473. static SQRESULT sq_LmDB_Cursor_put(HSQUIRRELVM v){
  474. SQ_FUNC_VARS_NO_TOP(v);
  475. GET_LmDB_Cursor_INSTANCE();
  476. MDB_val key, value;
  477. pop_val(v, 2, &key);
  478. pop_val(v, 3, &value);
  479. SQ_GET_INTEGER(v, 4, int_flags);
  480. sq_pushinteger(v, mdb_cursor_put(self, &key, &value, int_flags));
  481. return 1;
  482. }
  483. #define _DECL_LMDB_CURSOR_FUNC(name,nparams,pmask) {_SC(#name),sq_LmDB_Cursor_##name,nparams,pmask}
  484. static SQRegFunction LmDB_Cursor_obj_funcs[]={
  485. _DECL_LMDB_CURSOR_FUNC(close, 1, _SC(".")),
  486. _DECL_LMDB_CURSOR_FUNC(count, 1, _SC(".")),
  487. _DECL_LMDB_CURSOR_FUNC(del, 2, _SC(".i")),
  488. _DECL_LMDB_CURSOR_FUNC(get, 2, _SC("..")),
  489. _DECL_LMDB_CURSOR_FUNC(put, 4, _SC("...i")),
  490. {0,0}
  491. };
  492. #undef _DECL_LMDB_CURSOR_FUNC
  493. typedef struct {
  494. const SQChar *Str;
  495. SQInteger Val;
  496. } KeyIntType, * KeyIntPtrType;
  497. static KeyIntType LmDB_constants[] = {
  498. #define MK_CONST(c) {_SC(#c), MDB_##c}
  499. MK_CONST(TLS_FULL);
  500. MK_CONST(NEXT);
  501. MK_CONST(WRITEMAP);
  502. MK_CONST(PAGE_FULL);
  503. MK_CONST(MULTIPLE);
  504. MK_CONST(PREV_DUP);
  505. MK_CONST(DBS_FULL);
  506. MK_CONST(NOSYNC);
  507. MK_CONST(SUCCESS);
  508. MK_CONST(SET);
  509. MK_CONST(SET_RANGE);
  510. MK_CONST(NOSUBDIR);
  511. MK_CONST(SET_KEY);
  512. MK_CONST(INCOMPATIBLE);
  513. MK_CONST(FIXEDMAP);
  514. MK_CONST(PREV_NODUP);
  515. MK_CONST(NEXT_MULTIPLE);
  516. MK_CONST(FIRST_DUP);
  517. MK_CONST(APPEND);
  518. MK_CONST(GET_MULTIPLE);
  519. MK_CONST(READERS_FULL);
  520. MK_CONST(CURSOR_FULL);
  521. MK_CONST(INTEGERKEY);
  522. MK_CONST(PANIC);
  523. MK_CONST(NEXT_DUP);
  524. MK_CONST(REVERSEKEY);
  525. MK_CONST(MAP_RESIZED);
  526. MK_CONST(LAST_DUP);
  527. MK_CONST(NEXT_NODUP);
  528. MK_CONST(NOMETASYNC);
  529. MK_CONST(GET_CURRENT);
  530. MK_CONST(GET_BOTH_RANGE);
  531. MK_CONST(MAP_FULL);
  532. MK_CONST(GET_BOTH);
  533. MK_CONST(PREV);
  534. MK_CONST(NOTLS);
  535. MK_CONST(FIRST);
  536. MK_CONST(CORRUPTED);
  537. MK_CONST(MAPASYNC);
  538. MK_CONST(DUPFIXED);
  539. MK_CONST(NORDAHEAD);
  540. MK_CONST(APPENDDUP);
  541. MK_CONST(NOOVERWRITE);
  542. MK_CONST(INVALID);
  543. MK_CONST(NOLOCK);
  544. MK_CONST(KEYEXIST);
  545. MK_CONST(NOTFOUND);
  546. MK_CONST(REVERSEDUP);
  547. MK_CONST(CREATE);
  548. MK_CONST(RESERVE);
  549. MK_CONST(TXN_FULL);
  550. MK_CONST(LAST);
  551. MK_CONST(PAGE_NOTFOUND);
  552. MK_CONST(DUPSORT);
  553. MK_CONST(RDONLY);
  554. MK_CONST(VERSION_MISMATCH);
  555. MK_CONST(CURRENT);
  556. MK_CONST(NOMEMINIT);
  557. MK_CONST(NODUPDATA);
  558. MK_CONST(INTEGERDUP);
  559. {0,0}
  560. };
  561. /* This defines a function that opens up your library. */
  562. SQRESULT sqext_register_LmDB (HSQUIRRELVM v) {
  563. //add a namespace LmDB
  564. sq_pushstring(v, SQ_LMDB_ENV, -1);
  565. sq_newtable(v);
  566. sq_insert_reg_funcs(v, LmDB_obj_funcs);
  567. //add constants
  568. KeyIntPtrType KeyIntPtr;
  569. for (KeyIntPtr = LmDB_constants; KeyIntPtr->Str; KeyIntPtr++) {
  570. sq_pushstring(v, KeyIntPtr->Str, -1); //first the key
  571. sq_pushinteger(v, KeyIntPtr->Val); //then the value
  572. sq_newslot(v, -3, SQFalse); //store then
  573. }
  574. sq_newslot(v,-3,SQFalse); //add LmDB table to the root table
  575. return SQ_OK;
  576. }
  577. #ifdef __cplusplus
  578. }
  579. #endif
  580. #endif //USE_LMDB