3
\ @ s d Z dZddlZddlmZ ddlZddlZddlZddlZddl Z ej
Z da dd Z
eje
G dd d eZd
d ZG d d
d
ejZdS )z Implements ThreadPoolExecutor.z"Brian Quinlan (brian@sweetapp.com) N)_baseFc C sJ da ttj } x| D ]\}}|jd qW x| D ]\}}|j q2W d S )NT) _shutdownlist_threads_queuesitemsputjoin)r tq r 1/usr/lib64/python3.6/concurrent/futures/thread.py _python_exit! s r
c @ s e Zd Zdd Zdd ZdS ) _WorkItemc C s || _ || _|| _|| _d S )N)futurefnargskwargs)selfr r r r r r r __init__- s z_WorkItem.__init__c C sf | j j sd S y| j| j| j}W n2 tk
rT } z| j j| d } W Y d d }~X nX | j j| d S )N)r Z set_running_or_notify_cancelr r r
BaseExceptionZ
set_exceptionZ
set_result)r resultexcr r r run3 s
z
_WorkItem.runN)__name__
__module__ __qualname__r r r r r r r , s r c C s| yRxL|j dd}|d k r$|j ~q| }ts<|d ks<|jrJ|jd d S ~qW W n$ tk
rv tjjddd Y nX d S )NT)blockzException in worker)exc_info)getr r r r r ZLOGGERZcritical)Zexecutor_referenceZ
work_queueZ work_itemZexecutorr r r _worker@ s
r c @ sR e Zd Zej jZd ddZdd Ze j
jj e_ dd Z d
d
d Z
e j
j
j e
_ dS )ThreadPoolExecutorN c C sf |dkrt j pdd }|dkr(td|| _tj | _t | _d| _ t
j | _ |p^d| j
| _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.
N r z"max_workers must be greater than 0FzThreadPoolExecutor-%d)os cpu_count
ValueError _max_workersqueueZQueue _work_queueset_threadsr threadingZLock_shutdown_lock_counter_thread_name_prefix)r Z max_workersZthread_name_prefixr r r r [ s
zThreadPoolExecutor.__init__c O sN | j > | jrtdtj }t||||}| jj| | j |S Q R X d S )Nz*cannot schedule new futures after shutdown) r- r RuntimeErrorr ZFuturer r) r _adjust_thread_count)r r r r fwr r r submitr s zThreadPoolExecutor.submitc C sz | j fdd}t| j}|| jk rvd| jp,| |f }tj|ttj | || j fd}d|_
|j | jj | | j t
|< d S )Nc S s |j d d S )N)r )_r
r r r
weakref_cb s z;ThreadPoolExecutor._adjust_thread_count..weakref_cbz%s_%d)nametargetr T)r) lenr+ r' r/ r, ZThreadr weakrefrefZdaemonstartaddr )r r6 Z num_threadsZ thread_namer r r r r1 s
z'ThreadPoolExecutor._adjust_thread_countTc C sD | j d| _| jjd W d Q R X |r@x| jD ] }|j q0W d S )NT)r- r r) r r+ r )r waitr r r r shutdown s zThreadPoolExecutor.shutdown)Nr! )T)r r r itertoolscount__next__r. r r4 r Executor__doc__r1 r? r r r r r V s
r )rD
__author__atexitZconcurrent.futuresr r@ r( r, r: r$ WeakKeyDictionaryr r r
registerobjectr r rC r r r r r s