tkv.h 2.2 KB
Newer Older
H
refact  
Hongze Cheng 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
/*
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef _TD_TKV_H_
#define _TD_TKV_H_

#ifdef __cplusplus
extern "C" {
#endif

H
refact  
Hongze Cheng 已提交
23 24 25 26 27 28 29 30 31 32
typedef struct tkv_db_s tkv_db_t;

typedef struct {
  /* data */
} tkv_key_t;

typedef struct {
  bool    pinned;
  int64_t ref;  // TODO: use util library
                // TODO: add a RW latch here
H
refact  
Hongze Cheng 已提交
33 34
  uint64_t offset;
  void *   pObj;
H
refact  
Hongze Cheng 已提交
35 36 37 38 39 40
} tkv_obj_t;

typedef int (*tkv_key_comp_fn_t)(const tkv_key_t *, const tkv_key_t *);
typedef void (*tkv_get_key_fn_t)(const tkv_obj_t *, tkv_key_t *);
typedef int (*tkv_obj_encode_fn_t)(void **buf, void *pObj);
typedef void *(*tkv_obj_decode_fn_t)(void *buf, void **pObj);
H
refact  
Hongze Cheng 已提交
41
typedef int (*tkv_obj_comp_fn_t)(const tkv_obj_t *, const tkv_obj_t *);
H
refact  
Hongze Cheng 已提交
42 43

typedef struct {
H
refact  
Hongze Cheng 已提交
44
  uint64_t            memLimit;
H
refact  
Hongze Cheng 已提交
45 46 47
  tkv_get_key_fn_t    getKey;
  tkv_obj_encode_fn_t encode;
  tkv_obj_decode_fn_t decode;
H
refact  
Hongze Cheng 已提交
48
  tkv_obj_comp_fn_t   compare;
H
refact  
Hongze Cheng 已提交
49 50
} tkv_db_option_t;

H
refact  
Hongze Cheng 已提交
51 52 53
tkv_db_t *       tkvOpenDB(char *dir, tkv_db_option_t *);
int              tkvCloseDB(tkv_db_t *);
int              tkvPut(tkv_db_t *, tkv_obj_t *);
H
refact  
Hongze Cheng 已提交
54
int              tkvPutBatch(tkv_db_t *, tkv_obj_t **, int);  // TODO: use array here
H
refact  
Hongze Cheng 已提交
55
const tkv_obj_t *tkvGet(tkv_key_t *);
H
refact  
Hongze Cheng 已提交
56
int              tkvGetBatch(tkv_db_t *, tkv_key_t **, int, tkv_obj_t **);  // TODO: use array here
H
refact  
Hongze Cheng 已提交
57
int              tkvDrop(tkv_db_t *, tkv_key_t *);
H
refact  
Hongze Cheng 已提交
58
int              tkvDropBatch(tkv_db_t *, tkv_key_t **, int);  // TODO: use array here
H
refact  
Hongze Cheng 已提交
59
int              tkvCommit(tkv_db_t *, void * /*TODO*/);
H
refact  
Hongze Cheng 已提交
60 61 62 63 64 65 66

typedef struct {
} tkv_db_iter_t;

tkv_db_iter_t *  tkvIterNew(tkv_db_t *);
void             tkvIterFree(tkv_db_iter_t *);
const tkv_obj_t *tkvIterNext(tkv_db_iter_t *);
H
refact  
Hongze Cheng 已提交
67

H
refact  
Hongze Cheng 已提交
68 69 70 71 72
#ifdef __cplusplus
}
#endif

#endif /*_TD_TKV_H_*/