o iG@s6dZddlZddlZddlZddlZGdddZdS)zn Control Interface Command Handlers Provides handlers for all control commands with access to arbiter state. Nc@s0eZdZdZddZdefddZdefddZdefd d Zdefd d Z defd dZ d.de defddZ d.de defddZ de defddZd.de defddZd.de defddZdede defddZdefdd Zdefd!d"Zd/d$edefd%d&Zdefd'd(Zdefd)d*Zdefd+d,Zd-S)0CommandHandlersz Command handlers with access to arbiter state. All handler methods return dictionaries that will be sent as the response data. cCs ||_dS)z Initialize handlers with arbiter reference. Args: arbiter: The Gunicorn arbiter instance N)arbiter)selfrrI/home/ubuntu/.local/lib/python3.10/site-packages/gunicorn/ctl/handlers.py__init__s zCommandHandlers.__init__returnc Csg}t}|jjD].\}}z|j}t||d}Wn tt fy+d}Ynw| ||j |j |j |dq |jddd|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_heartbeatcS|dSNr rwrrrAz.CommandHandlers.show_workers..key)workerscount)time monotonicrWORKERSitemstmp last_updateroundOSError ValueErrorappendr r r sortlen)rrnowr workerrrrrr show_workers"s$    zCommandHandlers.show_workersc Cs|jjs ddggdSt|jdd}g}g}|rt|drt}|jD]5\}}z|j }t ||d}Wnt t t fyFd}Ynw|||jt|dgt|dd|d q&t|d r|jD]\} } t|j| g} || | d t| | d qfd |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)enabledr rapps dirty_arbiterrr app_pathsr )r r r)r r app_specs worker_count) import_pathr-current_workers worker_pidsT)rdirty_arbiter_pidgetattrhasattrrrrrrrrr r!AttributeErrorr"r r,listapp_worker_mapgetr$) rr*rr)r%r r&rrpathspecr0rrr show_dirtyEsN      zCommandHandlers.show_dirtyc Cs|jj}i}gd}|D]6}z+t||}t|rt|}nt|dr4t|tttt t t t dfs4t|}|||<Wq t yBYq w|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)rcfgr2callablestrr3 isinstanceintfloatboolr5dicttyper4)rrOconfig config_keysrvaluerrr show_configs"      zCommandHandlers.show_configc Csxt|jdi}|d}d}|rtt|d}||jjt|jj|jj|dd|dd|dd|jj p9dd 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)uptimer workers_currentworkers_targetr^r_r`r1) r2rr7rrr r$r num_workersr1)rstatsr]rarrr show_statss      zCommandHandlers.show_statsc Csg}|jjD]E}t|}||d}z%|j}|jtjkr"d|d<n|jtjkr-d|d<n |jtj kr7d|d<Wn t yEd|d<Ynw| |q|t |dS)zo Return bound socket information. Returns: Dictionary with listeners list )addressfdunixrWtcptcp6unknown) listenersr) r LISTENERSrQfilenosockfamilysocketAF_UNIXAF_INETAF_INET6 Exceptionr"r$)rrmlnraddr listener_inforprrrshow_listenerss(         zCommandHandlers.show_listenersrcCs@tdt|}|jj}|jj|7_|j|||jjdS)z Increase worker count. Args: count: Number of workers to add (default 1) Returns: Dictionary with added count and new total r{)addedprevioustotalmaxrSrrdwakeup)rr old_countrrr worker_adds  zCommandHandlers.worker_addcCsJtdt|}|jj}td||}||}||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{)removedr}r~r)rrr new_countactual_removedrrr worker_removes  zCommandHandlers.worker_remover c Cstt|}||jjvrdd|ddSz t|tjd|dWSty9}z dt|dWYd}~Sd}~ww)z Gracefully terminate a specific worker. Args: pid: Worker process ID Returns: Dictionary with killed PID or error FzWorker z not foundsuccesserrorT)rkilledN) rSrroskillsignalSIGTERMr rQ)rr errr worker_kills   zCommandHandlers.worker_killcC,|jjs dddStdt|}|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 runningrr{addrr1rrS_send_manage_messagerrrrr dirty_add4   zCommandHandlers.dirty_addcCr)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 Frrr{removerrrrr dirty_removeIrzCommandHandlers.dirty_remove operationc Cs^d}t|jdr|jjrt|jjdd}|stjd}|s#dddSzrdd lm}m }m }|d kr4|n|}t t j t j }|d |||jd ||d } ||| ||} || d|jkrp| dddiWS| d|jkr| di} d| dt| dWSdd| ddWSty} z dt| dWYd} ~ Sd} ~ ww)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{)rWidoprrWresultrTrmessagezUnexpected response type: )r3rr*r2renvironr7gunicorn.dirty.protocolrrrrrrs SOCK_STREAM settimeoutconnectMSG_TYPE_MANAGE write_message read_messagecloseMSG_TYPE_RESPONSEMSG_TYPE_ERRORrQrv) rrrdirty_socket_pathrrrrrprequestresponserrrrrr^sR        z$CommandHandlers._send_manage_messagecCt|jjtjddiS)zv Trigger graceful reload (equivalent to SIGHUP). Returns: Dictionary with status status reloading)rrrr rSIGHUPrrrrrHszCommandHandlers.reloadcCr)zp Reopen log files (equivalent to SIGUSR1). Returns: Dictionary with status r reopening)rrrr rSIGUSR1rrrrreopenszCommandHandlers.reopengracefulmodecCs8|dkrt|jjtjn t|jjtjd|dS)z Initiate shutdown. Args: mode: "graceful" (SIGTERM) or "quick" (SIGINT) Returns: Dictionary with status quick shutting_down)rr)rrrr rSIGINTr)rrrrrshutdowns  zCommandHandlers.shutdownc Cst}|jjddd}g}|jjD]-\}}z|j}t||d}Wn t t fy3d}Ynw| |d|j |j |dq|jdd 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)r rWroler Nweb)r rWr r rcSrrrrrrrrrz*CommandHandlers.show_all..rr*z dirty master)r web_workersweb_worker_countr*rIdirty_worker_count)rrrr rrrrrr r!r"r r r#r1_query_dirty_workersr$) rr% arbiter_inforr r&rrdirty_arbiter_inforIrrrshow_allsH   zCommandHandlers.show_allcCsd}t|jdr|jjrt|jjdd}|stjd}|s gSzEddlm}t t j t j }| d| ||jdd }|||||}||d |jkrc|d i}|d gWSWgStyoYgSw) z Query the dirty arbiter for worker information. Connects to the dirty arbiter socket and sends a status request. Returns: List of dirty worker info dicts, or empty list on error Nr*rrr)rg@z ctl-status-1)rWrrWrr)r3rr*r2rrr7rrrrrsrrrMSG_TYPE_STATUSrrrrrv)rrrrprrrrrrrs6         z$CommandHandlers._query_dirty_workerscCs,ddddddddd d d d d ddd}d|iS)z| Return list of available commands. Returns: Dictionary with commands and descriptions z8Show all processes (arbiter, web workers, dirty workers)z#List HTTP workers with their statuszList dirty workers and appsz$Show current effective configurationzShow server statisticszShow bound socketszSpawn N workers (default 1)zRemove N workers (default 1)z$Gracefully terminate specific workerz!Spawn N dirty workers (default 1)z"Remove N dirty workers (default 1)zGraceful reload (HUP)zReopen log files (USR1)zShutdown server (TERM/INT)zShow this help message)zshow allz show workersz show dirtyz show configz show statszshow listenerszworker add [N]zworker remove [N]zworker kill z dirty add [N]zdirty remove [N]rHrzshutdown [graceful|quick]helpcommandsr)rrrrrr1s"zCommandHandlers.helpN)r{)r)__name__ __module__ __qualname____doc__rrVr'r:r[rfrzrSrrrrrrQrrHrrrr5rrrrrrrs( #<# E  ;-r)rrrrrrrrrrrs