File: //lib/python3.6/site-packages/botocore/retries/__pycache__/adaptive.cpython-36.pyc
3
�T__ � @ sr d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ eje�Z dd� Z
G dd� de �Z G d d
� d
e �Z
dS ) � N)�bucket)�
throttling)�standardc C sz t j� }tjd|j� d�}t jd|d�}t|�}tjtj � d�}t
|||||d�}| j j j
d|j� | j j j
d|j� |S ) Nr )Zstarting_max_rateZ
start_time� )�max_rate�clock)Zretry_event_adapter)�
rate_adjustor� rate_clocker� token_bucket�throttling_detectorr z before-sendz needs-retry)r ZClockr ZCubicCalculator� current_timeZ TokenBucket� RateClockerr ZThrottlingErrorDetectorZRetryEventAdapter�ClientRateLimiter�metaZevents�register�on_sending_request�on_receiving_response)Zclientr r r
r r Zlimiter� r � /usr/lib/python3.6/adaptive.py�register_retry_handler
s$
r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) r g @c C s2 || _ || _|| _|| _|| _d| _tj� | _d S )NF) �_rate_adjustor�
_rate_clocker�
_token_bucket�_throttling_detector�_clock�_enabled� threading�Lock�_lock)�selfr r r
r r r r r �__init__* s zClientRateLimiter.__init__c K s | j r| jj� d S )N)r r �acquire)r Zrequest�kwargsr r r r 4 s z$ClientRateLimiter.on_sending_requestc K s� | j j� }| jj� }| j�� | jjf |�s<d}| jj|�}nFd}| j sL|}nt
|| j j �}| jj
||�}tjd||| j j� d| _ t
|| j| �| j _ W d Q R X d S )NFTzfThrottling response received, new send rate: %s measured rate: %s, token bucket capacity available: %s)r �recordr r r r Zis_throttling_errorr Zsuccess_receivedr �minr r Zerror_received�logger�debugZavailable_capacity�_MAX_RATE_ADJUST_SCALE)r r" �
measured_rateZ timestampr Znew_rateZ rate_to_user r r r 9 s$
z'ClientRateLimiter.on_receiving_responseN)�__name__�
__module__� __qualname__r' r r r r r r r r & s
r c @ s< e Zd ZdZdZdZeefdd�Zd dd�Zed d
� �Z d S )
r
z7Tracks the rate at which a client is sending a request.g�������?g �?c C sD || _ d| _|| _tj| j j� �| _d| j | _d| _ t
j � | _ d S )Nr r )
r �_measured_rate�
_smoothing�math�floorr � _last_bucket�_TIME_BUCKET_RANGE�_time_bucket_scale�_countr r r )r r Z smoothingZtime_bucket_ranger r r r X s zRateClocker.__init__r c C s� | j �| | jj� }tj|| j �| j }| j|7 _|| jkr|| jt|| j � }|| j | j
d| j | _
d| _|| _| j
S Q R X d S )Nr r ) r r r r. r/ r2 r3 r0 �floatr- r, )r Zamount�tr Z current_rater r r r# b s
zRateClocker.recordc C s | j S )N)r, )r r r r r( s s zRateClocker.measured_rateN)r )
r) r* r+ �__doc__Z_DEFAULT_SMOOTHINGr1 r r# �propertyr( r r r r r
Q s
r
)r. Zloggingr Zbotocore.retriesr r r Z getLoggerr) r% r �objectr r
r r r r �<module> s
+