static PyObject *
put(PyObject *self,PyObject *args){
const char *dbname;
const char *stext;
const int *kid;
int ecode;
bool result;
TCIDB *idb;
if (!PyArg_ParseTuple(args, "sis", &dbname, &kid, &stext))
return NULL;
/* create the object */
idb = tcidbnew();
/* open the database */
if(!tcidbopen(idb, dbname, IDBOCREAT | IDBOWRITER)){
ecode = tcidbecode(idb);
fprintf(stderr, "open error: %s\n", tcidberrmsg(ecode));
}
result = tcidbput(idb,(int64_t)kid,stext);
/* close the database */
if(!tcidbclose(idb)){
ecode = tcidbecode(idb);
fprintf(stderr, "close error: %s\n", tcidberrmsg(ecode));
}
/* delete the object */
tcidbdel(idb);
return Py_BuildValue("b",result);
}
static PyObject *
out(PyObject *self, PyObject *args){
const char *data_dir;
const int *key;
TCIDB *idb;
bool result;
int ecode;
if (PyArg_ParseTuple(args, "si", &data_dir, &key)) {
/* create the object */
idb = tcidbnew();
/* open the database */
if(!tcidbopen(idb, data_dir, IDBOWRITER | IDBOCREAT)){
ecode = tcidbecode(idb);
fprintf(stderr, "open error: %s\n", tcidberrmsg(ecode));
return NULL;
}
/* retrieve record */
result = tcidbout(idb, (int64_t) key);
/* close the database */
if(!tcidbclose(idb)){
ecode = tcidbecode(idb);
fprintf(stderr, "close error: %s\n", tcidberrmsg(ecode));
return NULL;
}
/* delete the object */
tcidbdel(idb);
return Py_BuildValue("b", result);
}
}
static PyObject *
search(PyObject *self, PyObject *args){
const char *stext;
const char *dbname;
TCIDB *idb;
int ecode, rnum, i;
uint64_t *result;
char *text;
PyObject* pList;
if (!PyArg_ParseTuple(args, "ss", &dbname, &stext))
return NULL;
/* create the object */
idb = tcidbnew();
/* open the database */
if(!tcidbopen(idb, dbname, IDBOREADER | IDBONOLCK)){
ecode = tcidbecode(idb);
fprintf(stderr, "open error: %s\n", tcidberrmsg(ecode));
}
/* search records */
result = tcidbsearch2(idb, stext, &rnum);
pList = PyList_New(rnum);
if(result){
for(i = 0; i < rnum; i++){
// printf("r[i]:%lld\n",result[i]);
PyList_SetItem(pList, i, Py_BuildValue("i", (int)result[i]));
}
tcfree(result);
} else {
ecode = tcidbecode(idb);
fprintf(stderr, "search error: %s\n", tcidberrmsg(ecode));
}
/* close the database */
if(!tcidbclose(idb)){
ecode = tcidbecode(idb);
fprintf(stderr, "close error: %s\n", tcidberrmsg(ecode));
}
/* delete the object */
tcidbdel(idb);
return Py_BuildValue("O",pList);
}
static PyObject *
get(PyObject *self, PyObject *args){
const char *data_dir;
const int *key;
TCIDB *idb;
char *val;
int ecode;
if (PyArg_ParseTuple(args, "si", &data_dir, &key)) {
/* create the object */
idb = tcidbnew();
/* open the database */
if(!tcidbopen(idb, data_dir, IDBOWRITER | IDBOCREAT)){
ecode = tcidbecode(idb);
fprintf(stderr, "open error: %s\n", tcidberrmsg(ecode));
return NULL;
}
/* retrieve record */
val = tcidbget(idb, (int64_t) key);
/* close the database */
if(!tcidbclose(idb)){
ecode = tcidbecode(idb);
fprintf(stderr, "close error: %s\n", tcidberrmsg(ecode));
return NULL;
}
/* delete the object */
tcidbdel(idb);
if (val) {
return Py_BuildValue("s", val);
} else {
PyErr_Format(PyExc_KeyError, "%i", key);
}
}
}
