U
'7`V @ s d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z m
Z
m Z m Z m
Z
ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ zddlZej Z W n e k
r dZe Z Y nX d Z G d
d dej Z!G dd de!Z"G dd de!Z#dS )z%Async gunicorn worker for aiohttp.web N) FrameType)Any AwaitableCallableOptionalUnion)AccessLogFormat)base)web )
set_result) Application) AccessLogger)GunicornWebWorkerGunicornUVLoopWebWorkerGunicornTokioWebWorkerc s e Zd ZejZejZe e dd fdd Z
dd fdd Z dddd Z ddd
d Z
d dd
dZd ed ddddZddddZeeddddZeeddddZee dddd Zeed d d Z ZS )!r N)argskwreturnc s$ t j|| d | _d| _d | _d S Nr )super__init___task exit_code_notify_waiter)selfr r __class__ A/opt/alt/python38/lib64/python3.8/site-packages/aiohttp/worker.pyr % s zGunicornWebWorker.__init__r c s0 t t | _t | j t d S )N)asyncioget_event_loopcloseZnew_event_looploopZset_event_loopr init_processr r r r r% , s
z GunicornWebWorker.init_processc C s| | j | | _z| j | j W n tk
rD | jd Y nX tj dkrb| j | j
| j t | j
d S )Nz Exception in gunicorn worker) )r$ Z create_task_runr Zrun_until_complete ExceptionlogZ exceptionsys version_infoZshutdown_asyncgensr# exitr r& r r r run5 s
zGunicornWebWorker.runc
s t | jtr| j}n,t| jr0| I d H }ntd| j| jjrP| j j
nd }t j || j | jj
|| | jjd}| I d H | jjr| | jnd }|}|d k st|j}|d k st| jD ]0}t j|||| jjd d d}| I d H qt }z| jr| |j } | jj rF| | jj krFd| _| j d| nB|t krz| j t! krzd| _| j d| n| " I d H qW n t#k
r Y nX |$ I d H d S ) NzUwsgi app should be either Application or async function returning Application, got {})ZloggerZkeepalive_timeout
access_logaccess_log_formatd _ )Z ssl_contextZshutdown_timeoutFz Max requests, shutting down: %sz!Parent changed, shutting down: %s)%
isinstanceZwsgir
r! Ziscoroutinefunction RuntimeErrorformatcfgZ accesslogr+ r0 r
Z AppRunnerZ keepalive_get_valid_log_formatr1 ZsetupZis_ssl_create_ssl_contextAssertionErrorserverZsocketsZSockSiteZgraceful_timeoutstartosgetpidaliveZnotifyZrequests_countZ max_requestsinfoZppidgetppid_wait_next_notify
BaseExceptionZcleanup)
r Zappr0 Zrunnerctxr; ZsockZsitepidZcntr r r r) B s\
zGunicornWebWorker._runzasyncio.Future[bool]c C s> | | j}|d k st| | _}| jd| j | |S )Ng ?)_notify_waiter_doner$ r: Z
create_futurer Z
call_later)r r$ waiterr r r rB z s z#GunicornWebWorker._wait_next_notify)rG r c C s4 |d kr| j }|d k r t|d || j kr0d | _ d S )NT)r r )r rG r r r rF s
z%GunicornWebWorker._notify_waiter_donec C s | j tj| jtjd | j tj| jtjd | j tj| jtjd | j tj| j tjd | j tj
| j tj
d | j tj | j
tj d ttjd ttj
d d S NF)r$ Zadd_signal_handlersignalSIGQUIT handle_quitSIGTERMZ handle_exitSIGINTSIGWINCHZ handle_winchSIGUSR1Z handle_usr1SIGABRT handle_abort siginterruptr& r r r init_signals sL z GunicornWebWorker.init_signals)sigframer c C s d| _ | j| | d S rH )r? r7 Z
worker_intrF r rT rU r r r rK s z GunicornWebWorker.handle_quitc C s& d| _ d| _| j| td d S )NFr )r? r r7 Z worker_abortr, r. rV r r r rQ s z GunicornWebWorker.handle_abort
SSLContext)r7 r c C s\ t dkrtdt | j}|| j| j | j|_| j rF|
| j | j rX| | j |S )z~Creates SSLContext instance for usage in asyncio.create_server.
See ssl.SSLSocket.__init__ for more details.
NzSSL is not supported.)
sslr5 rW Z ssl_versionZload_cert_chainZcertfileZkeyfileZ cert_reqsZ verify_modeZca_certsZload_verify_locationsZciphersZ set_ciphers)r7 rD r r r r9 s z%GunicornWebWorker._create_ssl_context)
source_formatr c C s. || j kr| jS td|r&tdn|S d S )Nz %\([^\)]+\)zGunicorn's style options in form of `%(name)s` are not supported for the log formatting. Please use aiohttp's format specification to configure access log formatting: http://docs.aiohttp.org/en/stable/logging.html#format-specification)DEFAULT_GUNICORN_LOG_FORMATDEFAULT_AIOHTTP_LOG_FORMATresearch
ValueError)r rY r r r r8 s
z'GunicornWebWorker._get_valid_log_format)N)__name__
__module__ __qualname__r Z
LOG_FORMATr[ GunicornAccessLogFormatdefaultrZ r r r% r/ r) rB r rF rS intr rK rQ staticmethodr9 strr8
__classcell__r r r r r s$
8 r c s" e Zd Zdd fdd Z ZS )r Nr c s0 dd l }t t| t d S r )uvloopr! r" r# set_event_loop_policyEventLoopPolicyr r% )r rh r r r r% s z$GunicornUVLoopWebWorker.init_processr_ r` ra r% rg r r r r r s r c s" e Zd Zdd fdd Z ZS )r Nr c s0 dd l }t t| t d S r )tokior! r" r# ri rj r r% )r rl r r r r% s z#GunicornTokioWebWorker.init_processrk r r r r r s r )$__doc__r! r= r\ rI r, typesr typingr r r r r Zgunicorn.configr rb Zgunicorn.workersr Zaiohttpr
Zhelpersr Zweb_appr
Zweb_logr rX rW ImportErrorobject__all__ZWorkerr r r r r r r <module> s0
@