3 \@sdZdZddlZddlmZddlZddlZddlZddlZddl Z ej Z da ddZ eje Gdd d eZd d ZGd d d ejZdS)zImplements ThreadPoolExecutor.z"Brian Quinlan (brian@sweetapp.com)N)_baseFcCsJdattj}x|D]\}}|jdqWx|D]\}}|jq2WdS)NT) _shutdownlist_threads_queuesitemsputjoin)rtqr /usr/lib64/python3.6/thread.py _python_exit!s  r c@seZdZddZddZdS) _WorkItemcCs||_||_||_||_dS)N)futurefnargskwargs)selfrrrrr r r __init__-sz_WorkItem.__init__cCsf|jjsdSy|j|j|j}Wn2tk rT}z|jj|d}WYdd}~XnX|jj|dS)N)rZset_running_or_notify_cancelrrr BaseExceptionZ set_exceptionZ set_result)rresultexcr r r run3s  z _WorkItem.runN)__name__ __module__ __qualname__rrr r r r r,src Cs|yRxL|jdd}|dk r$|j~q|}ts<|dks<|jrJ|jddS~qWWn$tk rvtjjdddYnXdS)NT)blockzException in worker)exc_info)getrrrrrZLOGGERZcritical)Zexecutor_referenceZ work_queueZ work_itemexecutorr r r _worker@s   r c@sReZdZejjZd ddZddZe j jj e_ ddZ d d d Z e j j j e _ dS)ThreadPoolExecutorNcCsf|dkrtjpdd}|dkr(td||_tj|_t|_d|_ t j |_ |p^d|j |_dS)aInitializes 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. Nrz"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 max_workersZthread_name_prefixr r r r[s  zThreadPoolExecutor.__init__c OsN|j>|jrtdtj}t||||}|jj||j|SQRXdS)Nz*cannot schedule new futures after shutdown) r.r RuntimeErrorrZFuturerr*r_adjust_thread_count)rrrrfwr r r submitrs zThreadPoolExecutor.submitcCsz|jfdd}t|j}||jkrvd|jp,||f}tj|ttj |||jfd}d|_ |j |jj ||jt |<dS)NcSs|jddS)N)r)_r r r r weakref_cbsz;ThreadPoolExecutor._adjust_thread_count..weakref_cbz%s_%d)nametargetrT)r*lenr,r(r0r-ZThreadr weakrefrefZdaemonstartaddr)rr8Z num_threadsZ thread_namer r r r r3s      z'ThreadPoolExecutor._adjust_thread_countTc CsD|jd|_|jjdWdQRX|r@x|jD] }|jq0WdS)NT)r.rr*rr,r)rwaitr r r r shutdowns  zThreadPoolExecutor.shutdown)Nr")T)rrr itertoolscount__next__r/rr6rExecutor__doc__r3rAr r r r r!Vs    r!)rF __author__atexitconcurrent.futuresrrBr)r-r<r%WeakKeyDictionaryrrr registerobjectrr rEr!r r r r s