File: //usr/local/lib/python3.6/site-packages/uhashring/__pycache__/ring_meta.cpython-36.pyc
3
U�` � @ s* d dl mZ d dlmZ G dd� d�ZdS )� )�Counter)�md5c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �MetaRingz,Implement a tunable consistent hashing ring.c C sD t � | _g | _i | _i | _|r2t|d� r2td��|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 S s t tt| �jd��j� d�S )Nzutf-8� )�intr �str�encode� hexdigest)�key� r �=/usr/local/lib/python3.6/site-packages/uhashring/ring_meta.py�<lambda> s z#MetaRing.__init__.<locals>.<lambda>N)r �
_distribution�_keys�_nodes�_ring�hasattr� TypeError�_hash_fn)�selfZhash_fnr r r
�__init__ s zMetaRing.__init__c C s
| j |�S )z>Returns an integer derived from the md5 hash of the given key.)r )r r r r r
�hashi s zMetaRing.hashic C sr x\|D ]T\}}xJt d|d |d �D ]0}| j| d7 < || j| j|� d|� ��< q&W qW t| jj� �| _dS )z,Generate a ketama compatible continuum/ring.r �vnodes�weight� �-N)�ranger r r �sorted�keysr )r �nodes� node_name� node_conf�wr r r
� _create_ring s
"zMetaRing._create_ringc C s� y| j j|�}W n* tk
r: tdj|| j j� ���Y nVX | jj|� x6td|d |d �D ] }| j| j |� d|� ��= q`W t
| jj� �| _ dS )zaRemove the given node from the continuum/ring.
:param node_name: the node name.
z(node '{}' not found, available nodes: {}r r r r N) r �pop� Exception�KeyError�formatr r r r r r r )r r! r" r# r r r
� _remove_node$ s zMetaRing._remove_nodeN)�__name__�
__module__� __qualname__�__doc__r r r$ r) r r r r
r s
r N)� collectionsr �hashlibr r r r r r
�<module> s