在V1.0版本,我们实现了基于array的kv存储引擎。本文继续完善,增加rbtree、hash、skiptable引擎。
实际上,在框架确定的基础上,其他的引擎只需要添加接口即可。
一、架构设计
二、具体实现
2.1 引擎层
//------------------------ array ------------------typedef struct kvs_array_item_s {char *key;char *value;
} kvs_array_item_t;// kvs_array_table 存储插入的 kv
kvs_array_item_t kvs_array_table [KVS_ARRAY_ITEM_SIEZ] = {0};//--------------------- rbtree hash skiptable--------------
void init_kvengine(void) {init_rbtree(&tree);init_hashtable(&hash);init_skiptable(&table);
}void dest_kvengine(void) {dest_rbtree(&tree);dest_hashtable(&hash);dest_skiptable(&table);
}
2.2 接口层
接口尽量做到一致,方便管理拓展。
//------------------------------ array --------------------------------// 查找 key 在 kvs_array_table 的位置
kvs_array_item_t *kvs_array_search_item (char *key);// KVS_CMD_EXIST: 判断 key 是否存在,存在返回 1
int kvs_array_exist (char *key);// KVS_CMD_SET:插入 kv
int kvs_array_set (char *key, char *value);// KVS_CMD_GET:获取 key 对应的value
char *kvs_array_get(char *key);// KVS_CMD_COUNT:统计以及插入多少个 key
int kvs_array_count (void) ;// KVS_CMD_DELETE:删除 key
int kvs_array_delete(char *key);//------------------------------ rbtree--------------------------------
// KVS_CMD_REXIST
int kvs_rbtree_exist (char *key) {return exist_kv_rbtree(&tree, key);
}// KVS_CMD_RSET
int kvs_rbtree_set (char *key, char *value) {return put_kv_rbtree(&tree, key, value);
}// KVS_CMD_RGET
char *kvs_rbtree_get (char *key) {return get_kv_rbtree(&tree, key);
}// KVS_CMD_RCOUNT
int kvs_rbtree_count(void) {return count_kv_rbtree(&tree);
}
// KVS_CMD_RDELETE
int kvs_rbtree_delete(char *key) {return delete_kv_rbtree(&tree, key);
}//------------------------------ hash--------------------------------int kvs_hash_set(char *key, char *value) {return put_kv_hashtable(&hash, key, value);
}char *kvs_hash_get(char *key) {return get_kv_hashtable(&hash, key);
}int kvs_hash_count(void) {return count_kv_hashtable(&hash);
}int kvs_hash_exist(char *key) {return exist_kv_hashtable(&hash, key);
}int kvs_hash_delete(char *key) {return delete_kv_hashtable(&hash, key);
}//------------------------------ skiptable--------------------------------int kvs_skiptable_set(char *key, char *value) {return put_kv_skiptable(&table, key, value);
}char *kvs_skiptable_get(char *key) {return get_kv_skiptable(&table, key);
}int kvs_skiptable_count(void) {return count_kv_skiptable(&table);
}int kvs_skiptable_exist(char *key) {return exist_kv_skiptable(&table, key);
}int kvs_skiptable_delete(char *key) {return delete_kv_skiptable(&table, key);
}
2.3 协议层
// 根据msg,解析其具体的命令协议
int kvs_parser_protocol (char *msg, char **tokens, int count) ;/*分割msg,比如 msg为 SET NAME ZXM ,分割为SET,NAME,ZXM,分别存储在tokens[]* tokens[0]: SET --------- 对应的是命令 cmd* tokens[1]: NAME --------- 对应的是命令 key* tokens[2]: ZXM --------- 对应的是命令 value*/
int kvs_spilt_tokens (char **tokens, char *msg) ;// 解析协议
int kvs_protocol (char *msg, int length);
2.4 网络层
纯c版本的协程实现NtyCo
// 为每一个连接端口创建一个协程
nty_coroutine_create(&co, server, port);
2.5 功能测试结果
2.6 qps测试结果
三、完成代码
Github地址:KV存储引擎项目