B
*>ƒbÉàÝã @ s¢ d Z dZddlZddlmZ ddlZddlZddlZddlZddl Z e
¡ Z da dd„ Z
e e
¡ G dd „ d eƒZd
d „ ZG d d
„ d
ejƒZG dd„ dejƒZdS )z Implements ThreadPoolExecutor.z"Brian Quinlan (brian@sweetapp.com)é N)Ú_baseFc C sJ da tt ¡ ƒ} x| D ]\}}| d ¡ qW x| D ]\}}| ¡ q2W d S )NT)Ú _shutdownÚlistÚ_threads_queuesÚitemsÚputÚjoin)r ÚtÚq© r ú>/opt/alt/python37/lib64/python3.7/concurrent/futures/thread.pyÚ _python_exit! s r
c @ s e Zd Zdd„ Zdd„ 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 sf | j ¡ sd S y| j| j| jŽ}W n2 tk
rT } z| j |¡ d } W d d }~X Y nX | j |¡ d S )N)r Z set_running_or_notify_cancelr r r Ú
BaseExceptionÚ
set_exceptionZ
set_result)r ÚresultÚexcr r r Úrun4 s
z
_WorkItem.runN)Ú__name__Ú
__module__Ú __qualname__r r r r r r r - s r c C sÚ |d k rPy ||Ž W n: t k
rN tjjddd | ƒ }|d k rJ| ¡ d S X y`xZ|jdd}|d k rt| ¡ ~qT| ƒ }tsŒ|d ksŒ|jr¨|d k ršd|_| d ¡ d S ~qTW W n$ t k
rÔ tjjddd Y nX d S )NzException in initializer:T)Úexc_info)ÚblockzException in worker) r r ZLOGGERZcriticalÚ_initializer_failedÚgetr r r )Zexecutor_referenceZ
work_queueÚ initializerÚinitargsZexecutorÚ work_itemr r r Ú_workerB s2
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% f s r% c @ sZ e Zd Ze ¡ jZddd„Zdd„ Ze j
jj e_ dd „ Z d
d „ Z
dd
d„Ze j
jj e_ dS )ÚThreadPoolExecutorNÚ r c C s |dkrt ¡ pdd }|dkr(tdƒ‚|dk r@t|ƒs@tdƒ‚|| _t ¡ | _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)ÚosÚ cpu_countÚ
ValueErrorÚcallableÚ TypeErrorÚ _max_workersÚqueueZ SimpleQueueÚ _work_queueÚsetÚ_threadsÚ_brokenr Ú threadingZLockÚ_shutdown_lockÚ_counterÚ_thread_name_prefixÚ _initializerÚ _initargs)r Z max_workersZthread_name_prefixr! r" r r r r q s
zThreadPoolExecutor.__init__c O sÀ t | ƒdkr| ^}}} n>| s&tdƒ‚n0d|krB| d¡}| ^}} ntdt | ƒd ƒ‚|jZ |jrnt|jƒ‚|jr|tdƒ‚trˆtdƒ‚t ¡ }t
||| |ƒ}|j |¡ |
¡ |S Q R X d S )Né zDdescriptor 'submit' of 'ThreadPoolExecutor' object needs an argumentr z6submit expected at least 1 positional argument, got %dr) z*cannot schedule new futures after shutdownz6cannot schedule new futures after interpreter shutdown)Úlenr/ Úpopr7 r5 r% r Ú RuntimeErrorr ZFuturer r2 r Ú_adjust_thread_count)r r r r ÚfÚwr r r Úsubmit‘ s*
zThreadPoolExecutor.submitc C s‚ | j fdd„}t| jƒ}|| jk r~d| jp,| |f }tj|tt | |¡| j | j
| j fd}d|_ |
¡ | j |¡ | j t|< d S )Nc S s | d ¡ d S )N)r )Ú_r
r r r Ú
weakref_cb³ s z;ThreadPoolExecutor._adjust_thread_count..weakref_cbz%s_%d)ÚnameÚtargetr T)r2 r= r4 r0 r9 r6 ZThreadr$ ÚweakrefÚrefr: r; ZdaemonÚstartÚaddr )r rE Z num_threadsZ thread_namer r r r r@ ° s
z'ThreadPoolExecutor._adjust_thread_countc C sd | j T d| _xFy| j ¡ }W n tjk
r6 P Y nX |d k r|j t| jƒ¡ qW W d Q R X d S )NzBA thread initializer failed, the thread pool is not usable anymore) r7 r5 r2 Z
get_nowaitr1 ZEmptyr r r% )r r# r r r r Å s z&ThreadPoolExecutor._initializer_failedTc C sD | j d| _| j d ¡ W d Q R X |r@x| jD ] }| ¡ q0W d S )NT)r7 r r2 r r4 r )r Úwaitr r r r ÚshutdownÒ s zThreadPoolExecutor.shutdown)Nr( Nr )T)r r r Ú itertoolsÚcountÚ__next__r8 r rC r ÚExecutorr& r@ r rM r r r r r' l s
r' )r& Ú
__author__ÚatexitZconcurrent.futuresr rN r1 r6 rH r+ ÚWeakKeyDictionaryr r r
ÚregisterÚobjectr r$ ZBrokenExecutorr% rQ r' r r r r Ú s
$