holdem/storage.c at master · wangchuan3533/holdem · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <db.h>

#define DATABASE "holdem.db"
DB *g_db;

int holdem_db_init()
{
    int ret;

    if ((ret = db_create(&g_db, NULL, 0)) != 0) {
        fprintf(stderr, "db_create: %s\n", db_strerror(ret));
        return -1;
    }
    if ((ret = g_db->open(g_db, NULL, DATABASE, NULL, DB_BTREE, DB_CREATE, 0664)) != 0) {
        g_db->err(g_db, ret, "%s", DATABASE);
        return -1;
    }
    return 0;
}

int holdem_db_close()
{
    int ret;

    if ((ret = g_db->close(g_db, 0)) != 0 && ret != 0) {
        return -1;
    }
    return 0;
}

int holdem_db_put(const void *key, size_t key_len, const void *value, size_t val_len)
{
    int ret;
    DBT k, v;

    memset(&k, 0, sizeof(k));
    memset(&v, 0, sizeof(v));
    k.data = (void *)key;
    k.size = key_len;
    v.data = (void *)value;
    v.size = val_len;

    if ((ret = g_db->put(g_db, NULL, &k, &v, 0)) != 0) {
        g_db->err(g_db, ret, "DB->put");
        return -1;
    }
    if ((ret = g_db->sync(g_db, 0)) != 0) {
        g_db->err(g_db, ret, "DB->sync");
        return -1;
    }
    return 0;
}

int holdem_db_get(const void *key, size_t key_len, void *value, size_t *val_len)
{
    int ret;
    DBT k, v;

    memset(&k, 0, sizeof(k));
    memset(&v, 0, sizeof(v));
    k.data = (void *)key;
    k.size = key_len;
    v.data = NULL;
    v.size = 0;

    if ((ret = g_db->get(g_db, NULL, &k, &v, 0)) != 0) {
        //g_db->err(g_db, ret, "DB->get");
        return -1;
    }
    memcpy(value, v.data, v.size);
    *val_len = v.size;
    return 0;
}