U /e@sUdZddlmZdZddlZddlZddlZddlZddlZddl Z ddl m Z e Z ded<dad d ZeeGd d d Zd dZGddde jZdS)zImplements ThreadPoolExecutor.) annotationsz"Brian Quinlan (brian@sweetapp.com)N)_basez8weakref.WeakKeyDictionary[threading.Thread, queue.Queue]_threads_queuesFcCsBdatt}|D]\}}|dq|D]\}}|q,dSNT) _shutdownlistritemsputjoin)rtqr J/tmp/pip-unpacked-wheel-g426oqom/distributed/_concurrent_futures_thread.py _python_exit(s     rc@seZdZddZddZdS) _WorkItemcCs||_||_||_||_dSN)futurefnargskwargs)selfrrrrr r r__init__6sz_WorkItem.__init__c Csb|jsdSz|j|j|j}Wn.tk rP}z|j|W5d}~XYnX|j|dSr)rZset_running_or_notify_cancelrrr BaseExceptionZ set_exceptionZ set_result)rresulter r rrun<s z _WorkItem.runN)__name__ __module__ __qualname__rrr r r rr5srcCszzP|jdd}|dk r"|~q|}ts:|dks:|jrJ|dWdS~qWn$tk rttjjdddYnXdS)NT)blockzException in worker)exc_info)getrrr rrZLOGGERcritical)Zexecutor_referenceZ work_queueZ work_itemexecutorr r r_workerHs  r$c@sPeZdZeZd ddZddZej jj e_ ddZ d d d Z ej j j e _ dS)ThreadPoolExecutorNcCsh|dkrtpdd}|dkr(td||_t|_t|_d|_ t |_ |p`dt |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_workersqueueQueue _work_queueset_threadsr threadingLock_shutdown_locknext_counter_thread_name_prefix)r max_workersZthread_name_prefixr r rrds   zThreadPoolExecutor.__init__c OsZ|jJ|jrtdt}t||||}|j|||W5QRSQRXdS)Nz*cannot schedule new futures after shutdown) r4r RuntimeErrorrZFuturerr/r _adjust_thread_count)rrrrfwr r rsubmit|s zThreadPoolExecutor.submitcCsz|jfdd}t|j}||jkrvd|jp,||f}tj|tt |||jfd}d|_ | |j ||jt |<dS)NcSs|ddSr)r )_r r r r weakref_cbsz;ThreadPoolExecutor._adjust_thread_count..weakref_cbz%s_%d)nametargetrT)r/lenr1r,r7r2Threadr$weakrefrefdaemonstartaddr)rr?Z num_threadsZ thread_namer r r rr:s   z'ThreadPoolExecutor._adjust_thread_countTc Cs@|jd|_|jdW5QRX|r<|jD] }|q.dSr)r4rr/r r1r )rwaitr r r rshutdowns  zThreadPoolExecutor.shutdown)Nr&)T) rrr itertoolscountr6rr=rExecutor__doc__r:rJr r r rr%_s   r%)rN __future__r __author__atexitrKr)r-r2rDconcurrent.futuresrWeakKeyDictionaryr__annotations__rrregisterrr$rMr%r r r rs"