tokyo cabinet源代碼分析（6）

2.5數據記錄的刪除

主要流程爲二叉樹節點刪除。

二叉樹中刪除記錄的過程: /*記錄在二叉樹中的刪除過程 *通過HASH函數將key映射到TCMAP數組的一個元素， *然後通過tcmapout進行相應的刪除工作*/ /* Remove a record of an on-memory hash database. */ bool tcmdbout(TCMDB *mdb, const void *kbuf, int ksiz){ assert(mdb && kbuf && ksiz >= 0); unsigned int mi; TCMDBHASH(mi, kbuf, ksiz); if(pthread_rwlock_wrlock((pthread_rwlock_t *)mdb->mmtxs + mi) != 0) return false; bool rv = tcmapout(mdb->maps[mi], kbuf, ksiz); pthread_rwlock_unlock((pthread_rwlock_t *)mdb->mmtxs + mi); return rv; }

通過tcmapout進行刪除。

/* Remove a record of a map object. */ bool tcmapout(TCMAP *map, const void *kbuf, int ksiz){ assert(map && kbuf && ksiz >= 0); if(ksiz > TCMAPKMAXSIZ) ksiz = TCMAPKMAXSIZ; uint32_t hash; /*HASH映射到buckets數組的一個元素*/ TCMAPHASH1(hash, kbuf, ksiz); int bidx = hash % map->bnum; TCMAPREC *rec = map->buckets[bidx]; TCMAPREC **entp = map->buckets + bidx; TCMAPHASH2(hash, kbuf, ksiz); hash &= ~TCMAPKMAXSIZ; /*遍歷二叉樹，進行相應的刪除*/ while(rec){ uint32_t rhash = rec->ksiz & ~TCMAPKMAXSIZ; uint32_t rksiz = rec->ksiz & TCMAPKMAXSIZ; /*如果HASH不相同，循環遍歷左子樹或者右子樹*/ if(hash > rhash){ entp = &(rec->left); rec = rec->left; } else if(hash < rhash){ entp = &(rec->right); rec = rec->right; } else { /*如果相同，則比較KEY值 *不相同繼續遍歷*/ char *dbuf = (char *)rec + sizeof(*rec); int kcmp = TCKEYCMP(kbuf, ksiz, dbuf, rksiz); if(kcmp < 0){ entp = &(rec->left); rec = rec->left; } else if(kcmp > 0){ entp = &(rec->right); rec = rec->right; } else { /*key相同，record記錄減一*/ map->rnum--; map->msiz -= rksiz + rec->vsiz; /*將刪除的record節點的前續、後繼節點進行重新連接*/ if(rec->prev) rec->prev->next = rec->next; if(rec->next) rec->next->prev = rec->prev; if(rec == map->first) map->first = rec->next; if(rec == map->last) map->last = rec->prev; if(rec == map->cur) map->cur = rec->next; /*如果該刪除節點只有左子樹，則直接跳過該節點 *指向它的左子樹節點*/ if(rec->left && !rec->right){ *entp = rec->left; } /*如果只有右子樹，則直接跳過該節點 *指向它的右子樹節點*/ else if(!rec->left && rec->right){ *entp = rec->right; /*如果沒有子樹，直接將節點刪除*/ } else if(!rec->left && !rec->left){ *entp = NULL; } else { /*如果相同，即需要刪除該節點 *entp指向它的左子樹，然後遍歷左子樹的 右子樹，到最右邊的葉子節點（即：值最大的節點）， 將原先節點的右子樹，作爲最大節點的右子樹*/ *entp = rec->left; TCMAPREC *tmp = *entp; while(tmp->right){ tmp = tmp->right; } tmp->right = rec->right; } TCFREE(rec); return true; } } } return false; }

在刪除過程中，方法和通常介紹的方法有所不同。如果被刪除節點A有左、右子樹，那麼左子樹都小於它，
右子樹大於它。而A節點的左子樹的最右節點是它左子樹中最大的節點，如果刪除了A節點，那麼它的右子樹節點均大於左子樹，所以可以直接將它右子樹，連接到左子樹中最大節點的右子樹中。