// #include // #include // #include // #include "kvstore.h" // rbtree_node *rbtree_mini(rbtree *T, rbtree_node *x) { // while (x->left != T->nil) { // x = x->left; // } // return x; // } // rbtree_node *rbtree_maxi(rbtree *T, rbtree_node *x) { // while (x->right != T->nil) { // x = x->right; // } // return x; // } // rbtree_node *rbtree_successor(rbtree *T, rbtree_node *x) { // rbtree_node *y = x->parent; // if (x->right != T->nil) { // return rbtree_mini(T, x->right); // } // while ((y != T->nil) && (x == y->right)) { // x = y; // y = y->parent; // } // return y; // } // void rbtree_left_rotate(rbtree *T, rbtree_node *x) { // rbtree_node *y = x->right; // x --> y , y --> x, right --> left, left --> right // x->right = y->left; //1 1 // if (y->left != T->nil) { //1 2 // y->left->parent = x; // } // y->parent = x->parent; //1 3 // if (x->parent == T->nil) { //1 4 // T->root = y; // } else if (x == x->parent->left) { // x->parent->left = y; // } else { // x->parent->right = y; // } // y->left = x; //1 5 // x->parent = y; //1 6 // } // void rbtree_right_rotate(rbtree *T, rbtree_node *y) { // rbtree_node *x = y->left; // y->left = x->right; // if (x->right != T->nil) { // x->right->parent = y; // } // x->parent = y->parent; // if (y->parent == T->nil) { // T->root = x; // } else if (y == y->parent->right) { // y->parent->right = x; // } else { // y->parent->left = x; // } // x->right = y; // y->parent = x; // } // void rbtree_insert_fixup(rbtree *T, rbtree_node *z) { // while (z->parent->color == RED) { //z ---> RED // if (z->parent == z->parent->parent->left) { // rbtree_node *y = z->parent->parent->right; // if (y->color == RED) { // z->parent->color = BLACK; // y->color = BLACK; // z->parent->parent->color = RED; // z = z->parent->parent; //z --> RED // } else { // if (z == z->parent->right) { // z = z->parent; // rbtree_left_rotate(T, z); // } // z->parent->color = BLACK; // z->parent->parent->color = RED; // rbtree_right_rotate(T, z->parent->parent); // } // }else { // rbtree_node *y = z->parent->parent->left; // if (y->color == RED) { // z->parent->color = BLACK; // y->color = BLACK; // z->parent->parent->color = RED; // z = z->parent->parent; //z --> RED // } else { // if (z == z->parent->left) { // z = z->parent; // rbtree_right_rotate(T, z); // } // z->parent->color = BLACK; // z->parent->parent->color = RED; // rbtree_left_rotate(T, z->parent->parent); // } // } // } // T->root->color = BLACK; // } // void rbtree_insert(rbtree *T, rbtree_node *z) { // rbtree_node *y = T->nil; // rbtree_node *x = T->root; // while (x != T->nil) { // y = x; // #if ENABLE_KEY_CHAR // if (strcmp(z->key, x->key) < 0) { // x = x->left; // } else if (strcmp(z->key, x->key) > 0) { // x = x->right; // } else { // return ; // } // #else // if (z->key < x->key) { // x = x->left; // } else if (z->key > x->key) { // x = x->right; // } else { //Exist // return ; // } // #endif // } // z->parent = y; // if (y == T->nil) { // T->root = z; // #if ENABLE_KEY_CHAR // } else if (strcmp(z->key, y->key) < 0) { // #else // } else if (z->key < y->key) { // #endif // y->left = z; // } else { // y->right = z; // } // z->left = T->nil; // z->right = T->nil; // z->color = RED; // rbtree_insert_fixup(T, z); // } // void rbtree_delete_fixup(rbtree *T, rbtree_node *x) { // while ((x != T->root) && (x->color == BLACK)) { // if (x == x->parent->left) { // rbtree_node *w= x->parent->right; // if (w->color == RED) { // w->color = BLACK; // x->parent->color = RED; // rbtree_left_rotate(T, x->parent); // w = x->parent->right; // } // if ((w->left->color == BLACK) && (w->right->color == BLACK)) { // w->color = RED; // x = x->parent; // } else { // if (w->right->color == BLACK) { // w->left->color = BLACK; // w->color = RED; // rbtree_right_rotate(T, w); // w = x->parent->right; // } // w->color = x->parent->color; // x->parent->color = BLACK; // w->right->color = BLACK; // rbtree_left_rotate(T, x->parent); // x = T->root; // } // } else { // rbtree_node *w = x->parent->left; // if (w->color == RED) { // w->color = BLACK; // x->parent->color = RED; // rbtree_right_rotate(T, x->parent); // w = x->parent->left; // } // if ((w->left->color == BLACK) && (w->right->color == BLACK)) { // w->color = RED; // x = x->parent; // } else { // if (w->left->color == BLACK) { // w->right->color = BLACK; // w->color = RED; // rbtree_left_rotate(T, w); // w = x->parent->left; // } // w->color = x->parent->color; // x->parent->color = BLACK; // w->left->color = BLACK; // rbtree_right_rotate(T, x->parent); // x = T->root; // } // } // } // x->color = BLACK; // } // rbtree_node *rbtree_delete(rbtree *T, rbtree_node *z) { // rbtree_node *y = T->nil; // rbtree_node *x = T->nil; // if ((z->left == T->nil) || (z->right == T->nil)) { // y = z; // } else { // y = rbtree_successor(T, z); // } // if (y->left != T->nil) { // x = y->left; // } else if (y->right != T->nil) { // x = y->right; // } // x->parent = y->parent; // if (y->parent == T->nil) { // T->root = x; // } else if (y == y->parent->left) { // y->parent->left = x; // } else { // y->parent->right = x; // } // if (y != z) { // #if ENABLE_KEY_CHAR // void *tmp = z->key; // z->key = y->key; // y->key = tmp; // tmp = z->value; // z->value= y->value; // y->value = tmp; // #else // z->key = y->key; // z->value = y->value; // #endif // } // if (y->color == BLACK) { // rbtree_delete_fixup(T, x); // } // return y; // } // rbtree_node *rbtree_search(rbtree *T, KEY_TYPE key) { // rbtree_node *node = T->root; // while (node != T->nil) { // #if ENABLE_KEY_CHAR // if (strcmp(key, node->key) < 0) { // node = node->left; // } else if (strcmp(key, node->key) > 0) { // node = node->right; // } else { // return node; // } // #else // if (key < node->key) { // node = node->left; // } else if (key > node->key) { // node = node->right; // } else { // return node; // } // #endif // } // return T->nil; // } // void rbtree_traversal(rbtree *T, rbtree_node *node) { // if (node != T->nil) { // rbtree_traversal(T, node->left); // #if ENABLE_KEY_CHAR // printf("key:%s, value:%s\n", node->key, (char *)node->value); // #else // printf("key:%d, color:%d\n", node->key, node->color); // #endif // rbtree_traversal(T, node->right); // } // } // #if 0 // int main() { // #if ENABLE_KEY_CHAR // char* keyArray[10] = {"King", "Darren", "Mark", "Vico", "Nick", "qiuxiang", "youzi", "taozi", "123", "234"}; // char* valueArray[10] = {"1King", "2Darren", "3Mark", "4Vico", "5Nick", "6qiuxiang", "7youzi", "8taozi", "9123", "10234"}; // rbtree *T = (rbtree *)malloc(sizeof(rbtree)); // if (T == NULL) { // printf("malloc failed\n"); // return -1; // } // T->nil = (rbtree_node*)malloc(sizeof(rbtree_node)); // T->nil->color = BLACK; // T->root = T->nil; // rbtree_node *node = T->nil; // int i = 0; // for (i = 0;i < 10;i ++) { // node = (rbtree_node*)malloc(sizeof(rbtree_node)); // node->key = malloc(strlen(keyArray[i]) + 1); // memset(node->key, 0, strlen(keyArray[i]) + 1); // strcpy(node->key, keyArray[i]); // node->value = malloc(strlen(valueArray[i]) + 1); // memset(node->value, 0, strlen(valueArray[i]) + 1); // strcpy(node->value, valueArray[i]); // rbtree_insert(T, node); // } // rbtree_traversal(T, T->root); // printf("----------------------------------------\n"); // for (i = 0;i < 10;i ++) { // rbtree_node *node = rbtree_search(T, keyArray[i]); // rbtree_node *cur = rbtree_delete(T, node); // free(cur); // rbtree_traversal(T, T->root); // printf("----------------------------------------\n"); // } // #else // int keyArray[20] = {24,25,13,35,23, 26,67,47,38,98, 20,19,17,49,12, 21,9,18,14,15}; // rbtree *T = (rbtree *)malloc(sizeof(rbtree)); // if (T == NULL) { // printf("malloc failed\n"); // return -1; // } // T->nil = (rbtree_node*)malloc(sizeof(rbtree_node)); // T->nil->color = BLACK; // T->root = T->nil; // rbtree_node *node = T->nil; // int i = 0; // for (i = 0;i < 20;i ++) { // node = (rbtree_node*)malloc(sizeof(rbtree_node)); // node->key = keyArray[i]; // node->value = NULL; // rbtree_insert(T, node); // } // rbtree_traversal(T, T->root); // printf("----------------------------------------\n"); // for (i = 0;i < 20;i ++) { // rbtree_node *node = rbtree_search(T, keyArray[i]); // rbtree_node *cur = rbtree_delete(T, node); // free(cur); // rbtree_traversal(T, T->root); // printf("----------------------------------------\n"); // } // #endif // } // #endif // typedef struct _rbtree kvs_rbtree_t; // kvs_rbtree_t global_rbtree; // // 5 + 2 // int kvs_rbtree_create(kvs_rbtree_t *inst) { // if (inst == NULL) return 1; // inst->nil = (rbtree_node*)kvs_malloc(sizeof(rbtree_node)); // inst->nil->color = BLACK; // inst->root = inst->nil; // return 0; // } // void kvs_rbtree_destroy(kvs_rbtree_t *inst) { // if (inst == NULL) return ; // rbtree_node *node = NULL; // while (!(node = inst->root)) { // rbtree_node *mini = rbtree_mini(inst, node); // rbtree_node *cur = rbtree_delete(inst, mini); // kvs_free(cur); // } // kvs_free(inst->nil); // return ; // } // int kvs_rbtree_set(kvs_rbtree_t *inst, char *key, char *value) { // if (!inst || !key || !value) return -1; // rbtree_node *node = (rbtree_node*)kvs_malloc(sizeof(rbtree_node)); // node->key = kvs_malloc(strlen(key) + 1); // if (!node->key) return -2; // memset(node->key, 0, strlen(key) + 1); // strcpy(node->key, key); // node->value = kvs_malloc(strlen(value) + 1); // if (!node->value) return -2; // memset(node->value, 0, strlen(value) + 1); // strcpy(node->value, value); // rbtree_insert(inst, node); // return 0; // } // char* kvs_rbtree_get(kvs_rbtree_t *inst, char *key) { // if (!inst || !key) return NULL; // rbtree_node *node = rbtree_search(inst, key); // if (!node) return NULL; // no exist // if (node == inst->nil) return NULL; // return node->value; // } // int kvs_rbtree_del(kvs_rbtree_t *inst, char *key) { // if (!inst || !key) return -1; // rbtree_node *node = rbtree_search(inst, key); // if (!node) return 1; // no exist // rbtree_node *cur = rbtree_delete(inst, node); // free(cur); // return 0; // } // int kvs_rbtree_mod(kvs_rbtree_t *inst, char *key, char *value) { // if (!inst || !key || !value) return -1; // rbtree_node *node = rbtree_search(inst, key); // if (!node) return 1; // no exist // if (node == inst->nil) return 1; // kvs_free(node->value); // node->value = kvs_malloc(strlen(value) + 1); // if (!node->value) return -2; // memset(node->value, 0, strlen(value) + 1); // strcpy(node->value, value); // return 0; // } // int kvs_rbtree_exist(kvs_rbtree_t *inst, char *key) { // if (!inst || !key) return -1; // rbtree_node *node = rbtree_search(inst, key); // if (!node) return 1; // no exist // if (node == inst->nil) return 1; // return 0; // }