File: //lib/python3/dist-packages/botocore/retries/__pycache__/adaptive.cpython-310.pyc
o
.&�a � @ sr d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ e�e�Z dd� Z
G dd� de �Z G d d
� d
e �Z
dS ) � N)�bucket)�
throttling)�standardc C sz t �� }tjd|�� d�}t jd|d�}t|�}tjt� � d�}t
|||||d�}| j j �
d|j� | j j �
d|j� |S ) Nr )�starting_max_rate�
start_time� )�max_rate�clock)�retry_event_adapter)�
rate_adjustor� rate_clocker� token_bucket�throttling_detectorr z before-sendz needs-retry)r �Clockr �CubicCalculator� current_time� TokenBucket� RateClockerr �ThrottlingErrorDetector�RetryEventAdapter�ClientRateLimiter�meta�events�register�on_sending_request�on_receiving_response)�clientr r r
r r �limiter� r �;/usr/lib/python3/dist-packages/botocore/retries/adaptive.py�register_retry_handler
s. �����r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) r g @c C s2 || _ || _|| _|| _|| _d| _t�� | _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
| j�� d S d S �N)r& r# �acquire)r* �request�kwargsr r r r 4 s �z$ClientRateLimiter.on_sending_requestc K s� | j �� }| j�� }| j�D | jjdi |��s | j�|�}n!| j s$|}nt
|| j j �}| j�
||�}t�d||| j j� d| _ t
|| j| �| j _ W d � d S 1 sTw Y d S )NzfThrottling response received, new send rate: %s measured rate: %s, token bucket capacity available: %sTr )r" �recordr% r r) r$ �is_throttling_errorr! �success_receivedr& �minr# r �error_received�logger�debug�available_capacity�_MAX_RATE_ADJUST_SCALE)r* r/ �
measured_rate� timestamp�new_rate� rate_to_user r r r 9 s(
��
�"�z'ClientRateLimiter.on_receiving_responseN)�__name__�
__module__� __qualname__r8 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| _|| _t�| j �� �| _d| j | _d| _ t
� � | _ d S )Nr r )
r% �_measured_rate�
_smoothing�math�floorr � _last_bucket�_TIME_BUCKET_RANGE�_time_bucket_scale�_countr' r( r) )r* r � smoothing�time_bucket_ranger r r r+ V s zRateClocker.__init__r c C s� | j �E | j�� }t�|| j �| j }| j|7 _|| jkr>| jt|| j � }|| j | j
d| j | _
d| _|| _| j
W d � S 1 sKw Y d S )Nr r ) r) r% r rB rC rF rG rD �floatrA r@ )r* �amount�tr � current_rater r r r0 ` s(
��
���$�zRateClocker.recordc C s | j S r, )r@ )r* r r r r9 q s zRateClocker.measured_rateN)r )
r= r> r? �__doc__�_DEFAULT_SMOOTHINGrE r+ r0 �propertyr9 r r r r r O s
�
r )rB �loggingr' �botocore.retriesr r r � getLoggerr= r5 r �objectr r r r r r �<module> s
)