:jddZddlZddlZddlmZmZddlmZddlm Z edZ Gdd Z y) z| Shutdown coordination for graceful application termination. Manages component shutdown sequence and completion monitoring. N)ListOptional) get_logger)LifecycleManagermanagerceZdZdZ ddeededefdZddeeddfd Z dd Z dd Z de fd Z dd eede fdZddZde fdZy)ShutdownCoordinatorz7Coordinates graceful shutdown of application components componentsshutdown_timeoutmonitor_intervalc||_td||_td||_t j |_d|_t j |_y)zInitialize shutdown coordinator Args: components: List of components to shutdown shutdown_timeout: Total timeout for shutdown process monitor_interval: Interval between completion checks in seconds g?rN) r maxr r threadingEventshutdown_eventcompletion_threadcompletion_stop)selfr r r s 7/root/.openclaw/workspace/harvester/manager/shutdown.py__init__zShutdownCoordinator.__init__sQ% #C)9 : #A'7 8'oo/=A(0Nintervalreturnc|jr0|jjrtjdy|%|dkDr ||_tj d|d|j jtj|jdd|_|jjtj d |jdy) zStart completion monitoring thread Args: interval: Override monitor interval for this session (optional) z"Completion monitor already runningNrzUsing custom monitor interval: szcompletion-monitorF)targetnamedaemonz0Started completion monitoring thread, interval: ) ris_aliveloggerwarningr inforclearrThread_completion_monitor_loopstart)rrs rstart_completion_monitorz,ShutdownCoordinator.start_completion_monitor*s  ! !d&<&<&E&E&G NN? @   HqL$,D ! KK9(1E F ""$!*!1!100%"  $$& FtG\G\F]]^_`rc|jr|jjsytjd|jj  |jj d|jjrtjdytjdy#t$r"}tjd|Yd}~yd}~wwxYw)z!Stop completion monitoring threadNz%Stopping completion monitoring threadg@timeoutz1Completion monitor thread did not stop gracefullyz!Completion monitor thread stoppedz#Error stopping completion monitor: ) rr r!r#rsetjoinr" Exceptionerror)res rstop_completion_monitorz+ShutdownCoordinator.stop_completion_monitorBs%%T-C-C-L-L-N  ;<   " D  " " ' ' ' 4%%..0RS ?@ D LL>qcB C C DsA B:$B:: C%C  C%c |jjsd}|jD]}|jrd}n|r0tj d|j jy|jj|jry|jjsyy#t$r"}tjd|Yd}~yd}~wwxYw)z8Monitor completion status and trigger shutdown when doneTFz,All components finished, triggering shutdownr*zError in completion monitor: N) ris_setr is_finishedr!r#rr,waitr r.r/)r all_finished componentr0s rr&z,ShutdownCoordinator._completion_monitor_loopSs >**113# !%I$002',   KK NO''++-'',,T5J5J,K**113" > LL8< = = >s(;B96B95&B9B99 C$CC$c  tjdtj}|j|jt t |jdz }d}t|jD]\}}|jj} tjd|dzdt |jd||jtj}tj|z |kr=|jsnKtjdtj|z |kr=tjd|d d }tjd|d tj|z } |rtjd| dd|Stjd| dd|S#t$r(}tj!d |d|d }Yd }~d }~wwxYw)zPerform graceful shutdown of all components Returns: bool: True if shutdown completed successfully zStarting graceful shutdownrTzStopping component /z: g?z Component z did not stop within timeoutFz stoppedzError stopping component Nz,Graceful shutdown completed successfully in z.1frz+Graceful shutdown completed with errors in )r!r#timer1r rlenr enumerate __class____name__stop is_runningsleepr"r.r/) r start_timecomponent_timeoutsuccessir7component_name timeout_startr0elapseds rgraceful_shutdownz%ShutdownCoordinator.graceful_shutdownjs  01YY[  $$&!11CDOO8La4PP%doo6 LAy&0099N  1!A#aDOO8L7MRP^O_`a !% iikM14EE$//JJsO iikM14EE NNZ/??[#\]#G j(8AB% 0))+ *  KKFwsmSTU V NNHQT UVW X  88H1#NO sB.G 4G HG>>Hr+c:|jj|S)zWait for completion signal Args: timeout: Maximum time to wait Returns: bool: True if completion was signaled r*)rr5)rr+s rwait_for_completionz'ShutdownCoordinator.wait_for_completions""'''88rcbtjd|jjy)zSignal shutdown requestzShutdown signal receivedN)r!r#rr,rs rsignal_shutdownz#ShutdownCoordinator.signal_shutdowns! ./ !rc6|jjS)zpCheck if shutdown has been requested Returns: bool: True if shutdown was requested )rr3rMs ris_shutdown_requestedz)ShutdownCoordinator.is_shutdown_requesteds ""))++r)g>@g@)N)rN)r> __module__ __qualname____doc__rrfloatrrr(r1r&boolrIrKrNrPrrr r sA #'"% 1)*1 1 1(a%aDa0D">..4.` 98E? 9d 9" ,t,rr ) rSrr:typingrr tools.loggerrbaserr!r rVrrrZs1  !#" I ],],r