一致性哈希 代码实例

以下是一致性哈希的 Python 代码实现，以字符串的哈希值作为节点的标识：

import bisect
import hashlib

class ConsistentHash:

    def __init__(self, replicas=100):
        self.replicas = replicas
        self.ring = {}
        self.keys = []

    def add_node(self, node):
        for i in range(self.replicas):
            replica_key = self.get_replica_key(node, i)
            self.ring[replica_key] = node
            self.keys.append(replica_key)
        self.keys.sort()

    def remove_node(self, node):
        for i in range(self.replicas):
            replica_key = self.get_replica_key(node, i)
            del self.ring[replica_key]
            self.keys.remove(replica_key)

    def get_node(self, key):
        if not self.ring:
            return None
        h = self.hash_key(key)
        idx = bisect.bisect_left(self.keys, h)
        if idx == len(self.keys):
            idx = 0
        return self.ring[self.keys[idx]]

    def get_replica_key(self, node, i):
        return self.hash_key(f"{node}:{i}")

    def hash_key(self, key):
        return int(hashlib.sha256(key.encode()).hexdigest(), 16)

以下是一个简单的测试：

cons_hash = ConsistentHash()

cons_hash.add_node("node1")
cons_hash.add_node("node2")
cons_hash.add_node("node3")

print(cons_hash.get_node("key1"))  # 输出 node1
print(cons_hash.get_node("key2"))  # 输出 node2

cons_hash.remove_node("node2")

print(cons_hash.get_node("key1"))  # 输出 node1
print(cons_hash.get_node("key2"))  # 输出 node3

以上代码实现了一致性哈希的基本功能，包括添加节点、删除节点、根据键值获取节点等。