jGDdZddlZddlZddlZddlZGddZdS)zn Control Interface Command Handlers Provides handlers for all control commands with access to arbiter state. NceZdZdZdZdefdZdefdZdefdZdefdZ defdZ dd e defd Z dd e defd Z d e defdZdd e defdZdd e defdZded e defdZdefdZdefdZddedefdZdefdZdefdZdefdZdS)CommandHandlersz Command handlers with access to arbiter state. All handler methods return dictionaries that will be sent as the response data. c||_dS)z Initialize handlers with arbiter reference. Args: arbiter: The Gunicorn arbiter instance N)arbiter)selfrs H/var/www/html/venv/lib/python3.11/site-packages/gunicorn/ctl/handlers.py__init__zCommandHandlers.__init__s returnc g}tj}|jjD]v\}} |j}t||z d}n#ttf$rd}YnwxYw| ||j |j |j |dw|d|t|dS)a} Return list of HTTP workers. Returns: Dictionary with workers list containing: - pid: Worker process ID - age: Worker age (spawn order) - requests: Number of requests handled (if available) - booted: Whether worker has finished booting - last_heartbeat: Seconds since last heartbeat N)pidagebootedabortedlast_heartbeatc|dSNrws rz.CommandHandlers.show_workers..As 1U8r key)workerscount)time monotonicrWORKERSitemstmp last_updateroundOSError ValueErrorappendrrrsortlen)rrnowrworkerr"rs r show_workerszCommandHandlers.show_workers"sn</5577  KC &$j4466 !&s['8!!"0       ++ ,,,"S\\:::s,A''A=<A=c B|jjsddggdSt|jdd}g}g}|r_t|drNt j}|jD]\}} |j }t||z d}n#tttf$rd}YnwxYw|||jt|dgt|dd|d t|d r|jD]g\} } t#|j| g} || | d t)| | d hd |jj||dS)a% Return dirty workers and apps information. Returns: Dictionary with: - enabled: Whether dirty arbiter is running - pid: Dirty arbiter PID - workers: List of dirty worker info - apps: List of dirty app specs FN)enabledrrapps dirty_arbiterrr app_pathsr)rrr.rr app_specs worker_count) import_pathr2current_workers worker_pidsT)rdirty_arbiter_pidgetattrhasattrrrrr r!r"r#r$r%AttributeErrorr&rr1listapp_worker_mapgetr() rr/rr.r)rr*r"rpathspecr5s r show_dirtyzCommandHandlers.show_dirtyEs|-      otDD   W]I>> .""C,4::<<   V*"(*"8"8":":K%*3+>&4   }k22 "/"9"?"?"A"AJD$"&}'C'G'Gb'Q'Q"R"RKKK'+(,(@(@+.{+;+;'2 !!<1    s4,B!!B=<B=c |jj}i}gd}|D]} t||}t|rt |}nat |drQt |tttttttdfst |}|||<#t$rYwxYw|S)zz Return current effective configuration. Returns: Dictionary of configuration values )bindr worker_classthreadstimeoutgraceful_timeout keepalive max_requestsmax_requests_jitterworker_connections preload_appdaemonpidfile proc_namereload dirty_workers dirty_apps dirty_timeoutcontrol_socketcontrol_socket_disable __class__N)rcfgr7callablestrr8 isinstanceintfloatboolr:dicttyper9)rrUconfig config_keysrvalues r show_configzCommandHandlers.show_configsl     C S))E??'JJEEUK00'S%tT4::N:P:P'JJE#s !     sBB.. B;:B;c t|jdi}|d}d}|r$tt j|z d}||jjt |jj|jj|dd|dd|dd|jj pdd S) a Return server statistics. Returns: Dictionary with: - uptime: Seconds since arbiter started - pid: Arbiter PID - workers_current: Current number of workers - workers_spawned: Total workers spawned - workers_killed: Total workers killed (if tracked) - reloads: Number of reloads (if tracked) _stats start_timeNr workers_spawnedrworkers_killedreloads)uptimerworkers_currentworkers_targetrerfrgr6) r7rr<r#rrr(r num_workersr6)rstatsrdrhs r show_statszCommandHandlers.show_statss h33YY|,,   849;;3Q77F<#"4<#788"l6$yy):A>>#ii(8!<<yyA..!%!?!G4   r cg}|jjD]}t|}||d} |j}|jt jkrd|d<n5|jt jkrd|d<n|jt j krd|d<n#t$rd|d<YnwxYw| ||t|dS)zo Return bound socket information. Returns: Dictionary with listeners list )addressfdunixr]tcptcp6unknown) listenersr) r LISTENERSrWfilenosockfamilysocketAF_UNIXAF_INETAF_INET6 Exceptionr&r()rrulnraddr listener_inforxs rshow_listenerszCommandHandlers.show_listenerss <) , ,Cs88DjjllM 2x;&.00,2M&))[FN22,1M&))[FO33,2M&) 2 2 2(1 f%%% 2   ] + + + +&Y@@@sABB"!B"rctdt|}|jj}|jxj|z c_|j|||jjdS)z Increase worker count. Args: count: Number of workers to add (default 1) Returns: Dictionary with added count and new total r)addedprevioustotalmaxrYrrkwakeup)rr old_counts r worker_addzCommandHandlers.worker_addsnAs5zz""L,    E)   !\-   r ctdt|}|jj}td||z }||z }||j_|j|||dS)z Decrease worker count. Args: count: Number of workers to remove (default 1) Returns: Dictionary with removed count and new total r)removedrrr)rrr new_countactual_removeds r worker_removezCommandHandlers.worker_removes}As5zz""L, 9u,-- "Y.#,   &!   r rct|}||jjvr dd|ddS tj|t jd|dS#t$r}dt|dcYd}~Sd}~wwxYw)z Gracefully terminate a specific worker. Args: pid: Worker process ID Returns: Dictionary with killed PID or error FzWorker z not foundsuccesserrorT)rkilledN) rYrroskillsignalSIGTERMr$rW)rres r worker_killzCommandHandlers.worker_kills#hh dl* * * 23222   GC ( ( (     Q       s#A A2A-'A2-A2c|jjsdddStdt|}|d|S)a Spawn additional dirty workers. Sends a MANAGE message to the dirty arbiter to spawn workers. Args: count: Number of dirty workers to add (default 1) Returns: Dictionary with added count or error FDirty arbiter not runningrraddrr6rrY_send_manage_messagerrs r dirty_addzCommandHandlers.dirty_add4sS|-  4  As5zz""((666r c|jjsdddStdt|}|d|S)a Remove dirty workers. Sends a MANAGE message to the dirty arbiter to remove workers. Args: count: Number of dirty workers to remove (default 1) Returns: Dictionary with removed count or error Frrrremoverrs r dirty_removezCommandHandlers.dirty_removeIsS|-  4  As5zz""((5999r operationcd}t|jdr'|jjrt|jjdd}|stjd}|sdddS dd lm}m }m }|d kr|n|}tj tj tj }|d |||jd ||d } ||| ||} || d|jkr| dddiS| d|jkr<| di} d| dt-| dSdd| ddS#t.$r} dt-| dcYd} ~ Sd} ~ wwxYw)z Send a worker management message to the dirty arbiter. Args: operation: "add" or "remove" count: Number of workers to add/remove Returns: Dictionary with result or error Nr/ socket_pathGUNICORN_DIRTY_SOCKETFz%Cannot find dirty arbiter socket pathrr) DirtyProtocol MANAGE_OP_ADDMANAGE_OP_REMOVErg$@r)r]idoprr]resultrTrmessagezUnexpected response type: )r8rr/r7renvironr<gunicorn.dirty.protocolrrrrzr{ SOCK_STREAM settimeoutconnectMSG_TYPE_MANAGE write_message read_messagecloseMSG_TYPE_RESPONSEMSG_TYPE_ERRORrWr~) rrrdirty_socket_pathrrrrrxrequestresponserrs rrz$CommandHandlers._send_manage_message^sk! 4< 1 1 dl6P  ' *M4!! ! H " /F G G    @  *           #,u"4"4:JB=1CDDD OOD ! ! ! LL* + + +&5 G  ' 'g 6 6 6%11$77H JJLLL||F##}'FFF||Hy$.?@@@f%%)EEE Wb11$"YYy#e**== %P(,,v:N:NPP     Q       s,(C'GAG*G G+G& G+&G+c\tj|jjtjddiS)zv Trigger graceful reload (equivalent to SIGHUP). Returns: Dictionary with status status reloading)rrrrrSIGHUPrs rrNzCommandHandlers.reloads(   &-000+&&r c\tj|jjtjddiS)zp Reopen log files (equivalent to SIGUSR1). Returns: Dictionary with status r reopening)rrrrrSIGUSR1rs rreopenzCommandHandlers.reopens(   &.111+&&r gracefulmodec|dkr*tj|jjtjn)tj|jjtjd|dS)z Initiate shutdown. Args: mode: "graceful" (SIGTERM) or "quick" (SIGINT) Returns: Dictionary with status quick shutting_down)rr)rrrrrSIGINTr)rrs rshutdownzCommandHandlers.shutdownsP 7?? GDL$fm 4 4 4 4 GDL$fn 5 5 5)4888r c Htj}|jjddd}g}|jjD]q\}} |j}t||z d}n#ttf$rd}YnwxYw| |d|j |j |dr|d d}g} |jjr$|jjd d d}|} ||t#||| t#| d S) z Return overview of all processes (arbiter, web workers, dirty arbiter, dirty workers). Returns: Dictionary with complete process hierarchy rmaster)rr]roler Nweb)rr]rrrc|dSrrrs rrz*CommandHandlers.show_all..s qxr rr/z dirty master)r web_workersweb_worker_countr/rOdirty_worker_count)rrrrrr r!r"r#r$r%r&rrr'r6_query_dirty_workersr() rr) arbiter_inforrr*r"rdirty_arbiter_inforOs rshow_allzCommandHandlers.show_allsn<#   </5577  KC &$j4466 !&s['8!!z dirty add [N]zdirty remove [N]rNrzshutdown [graceful|quick]helpcommandsr)rrs rrzCommandHandlers.help1sKSA7A22;!?!G@ D-/)E,  "H%%r N)r)r)__name__ __module__ __qualname____doc__r r\r+r?rarmrrYrrrrrrWrrNrrrr:rrrr rrrs\!;d!;!;!;!;F: D: : : : x!T!!!!F D    >AAAAA@   D    .  3 t    8st<77s747777*::#:d::::*CcC#C$CCCCJ ' ' ' ' ''''''99S9$9999"9 $9 9 9 9 v+d++++Z&d&&&&&&r r)rrrrzrrrr rrsq   x&x&x&x&x&x&x&x&x&x&r