CasperSecurity
o
`�wiC" � @ s� d Z dZddlmZ ddlZddlZddlZddlZddlZddl Z e�
� Z da e�
� Zdd� Ze�e� ee d�rEe jejejejd � G d
d � d e�Zd d
� ZG dd� dej�ZG dd� dej�Z dS )z Implements ThreadPoolExecutor.z"Brian Quinlan (brian@sweetapp.com)� )�_baseNFc C sf t �
daW d � n1 sw Y tt�� �} | D ] \}}|�d � q | D ]\}}|�� q(d S �NT)�_global_shutdown_lock� _shutdown�list�_threads_queues�items�put�join)r �t�q� r
�0/usr/lib/python3.10/concurrent/futures/thread.py� _python_exit s �
�r �register_at_fork)�before�after_in_child�after_in_parentc @ s& e Zd Zdd� Zdd� Zeej�ZdS )� _WorkItemc C s || _ || _|| _|| _d S �N)�future�fn�args�kwargs)�selfr r r r r
r
r �__init__/ s
z_WorkItem.__init__c
C sn | j �� sd S z | j| ji | j��}W n ty. } z| j �|� d } W Y d }~d S d }~ww | j �|� d S r )r � set_running_or_notify_cancelr r r �
BaseException�
set_exception�
set_result)r �result�excr
r
r �run5 s
��z
_WorkItem.runN) �__name__�
__module__� __qualname__r r" � classmethod�types� GenericAlias�__class_getitem__r
r
r
r r . s
r c C s� |d ur(z||� W n t y' tjjddd� | � }|d ur$|�� Y d S w z; |jdd�}|d urG|�� ~| � }|d urE|j�� ~q)| � }t sS|d u sS|j rb|d urZd|_ |�
d � W d S ~q* t yu tjjddd� Y d S w )NzException in initializer:T)�exc_info)�blockzException in worker) r r �LOGGER�critical�_initializer_failed�getr" �_idle_semaphore�releaser r )�executor_reference�
work_queue� initializer�initargs�executor� work_itemr
r
r �_workerE s@ �
� �r8 c @ s e Zd ZdZdS )�BrokenThreadPoolzR
Raised when a worker thread in a ThreadPoolExecutor failed initializing.
N)r# r$ r% �__doc__r
r
r
r r9 p s r9 c @ sd e Zd Ze�� jZ ddd�Zdd� Ze j
jj e_ dd � Z d
d � Z
dd
d�dd�Ze j
jj e_ dS )�ThreadPoolExecutorN� r
c C s� |du rt dt�� p dd �}|dkrtd��|dur#t|�s#td��|| _t�� | _ t
� d�| _ t
� | _d| _d| _t
�� | _|pGd | �� | _|| _|| _dS )
a� Initializes a new ThreadPoolExecutor instance.
Args:
max_workers: The maximum number of threads that can be used to
execute the given calls.
thread_name_prefix: An optional name prefix to give our threads.
initializer: A callable used to initialize worker threads.
initargs: A tuple of arguments to pass to the initializer.
N� � � r z"max_workers must be greater than 0z initializer must be a callableFzThreadPoolExecutor-%d)�min�os� cpu_count�
ValueError�callable� TypeError� _max_workers�queue� SimpleQueue� _work_queue� threading� Semaphorer0 �set�_threads�_brokenr �Lock�_shutdown_lock�_counter�_thread_name_prefix� _initializer� _initargs)r � max_workers�thread_name_prefixr4 r5 r
r
r r { s$
�
zThreadPoolExecutor.__init__c O s� | j �N t�; | jrt| j��| jrtd��tr td��t�� }t||||�}| j �
|� | � � |W d � W d � S 1 sDw Y W d � d S 1 sTw Y d S )Nz*cannot schedule new futures after shutdownz6cannot schedule new futures after interpreter shutdown) rP r rN r9 r � RuntimeErrorr �Futurer rI r �_adjust_thread_count)r r r r �f�wr
r
r �submit� s
R�zThreadPoolExecutor.submitc C s� | j jdd�r d S | jfdd�}t| j�}|| jk rGd| jp | |f }tj|t t
� | |�| j| j | j
fd�}|�� | j�|� | jt|< d S d S )Nr )�timeoutc S s |� d � d S r )r )�_r r
r
r �
weakref_cb� s z;ThreadPoolExecutor._adjust_thread_count.<locals>.weakref_cbz%s_%d)�name�targetr )r0 �acquirerI �lenrM rF rR rJ �Threadr8 �weakref�refrS rT �start�addr )r r_ � num_threads� thread_namer r
r
r rY � s&
�
�� �z'ThreadPoolExecutor._adjust_thread_countc C st | j �- d| _ z| j�� }W n
tjy Y nw |d ur'|j�t| j�� qW d � d S 1 s3w Y d S )NzBA thread initializer failed, the thread pool is not usable anymore) rP rN rI �
get_nowaitrG �Emptyr r r9 )r r7 r
r
r r. � s ��"�z&ThreadPoolExecutor._initializer_failedTF)�cancel_futuresc C s� | j �0 d| _|r& z| j�� }W n
tjy Y n w |d ur%|j�� q
| j�d � W d � n1 s6w Y |rG| j D ]}|�
� q@d S d S r ) rP r rI rk rG rl r �cancelr rM r
)r �waitrm r7 r r
r
r �shutdown� s& �
�
�
�zThreadPoolExecutor.shutdown)Nr< Nr
)T)r# r$ r% � itertools�count�__next__rQ r r\ r �Executorr: rY r. rp r
r
r
r r; v s
�&
r; ) r: �
__author__�concurrent.futuresr rq rG rJ r' re rA �WeakKeyDictionaryr r rO r r �_register_atexit�hasattrr rb �_at_fork_reinitr1 �objectr r8 �BrokenExecutorr9 rt r; r
r
r
r �<module> s.
�+