o
0iC" @ s d Z dZddlmZ ddlZddlZddlZddlZddlZddl Z e
Z da e
Zdd Zee ee drEe jejejejd G d
d d eZd d
ZG dd dejZG dd dejZ 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 _shutdownlist_threads_queuesitemsputjoin)r tq r
@/opt/alt/python310/lib64/python3.10/concurrent/futures/thread.py _python_exit s
r register_at_fork)ZbeforeZafter_in_childZafter_in_parentc @ s& e Zd Zdd Zdd ZeejZdS ) _WorkItemc C s || _ || _|| _|| _d S N)futurefnargskwargs)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 Z set_running_or_notify_cancelr r r
BaseException
set_exceptionZ
set_result)r resultexcr
r
r run5 s
z
_WorkItem.runN) __name__
__module__ __qualname__r r classmethodtypes 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 ZLOGGERZcritical_initializer_failedgetr _idle_semaphorereleaser r )Zexecutor_referenceZ
work_queue initializerinitargsZexecutor work_itemr
r
r _workerE s@
r. 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 r/ p s r/ c @ sd e Zd Ze jZ dddZdd Ze j
jj e_ dd Z d
d Z
dd
dddZe 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)minos cpu_count
ValueErrorcallable TypeError _max_workersqueueZ SimpleQueue _work_queue threadingZ Semaphorer) set_threads_brokenr Lock_shutdown_lock_counter_thread_name_prefix _initializer _initargs)r Z max_workersZthread_name_prefixr+ r, r
r
r r { s$
zThreadPoolExecutor.__init__c O s | j N t; | jrt| j| jrtdtr tdt }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) rD r rB r/ r RuntimeErrorr ZFuturer r> r _adjust_thread_count)r r r r fwr
r
r submit s
RzThreadPoolExecutor.submitc C s | j jddr d S | jfdd}t| j}|| jk rGd| jp | |f }tj|t t
| || j| j | j
fd}| | j| | jt|< d S d S )Nr )Ztimeoutc S s | d d S r )r )_r r
r
r
weakref_cb s z;ThreadPoolExecutor._adjust_thread_count..weakref_cbz%s_%d)nametargetr )r) acquirer> lenrA r< rF r? ZThreadr. weakrefrefrG rH startaddr )r rO Z num_threadsZ thread_namer r
r
r rJ s&
z'ThreadPoolExecutor._adjust_thread_countc C st | j - d| _ z| j }W n
tjy Y nw |d ur'|jt| j qW d d S 1 s3w Y d S )NzBA thread initializer failed, the thread pool is not usable anymore) rD rB r>
get_nowaitr= Emptyr r r/ )r r- 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
| jd W d n1 s6w Y |rG| j D ]}|
q@d S d S r ) rD r r> rX r= rY r Zcancelr rA r
)r waitrZ r- r r
r
r shutdown s&
zThreadPoolExecutor.shutdown)Nr2 Nr
)T)r r r itertoolscount__next__rE r rM r Executorr0 rJ r' r\ r
r
r
r r1 v s
&
r1 ) r0
__author__Zconcurrent.futuresr r] r= r? r" rT r7 WeakKeyDictionaryr r rC r r Z_register_atexithasattrr rR _at_fork_reinitr* objectr r. ZBrokenExecutorr/ r` r1 r
r
r
r s.
+