File: //lib/python3.6/site-packages/botocore/retries/__pycache__/bucket.cpython-36.pyc
3
�T_� � @ sD d Z ddlZddlZddlmZ G dd� de�ZG dd� de�ZdS )zEThis module implements token buckets used for client side throttling.� N)�CapacityNotAvailableErrorc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�Clockc C s d S )N� )�selfr r � /usr/lib/python3.6/bucket.py�__init__ s zClock.__init__c C s t j|� d S )N)�time�sleep)r �amountr r r r s z Clock.sleepc C s t j � S )N)r )r r r r � current_time s zClock.current_timeN)�__name__�
__module__� __qualname__r r r r r r r r s r c @ sp e Zd ZdZefdd�Zedd� �Zejdd� �Zedd� �Zed d
� �Z dd
d�Z
dd� Z dd� Z dd� Z
dS )� TokenBucketg �?c C sF d | _ d | _d| _|| _d | _|| _tj� | _tj | j�| _
|| _ d S )Nr ) �
_fill_rate�
_max_capacity�_current_capacity�_clock�_last_timestamp� _min_rate� threadingZLock�_lockZ Condition�_new_fill_rate_condition�max_rate)r r ZclockZmin_rater r r r s
zTokenBucket.__init__c C s | j S )N)r )r r r r r " s zTokenBucket.max_ratec
C s\ | j �L | j� t|| j�| _|dkr.|| _nd| _t| j| j�| _| j j� W d Q R X d S )N� ) r �_refill�maxr r r �minr Znotify)r �valuer r r r & s
c C s | j S )N)r )r r r r � max_capacity: s zTokenBucket.max_capacityc C s | j S )N)r )r r r r �available_capacity> s z TokenBucket.available_capacityr Tc C s | j � | j||d�S Q R X dS )aJ Acquire token or return amount of time until next token available.
If block is True, then this method will block until there's sufficient
capacity to acquire the desired amount.
If block is False, then this method will return True is capacity
was successfully acquired, False otherwise.
)r
�blockN)r �_acquire)r r
r! r r r �acquireB s
zTokenBucket.acquirec C sz | j � || jkr$| j|8 _dS |s.t� �| j|�}x*|dkrb| jj|� | j � | j|�}q:W | j|8 _dS d S )NTr )r r r �
_sleep_amountr �wait)r r
r! Z sleep_amountr r r r" O s
zTokenBucket._acquirec C s || j | j S )N)r r )r r
r r r r$ f s zTokenBucket._sleep_amountc C sT | j j� }| jd kr || _d S | j}|| j | j }t| j|| �}|| _|| _d S )N)r r r r r r r )r Z timestampZcurrent_capacityZ fill_amountZ new_capacityr r r r i s
zTokenBucket._refillN)r T)r r
r Z _MIN_RATEr �propertyr �setterr r r# r" r$ r r r r r r s
r )�__doc__r r Zbotocore.exceptionsr �objectr r r r r r �<module> s