B

��f�	�@s6dd
lm
Z

ddlmZ
ddlmZ
Gdd�d�ZdS)�)
�insort)
�Counter)
�md5c@sHeZ
dZd
Zddd�
Zddd�
Zdd	�Zed
d��
Zdd
�Z	dd�Z
dS)�
KetamaRingz6Implement a ketama compatible consistent hashing ring.�cCs.t�|_
g|_i|_|
|_i|_d
d�|_dS)zCreate a new HashRing.c


Ss|S)
N�)
�
xrr�F/opt/alt/python37/lib/python3.7/site-packages/uhashring/ring_ketama.py�<lambda>�z%KetamaRing.__init__.<locals>.<lambda>N)r�
_distribution�_keys�_nodes�	_replicas�_ring�
_listbytes)�selfZreplicasrrr	�__init__	s



zKetamaRing.__init__r
cCs`|�t
t|
�
�d
�
�
���
}|d}|d|d>|d|d>B|d|d>B|d	|BS)
z4Returns a ketama compatible hash from the given key.zutf-8r�����
�r
)rr�str�encode�digest)r�key�replicaZdh�rdrrr	�hashis

zKetamaRing.hashiccsl|d
t|j
�
|d|j}xFtd|�D]8}|
�d|��}x$td|j�D]}|j||d�V
qLWq,WdS)z�Calculate the weight factor of the given node and
        yield its hash key for every configured replica.

        :param node_name: the node name.
        Zvnodes�weightr
�
-)
rN)�lenr�_weight_sum�rangerr )r�	node_name�	node_conf�ks�
wZw_node_name�
irrr	�_hashi_weight_generators
 



z"KetamaRing._hashi_weight_generatorc

Cs
tt
|�S)
ziPython 2 compatible int iterator from str.

        :param data: the string to int iterate upon.
        )�map�ord)
�datarrr	r%szKetamaRing._listbytesc	Cs�d
}x|j�
�D]}||d7}qW||_t�}g}i}xN|j��D]@\}}x6|�||�D]&}|||<t||�
||d7<qZWqDW||_||_||_	dS)z,Generate a ketama compatible continuum/ring.r
r!rN)
r�valuesr$r�itemsr+rrr
r)	rZnodesr$r'rr
rr&�
hrrr	�_create_ring-s













zKetamaRing._create_ringc	CsLy|j�
|
�

Wn*tk
r:


td
�|
|j����
�
YnX|�|j�

dS)zaRemove the given node from the continuum/ring.

        :param node_name: the node name.
        z(node '{}' not found, available nodes: {}N)r�pop�	Exception�KeyError�format�keysr2)rr&rrr	�_remove_node@s



zKetamaRing._remove_nodeN)
r)
r
)�__name__�
__module__�__qualname__�__doc__rr r+�staticmethodrr2r8rrrr	rs



rN)�bisectr�collectionsr�hashlibrrrrrr	�<module>
s