0FieW2(ddlmZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m ZddlmZddlmZddlmZddlZddlmZmZerdd lmZejeZd Zd Zd Z d Z!GddZ"GddZ#dZ$dS)) annotationsN)Callable)Queue) TYPE_CHECKING)Future)IOLoop)get_mp_contextwait_for)Selfc |j|g|RdS#t$r.}tjdt |sYd}~dSd}~wwxYw)zQ Helper to silence "IOLoop is closing" exception on IOLoop.add_callback. zIOLoop is clos(ed|ing)N) add_callback RuntimeErrorresearchstr)loopfuncargsexcs 3lib/python3.11/site-packages/distributed/process.py_loop_add_callbackrs$&&&&&&& y13s88<<        s A #AA c\|s||dSdSN) cancelled set_result)futurevalues r#_future_set_result_unless_cancelledr)s:     !%     !!c|s||dStd|dS)Nz$Exception after Future was cancelled)exc_info)r set_exceptionloggererror)rrs r&_future_set_exception_unless_cancelledr%.sN     KS!!!!! ;c JJJJJrc ||i|}t|t||dS#t$r"}t|t||Yd}~dSd}~wwxYwr)rr Exceptionr%)rrrrkwargsresrs r_call_and_set_futurer*5sSdD#F## 4!DfcRRRRR VVV 4!GQTUUUUUUUUUVs# AA  AceZdZdZdZdZdS) _ProcessStateFN)__name__ __module__ __qualname__is_alivepidexitcoderrr,r,@sH CHHHrr,cDeZdZUdZded<d%dZdZdZd Zd&dZ e dZ e dZ e d'dZ e dZdZd(dZd(dZd)dZdZd*dZdZed Zed!Zed"Zed#Zejd$ZdS)+ AsyncProcessz A coroutine-compatible multiprocessing.Process-alike. All normally blocking methods are wrapped in Tornado coroutines. multiprocessing.Process_processNr3c |pi}t|stdt|t|_|pt jd|_t d\}|_ t |j ||||||j tjjf|_|jj|_t'j|t*|j|_t/|_t3|_d|_d|_|dS)Nz#`target` needs to be callable, not F)instance)duplextargetnamer)callable TypeErrortyper,_statercurrent_loopr Pipe_keep_child_aliveProcess_rundaskconfig global_configr7r=_nameweakreffinalize_asyncprocess_finalizer_proc_finalizerPyQueue_watch_qr _exit_future_exit_callback_closed_start_threads)selfrr<r=rr(parent_alive_pipes r__init__zAsyncProcess.__init__Ns<2 TR$v,,RRSS S#oo ;V^U;;; 5C4D4D4I4IQV4I4W4W141&((009!& ) 1    ]' &/ )4=    "HH"  rc2d|jjd|jdS)N< >) __class__r-rKrVs r__repr__zAsyncProcess.__repr__xs":4>*::TZ::::rc2|jrtddS)Nz(invalid operation on closed AsyncProcess)rT ValueErrorr^s r _check_closedzAsyncProcess._check_closed{s( < IGHH H I Irc dtj|jd|jzt j||j|j|j|j |j f|_ d|j _ |j d}t j|||j |_d|j_dS)Nz#AsyncProcess %s watch message queuer;Tc4|ddidS)Nopstop) put_nowaitqs r stop_threadz0AsyncProcess._start_threads..stop_threads LL$ ( ( ( ( (rrhF) threadingThread_watch_message_queuer=rLrefr7rCrArQrR_watch_message_threaddaemonstartrM_thread_finalizeratexit)rVrjs rrUzAsyncProcess._start_threadss%.%5,6B D!!    !  & & & "-1") "((*** ) ) ) ")!1$ t}!U!U!U(-%%%rr2intreturnNonecd|_|j|||j|dSr)r7rSrRr)rVr2s r_on_exitzAsyncProcess._on_exitsC   *    % % % $$X.....rcrfd}tj|}d|_|dS)zP Immediately exit the process when parent_alive_pipe is closed. c td#t$rtjdYdSwxYw)Nz'unexpected state: should be unreachable)recvrEOFErroros_exit)rWsrmonitor_parentz@AsyncProcess._immediate_exit_when_closed..monitor_parents` N"&&(((##LMMM     s&AA)r<TN)rkrlrprq)clsrWrts ` r_immediate_exit_when_closedz(AsyncProcess._immediate_exit_when_closedsN  N N N N N,  N 3 3 3  rc|||dtj_t jt jj|d||i|dS)N MainThreadold)priority) closerrkcurrent_threadr=rHrIupdaterJ)rr<rr(rWrEinherit_configs rrGzAsyncProcess._runs{ !!! ''(9:::*6  ""' 4;4nuUUUrprocessc t j  fd} }td d||d} | dkrt ||d|nX| dkrt ||djn5| d krt ||djn| d krdSJ|) Nc&tjtjdzf}d|_|d_j_t ddjdS)Nz"AsyncProcess %s watch process joinr;T[z] created process with pid ) rqrkrlr5_watch_processrpr0r1r#debug)threadr=rrirselfrefstates r_startz1AsyncProcess._watch_message_queue.._starts MMOOO%#29D@wq1F !FM LLNNN!EN EI LLHQHH59HH I I I I IrTrz] got message rerqr terminatekillrf)reprr=getr#rr*rr) rrrrrri exit_futurermsgrer=rs `` `` @@rrmz!AsyncProcess._watch_message_queues- OOwyy~ J J J J J J J J J J %%''C LL5Q55c55 6 6 6TBW}}$T3x=&AAAA{""$T3x=':KLLLLv$T3x=',GGGGv# ! rct|}||jx}}|d}d|_||_|} |t |j|j|d}n#d}wxYw|#td||j dSt d||j |dS)NFzE[%s] process %r exit status was already read will report exitcode 255z#[%s] process %r exited with code %r) rjoinr2r0rrCrxr#warningr1r) rrrrrirr2original_exit_coderVs rrzAsyncProcess._watch_processs OO (/(88%  H!wyy "4:t}hGGGDD4DKKKK  % NNW       LL>59h W W W W Ws A22A6c|t}|jd|d|S)zQ Start the child process. This method returns a future. rqrerrbrrQrgrVfuts rrqzAsyncProcess.startsB hh   3!?!?@@@ rasyncio.Future[None]c|t}|jd|d|S)zTerminate the child process. This method returns a future. See also -------- multiprocessing.Process.terminate rrrrs rrzAsyncProcess.terminate!sB "HH   s!C!CDDD rc|t}|jd|d|S)zSend SIGKILL to the child process. On Windows, this is the same as terminate(). This method returns a future. See also -------- multiprocessing.Process.kill rrrrs rrzAsyncProcess.kill/sB "HH   #!>!>??? rcK||jj Jd|jjdSt t j|j|d{VdS)z_ Wait for the child process to exit. This method returns a coroutine. Nzcan only join a started process)rbrAr1r2r asyncioshieldrR)rVtimeouts rrzAsyncProcess.join>sz {**,M*** ;  + Fw~d&788'BBBBBBBBBBBrc\|js$|d|_d|_dSdS)z Stop helper thread and release resources. This method returns immediately and does not ensure the child process has exited. NT)rTrrr7r^s rrzAsyncProcess.closeLs< |  " " $ $ $ DMDLLL  rrVr rCallable[[Self], None]ctj|r Jdt|s Jd|jj Jd||_dS)z Set a function to be called by the event loop when the process exits. The function is called with the AsyncProcess as sole argument. The function may not be a coroutine function. z-exit callback may not be a coroutine functionz exit callback should be callableNz5cannot set exit callback when process already started)inspectiscoroutinefunctionr>rAr1rS)rVrs rset_exit_callbackzAsyncProcess.set_exit_callbackVsy.    ; ; : ; ; ;~~AAAAAA KO # # B $ # #"rc|jjSr)rAr0r^s rr0zAsyncProcess.is_alivegs {##rc|jjSr)rAr1r^s rr1zAsyncProcess.pidjs {rc|jjSr)rAr2r^s rr2zAsyncProcess.exitcodens {##rc|jSr)rKr^s rr=zAsyncProcess.namers zrc|jjSrr7rpr^s rrpzAsyncProcess.daemonvs }##rc||j_dSrr)rVrs rrpzAsyncProcess.daemonzs$ r)NNNr3N)r2rtrurv)rr6)rurr)rVr rrrurv)r-r.r/__doc____annotations__rXr_rbrUrx classmethodrrGrmrrqrrrrrr0propertyr1r2r=rpsetterr3rrr5r5Fs &%%%((((T;;;III...0////[>   [  ((([(TXX[X<            C C C C   ####"$$$X$$X$X$$X$ ]%%]%%%rr5c|rE td||dS#t$rYdSwxYwdS)Nzreaping stray process )r0r#inforOSError)procs rrNrNst }}  KK777 8 8 8 NN         DD  s1A AA)% __future__rrrloggingmultiprocessingr~rrkrLcollections.abcrqueuerrPtypingrtornado.concurrentrtornado.iolooprrHdistributed.utilsr r typing_extensionsr getLoggerr-r#rrr%r*r,r5rNr3rrrs"""""" $$$$$$"""""" %%%%%%!!!!!! 66666666'&&&&&&  8 $ $!!! KKKSSS v%v%v%v%v%v%v%v%r r