3
\R
@ sP d Z ddlZddlZddlZddlZdZdZdZdZdZ dZ
d Z d
Z d Z
e e
e e e
gZe d e
d
e de de
diZejdZG dd deZG dd deZG dd deZG dd deZG dd deZG dd d eZG d d d eZG d d d eZd!d" Z d#d$ Z d/d%d&Z ej d'd(Z defd)d*Z!G d+d, d,eZ"G d-d. d.eZ#dS )0z"Brian Quinlan (brian@sweetapp.com) NFIRST_COMPLETEDFIRST_EXCEPTION
ALL_COMPLETED
_AS_COMPLETEDPENDINGRUNNING CANCELLEDCANCELLED_AND_NOTIFIEDFINISHEDpendingrunning cancelledfinishedzconcurrent.futuresc @ s e Zd ZdS )ErrorN)__name__
__module__ __qualname__ r r 0/usr/lib64/python3.6/concurrent/futures/_base.pyr , s r c @ s e Zd ZdS )CancelledErrorN)r r r r r r r r 0 s r c @ s e Zd ZdS ) TimeoutErrorN)r r r r r r r r 4 s r c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
_Waiterc C s t j | _g | _d S )N) threadingZEventeventfinished_futures)selfr r r __init__: s
z_Waiter.__init__c C s | j j| d S )N)r append)r futurer r r
add_result> s z_Waiter.add_resultc C s | j j| d S )N)r r )r r r r r
add_exceptionA s z_Waiter.add_exceptionc C s | j j| d S )N)r r )r r r r r
add_cancelledD s z_Waiter.add_cancelledN)r r r r r r r! r r r r r 8 s r c s@ e Zd Z fddZ fddZ fddZ fddZ ZS ) _AsCompletedWaiterc s t t| j tj | _d S )N)superr" r r Locklock)r ) __class__r r r J s z_AsCompletedWaiter.__init__c
s0 | j tt| j| | jj W d Q R X d S )N)r% r# r" r r set)r r )r& r r r N s z _AsCompletedWaiter.add_resultc
s0 | j tt| j| | jj W d Q R X d S )N)r% r# r" r r r' )r r )r& r r r S s z _AsCompletedWaiter.add_exceptionc
s0 | j tt| j| | jj W d Q R X d S )N)r% r# r" r! r r' )r r )r& r r r! X s z _AsCompletedWaiter.add_cancelled)r r r r r r r!
__classcell__r r )r& r r" G s r" c s4 e Zd Z fddZ fddZ fddZ ZS )_FirstCompletedWaiterc s t j| | jj d S )N)r# r r r' )r r )r& r r r ` s z _FirstCompletedWaiter.add_resultc s t j| | jj d S )N)r# r r r' )r r )r& r r r d s z#_FirstCompletedWaiter.add_exceptionc s t j| | jj d S )N)r# r! r r' )r r )r& r r r! h s z#_FirstCompletedWaiter.add_cancelled)r r r r r r! r( r r )r& r r) ] s r) c sH e Zd Z fddZdd Z fddZ fddZ fd d
Z ZS ) _AllCompletedWaiterc s$ || _ || _tj | _t j d S )N)num_pending_callsstop_on_exceptionr r$ r% r# r )r r+ r, )r& r r r o s
z _AllCompletedWaiter.__init__c
C s4 | j $ | jd8 _| js&| jj W d Q R X d S )N )r% r+ r r' )r r r r _decrement_pending_callsu s z,_AllCompletedWaiter._decrement_pending_callsc s t j| | j d S )N)r# r r. )r r )r& r r r { s z _AllCompletedWaiter.add_resultc s* t j| | jr | jj n| j d S )N)r# r r, r r' r. )r r )r& r r r s z!_AllCompletedWaiter.add_exceptionc s t j| | j d S )N)r# r! r. )r r )r& r r r! s z!_AllCompletedWaiter.add_cancelled) r r r r r. r r r! r( r r )r& r r* l s
r* c @ s$ e Zd Zdd Zdd Zdd ZdS )_AcquireFuturesc C s t |td| _d S )N)key)sortedidfutures)r r3 r r r r s z_AcquireFutures.__init__c C s x| j D ]}|jj qW d S )N)r3
_conditionacquire)r r r r r __enter__ s z_AcquireFutures.__enter__c G s x| j D ]}|jj qW d S )N)r3 r4 release)r argsr r r r __exit__ s z_AcquireFutures.__exit__N)r r r r r6 r9 r r r r r/ s r/ c C s |t krt }nZ|tkr t }nJtdd | D }|tkrHt|dd}n"|tkr^t|dd}n td| x| D ]}|j j
| qpW |S )Nc s s | ]}|j ttgkV qd S )N)_stater r
).0fr r r s z._create_and_install_waiters..T)r, Fz Invalid return condition: %r) r r" r r) sumr r* r
ValueError_waitersr )fs return_whenwaiterZ
pending_countr< r r r _create_and_install_waiters s
rD c c sX xR| rR| d }x|D ]}|j | qW |j |jj | W d Q R X ~| j V qW d S )Nr- )remover4 r@ pop)rA rC ref_collectr< Z futures_setr r r _yield_finished_futures s
rI c c sJ |d k r|t j }t| } t| }t| * tdd | D }| | }t| t}W d Q R X t|}zt||| fdE d H x|r|d krd }n(|t j }|dk rt dt||f |j
j | |j |j
}g |_
|j
j W d Q R X |j t||| |fdE d H q~W W d x,| D ]$}|j |jj| W d Q R X q W X d S )Nc s s | ]}|j ttgkr|V qd S )N)r: r r
)r; r< r r r r= s z as_completed..)rH r z %d (of %d) futures unfinished)time monotonicr' lenr/ rD r listrI r r waitr% r clearreverser4 r@ rF ) rA timeoutend_timeZ
total_futuresr r rC Z wait_timeoutr< r r r as_completed s@
rS DoneAndNotDoneFuturesz
done not_donec C s t | tdd | D }t| | }|tkr>|r>t||S |tkrf|rftdd |D rft||S t|t| krt||S t| |}W d Q R X |jj | x*| D ]"}|j
|j j | W d Q R X qW |j
|j t|t| | S )Nc s s | ]}|j ttgkr|V qd S )N)r: r r
)r; r< r r r r= s zwait..c s s( | ] }|j r|j d k r|V qd S )N)r
exception)r; r< r r r r= $ s )r/ r' r rT r anyrL rD r rN r4 r@ rF updater )rA rQ rB doneZnot_donerC r< r r r rN s"
rN c @ s e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dd Z d ddZ d ddZ
dd Zdd Zd d ZdS ) Futurec C s, t j | _t| _d | _d | _g | _g | _d S )N) r Z Conditionr4 r r: _result
_exceptionr@ _done_callbacks)r r r r r 8 s
zFuture.__init__c C sB x<| j D ]2}y || W q tk
r8 tjd| Y qX qW d S )Nz!exception calling callback for %r)r\ ExceptionLOGGERrU )r callbackr r r _invoke_callbacksA s
zFuture._invoke_callbacksc C s | j v | jtkr`| jrz <%s at %#x state=%s returned %s>z<%s at %#x state=%s>) r4 r: r
r[ r& r r2 _STATE_TO_DESCRIPTION_MAPrZ )r r r r __repr__H s"
zFuture.__repr__c
C sR | j : | jttgkrdS | jttgkr,dS t| _| j j W d Q R X | j dS )NFT)r4 r: r r
r r
notify_allr` )r r r r cancel\ s z
Future.cancelc
C s | j | jttgkS Q R X d S )N)r4 r: r r )r r r r r
o s zFuture.cancelledc C s | j | jtkS Q R X d S )N)r4 r: r )r r r r r t s zFuture.runningc C s" | j | jtttgkS Q R X d S )N)r4 r: r r r
)r r r r rX y s z Future.donec C s | j r| j n| jS d S )N)r[ rZ )r r r r Z __get_result~ s zFuture.__get_resultc C s> | j & | jtttgkr(| jj| d S W d Q R X || d S )N)r4 r: r r r
r\ r )r fnr r r add_done_callback s
zFuture.add_done_callbackNc
C sx | j h | jttgkr t n| jtkr0| j S | j j| | jttgkrRt n| jtkrd| j S t W d Q R X d S )N) r4 r: r r r r
_Future__get_resultrN r )r rQ r r r result s
z
Future.resultc
C st | j d | jttgkr t n| jtkr.| jS | j j| | jttgkrPt n| jtkr`| jS t W d Q R X d S )N) r4 r: r r r r
r[ rN r )r rQ r r r rU s
zFuture.exceptionc C st | j d | jtkr6t| _x| jD ]}|j| q W dS | jtkrJt| _dS tj dt
| | j t dW d Q R X d S )NFTz!Future %s in unexpected state: %szFuture in unexpected state) r4 r: r r r@ r! r r r^ Zcriticalr2 RuntimeError)r rC r r r set_running_or_notify_cancel s
z#Future.set_running_or_notify_cancelc
C sN | j 6 || _t| _x| jD ]}|j| q W | j j W d Q R X | j d S )N)r4 rZ r
r: r@ r rc r` )r rh rC r r r
set_result s zFuture.set_resultc
C sN | j 6 || _t| _x| jD ]}|j| q W | j j W d Q R X | j d S )N)r4 r[ r
r: r@ r rc r` )r rU rC r r r
set_exception s zFuture.set_exception)N)N)r r r r r` rb rd r
r rX rg rf rh rU rj rk rl r r r r rY 5 s
"(
rY c @ s> e Zd Zdd ZdddddZdd d
Zd d Zd
d ZdS )Executorc O s
t d S )N)NotImplementedError)r re r8 kwargsr r r submit s zExecutor.submitNr- )rQ chunksizec sB d k rt j fddt| D fdd}| S )Nc s g | ]}j f| qS r )rp )r; r8 )re r r r
? s z Executor.map..c 3 sd zFj x8rBd kr(j j V q j j tj V q W W d xD ] } | j qNW X d S )N)rP rG rh rJ rK rd )r )rR rA rQ r r result_iteratorC s
z%Executor.map..result_iterator)rJ rK zip)r re rQ rq iterablesrs r )rR re rA r rQ r map& s
z Executor.mapTc C s d S )Nr )r rN r r r shutdownR s zExecutor.shutdownc C s | S )Nr )r r r r r6 _ s zExecutor.__enter__c C s | j dd dS )NT)rN F)rw )r exc_typeZexc_valZexc_tbr r r r9 b s zExecutor.__exit__)T)r r r rp rv rw r6 r9 r r r r rm s
,
rm )N)$
__author__ collectionsZloggingr rJ r r r r r r r r r
Z_FUTURE_STATESra Z getLoggerr^ r] r r r objectr r" r) r* r/ rD rI rS
namedtuplerT rN rY rm r r r r sR
>1 d