Marc/server
1
0
Fork 0
forked from github/server
Code Issues Pull requests Projects Releases Wiki Activity

implement get/put for db

Browse source
This commit is contained in:
Enno Rehling 2017-11-19 12:50:24 +01:00
parent 2f5f685e55
commit 2ff820d1b4
1 changed files with 40 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
src/kernel/db/berkeley.c
View file

@ -1,6 +1,9 @@
#include <platform.h> #include <platform.h>
#include "driver.h" #include "driver.h"
#include <kernel/config.h>
#include <kernel/orderdb.h>
#ifdef __APPLE__ #ifdef __APPLE__
#define _DARWIN_C_SOURCE #define _DARWIN_C_SOURCE
#include <sys/types.h> #include <sys/types.h>
@ -8,8 +11,10 @@
#include <db.h> #include <db.h>
#include <assert.h> #include <assert.h>
#include <string.h>
static DB *g_dbp; static DB *g_dbp;
static int order_id = -1;
void db_driver_open(void) void db_driver_open(void)
{ {
@ -17,11 +22,18 @@ void db_driver_open(void)
u_int32_t flags = DB_CREATE|DB_TRUNCATE; u_int32_t flags = DB_CREATE|DB_TRUNCATE;
const char * dbname; const char * dbname;
dbname = config_get("game.dbname");
if (!dbname) {
dbname = "temporary.db";
}
ret = db_create(&g_dbp, NULL, 0); ret = db_create(&g_dbp, NULL, 0);
assert(ret==0); assert(ret==0);
ret = g_dbp->open(g_dbp, NULL, dbname, NULL, DB_BTREE, flags, 0); ret = g_dbp->open(g_dbp, NULL, dbname, NULL, DB_BTREE, flags, 0);
assert(ret==0); assert(ret==0);
assert(order_id == -1);
order_id = 0;
} }
void db_driver_close(void) void db_driver_close(void)
@ -29,15 +41,41 @@ void db_driver_close(void)
int ret; int ret;
ret = g_dbp->close(g_dbp, 0); ret = g_dbp->close(g_dbp, 0);
assert(ret==0); assert(ret==0);
order_id = -1;
} }
int db_driver_order_save(struct order_data *od) int db_driver_order_save(struct order_data *od)
{ {
return 0; int ret;
DBT key, data;
assert(od && od->_str);
++order_id;
key.data = &order_id;
key.size = sizeof(int);
data.data = (void *)od->_str;
data.size = strlen(od->_str) + 1;
data.flags = DB_DBT_USERMEM;
ret = g_dbp->put(g_dbp, NULL, &key, &data, DB_NOOVERWRITE);
assert(ret == 0);
return order_id;
} }
struct order_data *db_driver_order_load(int id) struct order_data *db_driver_order_load(int id)
{ {
return NULL; int ret;
order_data *od = NULL;
DBT key, data;
assert(id>0 && id <= order_id);
memset(&key, 0, sizeof(DBT));
memset(&data, 0, sizeof(DBT));
key.data = &id;
key.size = sizeof(int);
ret = g_dbp->get(g_dbp, NULL, &key, &data, 0);
if (ret == 0) {
odata_create(&od, data.ulen, data.data);
}
return od;
} }