tokyo cabinet源代碼分析（5）

2.4數據記錄的查找

   在前面的部分對於記錄的插入進行了闡述。本節對通過key查找value方法進行了分析。

2.4.1TCMAP數組查找

  先映射到MAP數組的一個元素，然後基於該元素對於hash buckets數組進行訪問。

通過tcmdbget進行查找 /* Retrieve a record in an on-memory hash database. */ void *tcmdbget(TCMDB *mdb, const void *kbuf, int ksiz, int *sp){ assert(mdb && kbuf && ksiz >= 0 && sp); unsigned int mi; /*將key映射到map數組的一個元素上*/ TCMDBHASH(mi, kbuf, ksiz); /*通過mutex鎖唯一訪問*/ if(pthread_rwlock_rdlock((pthread_rwlock_t *)mdb->mmtxs + mi) != 0) return NULL; int vsiz; const char *vbuf = tcmapget(mdb->maps[mi], kbuf, ksiz, &vsiz); char *rv; if(vbuf){ TCMEMDUP(rv, vbuf, vsiz); *sp = vsiz; } else { rv = NULL; } pthread_rwlock_unlock((pthread_rwlock_t *)mdb->mmtxs + mi); return rv; }

2.4.2 hash buckets二叉樹查找

  查找二叉樹結構，獲得value對象，返回的是分配好的內存地址，所以查找使用以後，

需要進行釋放。

/*通過tcmapget遍歷二叉搜索樹，查找value*/ /* Retrieve a record in a map object. */ const void *tcmapget(const TCMAP *map, const void *kbuf, int ksiz, int *sp){ assert(map && kbuf && ksiz >= 0 && sp); if(ksiz > TCMAPKMAXSIZ) ksiz = TCMAPKMAXSIZ; uint32_t hash; TCMAPHASH1(hash, kbuf, ksiz); /*先定位到bucket一個元素上*/ TCMAPREC *rec = map->buckets[hash%map->bnum]; TCMAPHASH2(hash, kbuf, ksiz); hash &= ~TCMAPKMAXSIZ; /*遍歷二叉樹 *hash值相同，然後比較key值 */ while(rec){ uint32_t rhash = rec->ksiz & ~TCMAPKMAXSIZ; uint32_t rksiz = rec->ksiz & TCMAPKMAXSIZ; if(hash > rhash){ rec = rec->left; } else if(hash < rhash){ rec = rec->right; } else { /*dbuf的結構是rec結構體之後存放了key *TCKEYCMP先比較了兩個的keybuffer的大小， *然後再比較值*/ char *dbuf = (char *)rec + sizeof(*rec); int kcmp = TCKEYCMP(kbuf, ksiz, dbuf, rksiz); if(kcmp < 0){ rec = rec->left; } else if(kcmp > 0){ rec = rec->right; } else { *sp = rec->vsiz; /*找到之後，翻譯value所在buffer的地址 *通過TCMEMDUP複製內存, *所在在使用完了Value以後，應該釋放*/ return dbuf + rksiz + TCALIGNPAD(rksiz); } } }