B

��f��@s*dd
lm
Z

ddlmZ
Gdd�d�ZdS)�)
�Counter)
�md5c@s0eZ
dZd
Zdd�Zdd�Zdd�Zdd	�Zd
S)�MetaRingz,Implement a tunable consistent hashing ring.cCsBt�|_
g|_i|_i|_|
r0t|
d
�s0td�
�
|
p:dd�|_dS)zaCreate a new HashRing.

        :param hash_fn: use this callable function to hash keys.
        �__call__z%hash_fn should be a callable functionc

Sstt
t|�
�d
�
�
��d�S)Nzutf-8�)�intr�str�encode�	hexdigest)
�key�r�D/opt/alt/python37/lib/python3.7/site-packages/uhashring/ring_meta.py�<lambda>�z#MetaRing.__init__.<locals>.<lambda>N)r�
_distribution�_keys�_nodes�_ring�hasattr�	TypeError�_hash_fn)�selfZhash_fnrrr
�__init__s




zMetaRing.__init__cCs
|�|
�
S)
z>Returns an integer derived from the md5 hash of the given key.)
r)rrrrr
�hashiszMetaRing.hashic	Csrx\|
D]T\}}xJtd
|d|d�D]0}|j
|d7<||j|�|�d|���
<q&WqWt|j���
|_dS)z,Generate a ketama compatible continuum/ring.r
�vnodes�weight�
�
-N)�rangerrr�sorted�keysr)rZnodes�	node_name�	node_conf�
wrrr
�_create_rings



"
zMetaRing._create_ringc	Cs�y|j�
|
�
}Wn*tk
r:


td
�|
|j����
�
YnVX|j�
|
�

x6td|d|d�D]}|j|�	|
�d|���
=q`Wt
|j���
|_dS)zaRemove the given node from the continuum/ring.

        :param node_name: the node name.
        z(node '{}' not found, available nodes: {}r
rrrN)r�pop�	Exception�KeyError�formatr rrrrrr)rr!r"r#rrr
�_remove_node"s






zMetaRing._remove_nodeN)�__name__�
__module__�__qualname__�__doc__rrr$r)rrrr
rs

rN)�collectionsr�hashlibrrrrrr
�<module>
s