| Current File : //proc/self/root/usr/lib/python3/dist-packages/botocore/retries/__pycache__/bucket.cpython-312.pyc |
�
P��e� � �L � d Z ddlZddlZddlmZ G d� d� Z G d� d� Zy)zEThis module implements token buckets used for client side throttling.� N)�CapacityNotAvailableErrorc � � e Zd Zd� Zd� Zd� Zy)�Clockc � � y �N� ��selfs �9/usr/lib/python3/dist-packages/botocore/retries/bucket.py�__init__zClock.__init__ s � � � c �. � t j |� y r )�time�sleep�r
�amounts r r z Clock.sleep s � � �
�
�6�r
c �* � t j � S r )r r s r � current_timezClock.current_time s � ��y�y�{�r
N)�__name__�
__module__� __qualname__r r r r r
r r r s � �
��r
r c � � e Zd ZdZefd�Zed� � Zej d� � Zed� � Zed� � Z d d�Z
d� Z d � Z d
� Z
y )
� TokenBucketg �?c �� � d | _ d | _ d| _ || _ d | _ || _ t
j � | _ t
j | j � | _
|| _ y )Nr ) �
_fill_rate�
_max_capacity�_current_capacity�_clock�_last_timestamp� _min_rate� threading�Lock�_lock� Condition�_new_fill_rate_condition�max_rate)r
r& �clock�min_rates r r zTokenBucket.__init__ s[ � � ��� !���!"�� ��� � #�� �!��� �^�^�%��
�(1�(;�(;�D�J�J�(G��%� ��
r
c � � | j S r )r r s r r& zTokenBucket.max_rate! s � ���� r
c �B � | j 5 | j � t || j � | _ |dk\ r|| _ nd| _ t
| j | j
� | _ | j j � d d d � y # 1 sw Y y xY w)N� ) r% �_refill�maxr r r �minr �notify)r
�values r r& zTokenBucket.max_rate% s� � �
�
*�
*� 3�
�L�L�N� !�%�� � � 8�D�O���z�%*��"�%&��"� &)��&�&��(:�(:�&�D� "�
� )� )� 0� 0� 2�# 3� 3� 3�s �A?B�B c � � | j S r )r r s r � max_capacityzTokenBucket.max_capacity: s � ��!�!�!r
c � � | j S r )r r s r �available_capacityz TokenBucket.available_capacity> s � ��%�%�%r
c �l � | j 5 | j ||�� cddd� S # 1 sw Y yxY w)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 r6 s r �acquirezTokenBucket.acquireB s2 � � �
*�
*� =��=�=��e�=�<� =� =� =�s �*�3c �` � | j � || j k r| xj |z c_ y|s
t � �| j |� }|dkD rB| j j |� | j � | j |� }|dkD r�B| xj |z c_ y)NTr )r, r r �
_sleep_amountr% �wait)r
r r6 � sleep_amounts r r7 zTokenBucket._acquireO s� � � � � �� �T�+�+� +� � "� "�f� ,� "���/�1�1� �-�-�f�5�L� ��"� �-�-�2�2�<�@�� � �� #� 1� 1�&� 9� � ��"�
� "� "�f� ,� "�r
c �: � || j z
| j z S r )r r r s r r: zTokenBucket._sleep_amountf s � ���/�/�/�4�?�?�B�Br
c � � | j j � }| j �|| _ y | j }|| j z
| j z }t | j ||z � }|| _ || _ y r )r r r r r r. r )r
� timestamp�current_capacity� fill_amount� new_capacitys r r, zTokenBucket._refilli sx � ��K�K�,�,�.� � � � � '�#,�D� � � �1�1�� �4�#7�#7�7�4�?�?�J� ��4�-�-�/?�+�/M�N� �!-�� � (�� r
N)r+ T)r r r � _MIN_RATEr �propertyr& �setterr2 r4 r8 r7 r: r, r r
r r r sx � ��I�1:� !� � � � � �_�_�3� �3�( �"� �"� �&� �&� =��.C� )r
r )�__doc__r! r �botocore.exceptionsr r r r r
r �<module>rH s' �� K� � � 9�� �_)� _)r