3

��x`
�@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�?/usr/local/lib/python3.6/site-packages/uhashring/ring_ketama.py�<lambda>sz%KetamaRing.__init__.<locals>.<lambda>N)r�
_distribution�_keys�_nodes�	_replicas�_ring�
_listbytes)�selfZreplicasrrr	�__init__	s



zKetamaRing.__init__r
cCs`|jt
t|
�
jd
�
�
j��
}|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�replicaZdhZrdrrr	�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|jj
�D]}||d7}qW||_t�}g}i}xN|jj�D]@\}}x6|j||�D]&}|||<t||�
||d7<qZWqDW||_||_||_	dS)z,Generate a ketama compatible continuum/ring.r
rrN)
r
�valuesr"r�itemsr)rrrr)	r�nodesr"r%rrrr$�
hrrr	�_create_ring/s













zKetamaRing._create_ringcCsLy|jj
|
�

Wn*tk
r:


td
j|
|jj���
�
YnX|j|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�keysr1)rr$rrr	�_remove_nodeBs




zKetamaRing._remove_nodeN)
r)
r
)�__name__�
__module__�__qualname__�__doc__rrr)�staticmethodrr1r7rrrr	rs



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