我在实现Geecache一致性哈希算法时,遇到了一些问题,记录一下:
当我在实现向Map中添加真实节点的时候
func (m *Map) Add(keys ...string) {
for _, key := range keys {
for i := 0; i < m.replicas; i++ {
hash := int(m.hash([]byte(strconv.Itoa(i) + key)))
m.keys = append(m.keys, hash)
m.hashMap[hash] = key
}
}
}
我没有对m.keys切片进行排序,这就导致接下来我在根据key(表示数据的唯一标识符)查找节点的时候出现错误,因为sort.Search函数要求被搜索的切片是有序的,如果m.keys
切片没有经过排序,sort.Search
将无法正确找到虚拟节点的位置,从而导致返回的节点结果不正确。
func (m *Map) Get(key string) string {
if len(m.keys) == 0 {
return ""
}
hash := int(m.hash([]byte(key)))
//sort.Search 函数会返回使得表达式 m.keys[i] >= hash 为真的最小索引 i 值
idx := sort.Search(len(m.keys), func(i int) bool {
return m.keys[i] >= hash
})
//根据插入点的索引,从 m.keys 切片中取出对应的哈希值,并通过哈希值从 m.hashMap 中找到对应的节点名称
return m.hashMap[m.keys[idx%len(m.keys)]]
}
因此应该在Add函数的最后添加sort.Ints(m.keys)语句,将m.keys
切片排序:
func (m *Map) Add(keys ...string) {
for _, key := range keys {
for i := 0; i < m.replicas; i++ {
hash := int(m.hash([]byte(strconv.Itoa(i) + key)))
m.keys = append(m.keys, hash)
m.hashMap[hash] = key
}
}
sort.Ints(m.keys)
}