�
[%�h� � �l � d dl Z d dlZd dlmZmZ d dlmZ e j e� � Z G d� d� � Z
dS )� N)�ThreadPoolExecutor�Future)�BoundedSemaphorec �> � e Zd ZdZd� Zd� Zd� Zd� Zd d�Zde fd �Z
d
S ) �BoundedThreadExecutora9
BoundedExecutor behaves as a ThreadPoolExecutor which will block on
calls to submit() once the limit given as "bound" work items are queued for
execution.
While processing incoming connections, we need two things:
- the server must be able to process incoming connections
in multiple threads (one for each client) because we spend some time
trying to redirect incoming data to our servers
- the server must have some limit in order not to create new
threads indefinitely
This executor has two limits:
:param max_workers - which limits number of simultaneously running threads
:param maxqueuesize - which limits number of tasks to wait for the available thread
When both max_workers and maxqueuesize overflow, .submit function raises
queue.Full exception.
c �^ � t |�� � | _ t ||z � � | _ d S )N)� max_workers)r �executorr � semaphore)�self� maxqueuesizer s �J/opt/cloudlinux/venv/lib64/python3.11/site-packages/xray/agent/executor.py�__init__z BoundedThreadExecutor.__init__# s, � �*�{�C�C�C��
�)�,��*D�E�E����� c � � | S �N� )r s r � __enter__z BoundedThreadExecutor.__enter__' s � �� r c �. � | � � � d S r )�shutdown)r �exc_type�exc_val�exc_tbs r �__exit__z BoundedThreadExecutor.__exit__* s � � �
�
�����r c � � | j � d�� � }|s t j � | j j |g|�R i |��}|� | j � � |S # | j � � � � xY w)NF)�blocking) r �acquire�queue�Fullr
�submit�add_done_callback�_on_future_complete�release)r �fn�args�kwargs�success�futures r r z BoundedThreadExecutor.submit- s� � ��.�(�(�%�(�8�8��� ��*� � �)�T�]�)�"�>�t�>�>�>�v�>�>�F�
� $� $�T�%=� >� >� >��M�� � �N� "� "� $� $� $� ���s �A � A9Tc �: � | j � |� � d S r )r
r )r �waits r r z BoundedThreadExecutor.shutdown; s � � �
� � �t�$�$�$�$�$r �xc � � | j � � � |� � � r t � d� � d S d S )Nz Future completed with exception)r r# � exception�logger)r r+ s r r" z)BoundedThreadExecutor._on_future_complete> sN � � �� � � � � � �;�;�=�=� @� � � � >� ?� ?� ?� ?� ?� @� @r N)T) �__name__�
__module__� __qualname__�__doc__r r r r r r r" r r r r r s� � � � � � �� �&F� F� F�� � �� � � � � � %� %� %� %�@�V� @� @� @� @� @� @r r ) �loggingr �concurrent.futuresr r � threadingr � getLoggerr/ r. r r r r �<module>r7 s� �� ����
� � � � 9� 9� 9� 9� 9� 9� 9� 9� &� &� &� &� &� &� �� �8� $� $��3@� 3@� 3@� 3@� 3@� 3@� 3@� 3@� 3@� 3@r |