a
XC?hC" ã @ sÊ d Z dZddlmZ ddlZddlZddlZddlZddlZddl Z e
¡ Z da e
¡ Zdd„ Ze e¡ ee dƒrŠe 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 s 0 Y tt ¡ ƒ} | D ]\}}| d ¡ q8| D ]\}}| ¡ qPd S ©NT)Ú_global_shutdown_lockÚ _shutdownÚlistÚ_threads_queuesÚitemsÚputÚjoin)r ÚtÚq© r
ú>/opt/alt/python39/lib64/python3.9/concurrent/futures/thread.pyÚ _python_exit s " r Úregister_at_fork)ZbeforeZafter_in_childZafter_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 sl | j ¡ sd S z| j| ji | j¤Ž}W n4 tyZ } z | j |¡ d } W Y d }~nd }~0 0 | j |¡ d S r )r Z set_running_or_notify_cancelr r r Ú
BaseExceptionÚ
set_exceptionZ
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 urPz ||Ž W n: t yN tjjddd | ƒ }|d urH| ¡ Y d S 0 zx|jdd}|d urŒ| ¡ ~| ƒ }|d urˆ|j ¡ ~qR| ƒ }t s¤|d u s¤|j rÂ|d ur²d|_ |
d ¡ W d S ~qRW n" t yê tjjddd Y n0 d S )NzException in initializer:T)Úexc_info)ÚblockzException in worker) r r ZLOGGERZcriticalÚ_initializer_failedÚgetr Ú_idle_semaphoreÚreleaser r )Zexecutor_referenceZ
work_queueÚ initializerÚinitargsZexecutorÚ work_itemr
r
r Ú_workerE s8
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 @ s` e Zd Ze ¡ jZddd„Zdd„ Ze j
jj e_ dd „ Z d
d „ Z
dd
dœdd„Ze j
jj e_ dS )ÚThreadPoolExecutorNÚ r
c C s¢ |du r t dt ¡ pdd ƒ}|dkr.tdƒ‚|durFt|ƒsFtdƒ‚|| _t ¡ | _ t
d¡| _ t
ƒ | _d| _d| _t
¡ | _|pŽd | ¡ | _|| _|| _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ÚqueueZ 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 š tv | jr t| jƒ‚| jr,tdƒ‚tr8tdƒ‚t ¡ }t||||ƒ}| j
|¡ | ¡ |W d ƒ W d ƒ S 1 sˆ0 Y W d ƒ n1 s¦0 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 ÚfÚwr
r
r Úsubmit¡ s
zThreadPoolExecutor.submitc C sŽ | j jddrd S | jfdd„}t| jƒ}|| jk rŠd| jp>| |f }tj|t t
| |¡| j| j | j
fd}| ¡ | j |¡ | jt|< 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)ÚnameÚtargetr )r) Úacquirer> ÚlenrA r< rF r? ZThreadr. ÚweakrefÚrefrG rH ÚstartÚaddr )r rO Z num_threadsZ thread_namer r
r
r rJ ´ s$
ÿ
ýÿ z'ThreadPoolExecutor._adjust_thread_countc C st | j Z d| _z| j ¡ }W n tjy4 Y qRY n0 |d ur|j t| jƒ¡ qW d ƒ n1 sf0 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 b d| _|rNz| j ¡ }W n tjy8 Y qNY n0 |d ur|j ¡ q| j d ¡ W d ƒ n1 sn0 Y |r| j D ] }|
¡ q‚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 Ú itertoolsÚcountÚ__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_atexitÚhasattrr rR Ú_at_fork_reinitr* Úobjectr r. ZBrokenExecutorr/ r` r1 r
r
r
r Ú s,
þ+