bdb_tool.cpp

Go to the documentation of this file.

 
#include "bdb_tool.h"
#include "jbxl_state.h"
 
 
#ifdef  ENABLE_BERKELEYDB
 
DB*  open_fdb(char* fname, unsigned int mode)
{
    DB* dp;
 
    if (fname==NULL) return NULL;
 
    int ret = db_create(&dp, NULL, 0);
    if (ret==0) {
        ret = dp->open(dp, NULL, fname, NULL, DB_HASH, (u_int32_t)mode, 0600);
    }
 
    if (ret!=0) {
        DEBUG_MODE {
            dp->err(dp, ret, "DB->open");
            PRINT_MESG("dbname is %s\n", fname);
        }
        return NULL;
    }
    return dp;
}
 
 
void  close_fdb(DB** dp)
{
    if (dp==NULL || *dp==NULL) return;
 
    (*dp)->close((*dp), 0);
    *dp = NULL;
    return;
}
 
 
int  check_fdb_file(char* fname)
{
    DB* dp = open_fdb(fname, DB_CREATE);
 
    if (dp==NULL)  return FALSE;
    close_fdb(&dp);
    return TRUE;
}
 
 
int   set_fdb_data(DB* dp, Buffer id, Buffer dbdata)
{
    DBT  key, data;
    int  ret;
 
    memset(&key,  0, sizeof(DBT));
    memset(&data, 0, sizeof(DBT));
 
    key.data = id.buf;
    key.size = id.vldsz;
    if (key.size<=0) {
        key.size = strlen((char*)id.buf);
        if (key.size<=0) return JBXL_BDB_DATASZ_ERROR;
    }
 
    data.data = dbdata.buf;
    data.size = dbdata.vldsz;
    if (data.size<=0) return JBXL_BDB_DATASZ_ERROR;
 
    ret = dp->put(dp, NULL, &key, &data, 0);
    if (ret!=0) {
        DEBUG_MODE dp->err(dp, ret, "DB->put");
        return JBXL_BDB_PUT_ERROR;
    }
    return data.size;
}
 
 
int   set_fdb_data_file(char* fname, Buffer id, Buffer dbdata)
{
    DBT  key, data;
    DB*  dp;
    int  ret;
 
    dp = open_fdb(fname, DB_CREATE);
    if (dp==NULL) return JBXL_BDB_OPEN_ERROR;
 
    memset(&key,  0, sizeof(DBT));
    memset(&data, 0, sizeof(DBT));
 
    key.data  = (char*)(id.buf);
    key.size = id.vldsz;
    if (key.size<=0) {
        key.size = strlen((char*)id.buf);
        if (key.size<=0) return JBXL_BDB_KEY_ERROR;
    }
 
    data.data = (char*)dbdata.buf;
    data.size = dbdata.vldsz;
    if (data.size<=0) return JBXL_BDB_DATASZ_ERROR;
 
    ret = dp->put(dp, NULL, &key, &data, 0);
    close_fdb(&dp);
 
    if (ret!=0) {
        DEBUG_MODE dp->err(dp, ret, "DB->put");
        return JBXL_BDB_PUT_ERROR;
    }
    return data.size;
}
 
 
Buffer  get_fdb_data(DB* dp, Buffer id)
{
    int  ret;
    DBT  key, data;
    Buffer dbdata;
 
    memset(&key,  0, sizeof(DBT));
    memset(&data, 0, sizeof(DBT));
 
    dbdata   = init_Buffer();
    key.data = (void*)id.buf;
    key.size = (u_int32_t)id.vldsz;
    if (key.size<=0) {
        key.size = strlen((char*)id.buf);
        if (key.size<=0) {
            dbdata.state = JBXL_BDB_MEMOEY_ERROR;
            return dbdata;
        }
    }
 
    ret = dp->get(dp, NULL, &key, &data, 0);
    if (ret!=0) {
        dbdata.state = JBXL_DBD_MEMORY_ERROR;
        return dbdata;
    }
 
    dbdata = make_Buffer(data.size+1);        // +1 is for string
    if (dbdata.buf==NULL) {
        dbdata.state = JBXL_BDB_GET_ERROR;
        return dbdata;
    }
    memcpy(dbdata.buf, data.data, data.size);
    dbdata.vldsz = data.size;
    dbdata.buf[data.size] = '\0';            // for string
 
    return dbdata;
}
 
 
Buffer  get_fdb_data_file(char* fname, Buffer id)
{
    int  ret;
    DBT  key, data;
    DB*  dp;
    Buffer dbdata;
 
    dbdata = init_Buffer();
    dp = open_fdb(fname, DB_RDONLY);
    if (dp==NULL) {
        dbdata.state = JBXL_BDB_OPEN_ERROR;
        return dbdata;
    }
 
    memset(&key,  0, sizeof(DBT));
    memset(&data, 0, sizeof(DBT));
 
    key.data = (char*)(id.buf);
    key.size = id.vldsz;
    if (key.size<=0) {
        key.size = strlen((char*)id.buf);
        if (key.size<=0) {
            dbdata.state = JBXL_BDB_KEY_ERROR;
            return dbdata;
        }
    }
 
    ret = dp->get(dp, NULL, &key, &data, 0);
    close_fdb(&dp);
    if (ret!=0) {
        dbdata.state = JBXL_BDB_GET_ERROR;
        return dbdata;
    }
 
    dbdata = make_Buffer(data.size+1);        // +1 is for string
    if (dbdata.buf==NULL) {
        dbdata.state = JBXL_BDB_MEMORY_ERROR;
        return dbdata;
    }
 
    dbdata.vldsz = data.size;
    memcpy(dbdata.buf, data.data, data.size);
    dbdata.buf[data.size] = '\0';            // for string
 
    return dbdata;
}
 
 
int   del_fdb_data(DB* dp, Buffer id)
{
    DBT  key;
    int  ret;
 
    memset(&key, 0, sizeof(DBT));
    key.data = (char*)(id.buf);
    key.size = id.vldsz;
    if (key.size<=0) {
        key.size = strlen((char*)id.buf);
        if (key.size<=0) return FALSE;
    }
    ret = dp->del(dp, NULL, &key, 0);
 
    if (ret!=0) {
        DEBUG_MODE dp->err(dp, ret, "DB->del");
        return FALSE;
    }
    return TRUE;
}
 
 
int   del_fdb_data_file(char* fname, Buffer id)
{
    DBT  key;
    DB*  dp;
    int  ret;
 
    dp = open_fdb(fname, 0);
    if (dp==NULL) return FALSE;
 
    memset(&key, 0, sizeof(DBT));
    key.data = (char*)(id.buf);
    key.size = id.vldsz;
    if (key.size<=0) {
        key.size = strlen((char*)id.buf);
        if (key.size<=0) return FALSE;
    }
 
    ret = dp->del(dp, NULL, &key, 0);
    close_fdb(&dp);
 
    if (ret!=0) {
        DEBUG_MODE dp->err(dp, ret, "DB->del");
        return FALSE;
    }
    return TRUE;
}
 
 
tList*  get_all_fdb_data(DB* dp)
{
    int  ret;
    DBT  key, data;
    DBC* crsp;
 
    tList* lt = NULL;
    tList* lp = NULL;
    Buffer id, dt;
 
    if (dp==NULL) return NULL;
 
    dp->cursor(dp, NULL, &crsp, 0);
    if (crsp==NULL) return NULL;
    
    memset(&key,  0, sizeof(DBT));
    memset(&data, 0, sizeof(DBT));
 
    ret = crsp->c_get(crsp, &key, &data, DB_NEXT);
    while (ret==0) {
        id = make_Buffer_bin(key.data,  key.size);
        dt = make_Buffer_bin(data.data, data.size);
        lt = add_tList_node_Buffer(lt, id, dt);
        if (lp==NULL) lp = lt;
        free_Buffer(&id);
        free_Buffer(&dt);
 
        ret = crsp->c_get(crsp, &key, &data, DB_NEXT);
    }
 
    crsp->c_close(crsp);
 
    return lp;
}
 
 
tList*  get_all_fdb_data_file(char* fname)
{
    DB*    dp;
    tList* lp = NULL;
 
    if (fname==NULL) return NULL;
 
    dp = open_fdb(fname, DB_RDONLY);
    if (dp==NULL) return NULL;
 
    lp = get_all_fdb_data(dp);
    close_fdb(&dp);
 
    return lp;
}
 
 
#endif