nmdb/be-tc.c

   1
   2 #if BE_ENABLE_TC
   3
   4 #include <tchdb.h>      /* Tokyo Cabinet's hash API */
   5 #include <stdlib.h>
   6
   7 #include "be.h"
   8
   9
  10 int tc_set(struct db_conn *db, const unsigned char *key, size_t ksize,
  11                 unsigned char *val, size_t vsize);
  12 int tc_get(struct db_conn *db, const unsigned char *key, size_t ksize,
  13                 unsigned char *val, size_t *vsize);
  14 int tc_del(struct db_conn *db, const unsigned char *key, size_t ksize);
  15 int tc_close(struct db_conn *db);
  16
  17
  18 struct db_conn *tc_open(const char *name, int flags)
  19 {
  20         struct db_conn *db;
  21         TCHDB *tc_db = tchdbnew();
  22
  23         if (!tchdbopen(qdbm_db, name, HDBOWRITER | HDBOCREAT))
  24                 return NULL;
  25
  26         db = malloc(sizeof(struct db_conn));
  27         if (db == NULL) {
  28                 tchdbclose(tc_db);
  29                 tchdbdel(tc_db);
  30                 return NULL;
  31         }
  32
  33         db->conn = tc_db;
  34         db->set = tc_set;
  35         db->get = tc_get;
  36         db->del = tc_del;
  37         db->close = tc_close;
  38
  39         return db;
  40 }
  41
  42
  43 int tc_close(struct db_conn *db)
  44 {
  45         int r = tchdbclose(db->conn);
  46         tchdbdel(db->conn);
  47         free(db);
  48         return r;
  49 }
  50
  51
  52 int tc_set(struct db_conn *db, const unsigned char *key, size_t ksize,
  53                 unsigned char *val, size_t vsize)
  54 {
  55         return tchdbput(db->conn, key, ksize, val, vsize);
  56 }
  57
  58
  59 int tc_get(struct db_conn *db, const unsigned char *key, size_t ksize,
  60                 unsigned char *val, size_t *vsize)
  61 {
  62         int rv;
  63
  64         rv = tchdbget3(db->conn, key, ksize, val, *vsize);
  65         if (rv >= 0) {
  66                 *vsize = rv;
  67                 return 1;
  68         } else {
  69                 return 0;
  70         }
  71 }
  72
  73 int tc_del(struct db_conn *db, const unsigned char *key, size_t ksize)
  74 {
  75         return tchdbout(db->conn, key, ksize);
  76 }
  77
  78 #else
  79
  80 #include <stddef.h>     /* NULL */
  81
  82 struct db_conn *tc_open(const char *name, int flags)
  83 {
  84         return NULL;
  85 }
  86
  87 #endif
  88