o NK&h[@sjddlmZddlmZmZddlmZmZmZddl m Z m Z ddl m Z ddlmZGdddZd S) ) Connection)environgetpid)AnyCallableOptional)Colorslogger) ProcessState) WorkerStatec@s&eZdZdZdedeeeffddZddZ   d1d ed e defdeeefde de e de de de dd fddZde dd fddZddZ   d2dede d e fd!d"ZeZ d#e fd$d%Zd3d&e fd'd(Zede fd)d*Zedefd+d,Zed-d.Zedeeeffd/d0Zd S)4WorkerMultiplexera)Multiplexer for Sanic workers. This is instantiated inside of worker porocesses only. It is used to communicate with the monitor process. Args: monitor_publisher (Connection): The connection to the monitor. worker_state (Dict[str, Any]): The state of the worker. monitor_publisher worker_statecCs||_t||j|_dS)N)_monitor_publisherr name_state)selfr rrK/var/www/html/venv/lib/python3.10/site-packages/sanic/worker/multiplexer.py__init__szWorkerMultiplexer.__init__c Csbttjdtjtjdtjdtj|j|ji|j j |jdt j ji|j j |j<dS)z Acknowledge the worker is ready.z Process ack: z%s z[%s]stateN) r debugrBLUEBOLDSANICENDrpidrr ACKEDrrrracks zWorkerMultiplexer.ackFNTidentfunc.kwargs transient restartabletracked auto_startworkersreturnc Cs$||||||||f} |j| dS)aManages the initiation and monitoring of a worker process. Args: ident (str): A unique identifier for the worker process. func (Callable[..., Any]): The function to be executed in the worker process. kwargs (Dict[str, Any]): A dictionary of arguments to be passed to `func`. transient (bool, optional): Flag to mark the process as transient. If `True`, the Worker Manager will restart the process with any global restart (e.g., auto-reload). Defaults to `False`. restartable (Optional[bool], optional): Flag to mark the process as restartable. If `True`, the Worker Manager can restart the process if prompted. Defaults to `None`. tracked (bool, optional): Flag to indicate whether the process should be tracked after its completion. Defaults to `False`. auto_start (bool, optional): Flag to indicate whether the process should be started workers (int, optional): The number of worker processes to run. Defaults to 1. This method packages the provided arguments into a bundle and sends them back to the main process to be managed by the Worker Manager. Nrsend) rr!r"r#r$r%r&r'r(bundlerrrmanage*s zWorkerMultiplexer.manageservingcCs(i|jj|jd|i|jj|j<dS)zmSet the worker to serving. Args: serving (bool): Whether the worker is serving. r.N)rr)rr.rrr set_servingTs  zWorkerMultiplexer.set_servingcCs2z |jj|j=WdStytdYdSw)zRun cleanup at worker exit.z#Monitor process has already exited.N)rrConnectionRefusedErrorr rrrrrexit_s  zWorkerMultiplexer.exitr all_workers zero_downtimecCsP|r|rtd|s|rdn|j}|ds|d7}|r |d7}|j|dS)zRestart the worker. Args: name (str): The name of the process to restart. all_workers (bool): Whether to restart all workers. zero_downtime (bool): Whether to restart with zero downtime. z@Ambiguous restart with both a named process and all_workers=Truez__ALL_PROCESSES__::z:STARTUP_FIRSTN) ValueErrorrendswithrr+)rrr3r4rrrrestartfs  zWorkerMultiplexer.restart num_workerscCsd|}|j|dS)zvScale the number of workers. Args: num_workers (int): The number of workers to scale to. z __SCALE__:Nr*)rr9messagerrrscales zWorkerMultiplexer.scaleearlycCs|rdnd}|j|dS)zcTerminate the worker. Args: early (bool): Whether to terminate early. __TERMINATE_EARLY__ __TERMINATE__Nr*)rr<r:rrr terminates zWorkerMultiplexer.terminatecCstS)zThe process ID of the worker.)rrrrrrzWorkerMultiplexer.pidcCs tddS)zThe name of the worker.SANIC_WORKER_NAMEr2)rgetrrrrrs zWorkerMultiplexer.namecCs|jS)zThe state of the worker.)rrrrrrr@zWorkerMultiplexer.statecCs |jS)zThe state of all workers.)rfullrrrrr(s zWorkerMultiplexer.workers)FNFTr )r2FF)F)__name__ __module__ __qualname____doc__rdictstrrrrrboolrintr-r/r1r8reloadr;r?propertyrrrr(rrrrr sp       *      r N)multiprocessing.connectionrosrrtypingrrr sanic.logrr sanic.worker.processr sanic.worker.stater r rrrrs