Z @ s d Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z d d l m Z
d d l m Z d d l
m Z d d l m Z m Z y d d l Z Wn e k
r d Z Yn Xd Z Gd d d e j Z Gd
d
d
e Z Gd d d e Z d S)z%Async gunicorn worker for aiohttp.web N)suppress)AccessLogFormat)base)web ) AccessLogger
set_resultGunicornWebWorkerGunicornUVLoopWebWorkerGunicornTokioWebWorkerc s e Z d Z e j Z e j Z f d d Z f d d Z
d d Z d d Z d d
Z
d d d
Z d d Z d d Z d d Z e d d Z d d Z S)r c s; t j | | d | _ d | _ d | _ d | _ d S)Nr )super__init___runner_task exit_code_notify_waiter)selfargskw) __class__
/worker.pyr
" s
zGunicornWebWorker.__init__c s@ t j j t j | _ t j | j t j d S)N)asyncioget_event_loopcloseZnew_event_looploopZset_event_loopr init_process)r )r r r r * s z GunicornWebWorker.init_processc
C s | j j r | j j n d } t d | j d | j j d | d | j | j j } t j | j
| | _ | j j
| j j | j j | j | _ t t | j j
| j Wd QRXt | j d r | j j
| j j | j j t j | j d S)NloggerZkeepalive_timeout
access_logaccess_log_formatshutdown_asyncgens)cfgZ accessloglogr dictZ keepalive_get_valid_log_formatr r Z AppRunnerZwsgir r Zrun_until_completeZsetupZ create_task_runr r Exceptionhasattrr r sysexitr )r r Zparamsr r r run3 s
zGunicornWebWorker.runc sl | j j r | j | j n d } xO | j D]D } t j | j | d | d | j j d d } | j Id Hq. Wt j
} y x | j r?| j | j j
j } | j j r | | j j k r d | _ | j j d | q | t j
k r-| j t j k r-d | _ | j j d | q | j Id Hq WWn t k
rUYn X| j j Id Hd S)NZ ssl_contextZshutdown_timeoutd _ Fz Max requests, shutting down: %sz!Parent changed, shutting down: %s)r! Zis_ssl_create_ssl_contextZsocketsr ZSockSiter Zgraceful_timeoutstartosgetpidalivenotifyZserverZrequests_countZ max_requestsr" infoZppidgetppid_wait_next_notify
BaseExceptioncleanup)r ctxsockZsitepidZcntr r r r% G s* $
'
zGunicornWebWorker._runc C s= | j | j j | _ } | j j d | j | | S)Ng ?)_notify_waiter_doner Z
create_futurer Z
call_later)r waiterr r r r5 e s
z#GunicornWebWorker._wait_next_notifyNc C sJ | d k r | j } | d k r. t | d | | j k rF d | _ d S)NT)r r )r r< r r r r; m s
z%GunicornWebWorker._notify_waiter_donec C s | j j t j | j t j d | j j t j | j t j d | j j t j | j t j d | j j t j | j t j d | j j t j
| j t j
d | j j t j | j
t j d t j t j d t j t j
d d S)NF)r Zadd_signal_handlersignalSIGQUIT handle_quitSIGTERMZ handle_exitSIGINTSIGWINCHZ handle_winchSIGUSR1Z handle_usr1SIGABRT handle_abort siginterrupt)r r r r init_signalsv s
z GunicornWebWorker.init_signalsc C s' d | _ | j j | | j d S)NF)r1 r! Z
worker_intr; )r sigframer r r r? s z GunicornWebWorker.handle_quitc C s3 d | _ d | _ | j j | t j d d S)NFr )r1 r r! Z worker_abortr( r) )r rH rI r r r rE s z GunicornWebWorker.handle_abortc C s t d k r t d t j | j } | j | j | j | j | _ | j re | j
| j | j r~ | j | j | S)z Creates SSLContext instance for usage in asyncio.create_server.
See ssl.SSLSocket.__init__ for more details.
NzSSL is not supported.)
ssl RuntimeErrorZ
SSLContextZ ssl_versionZload_cert_chainZcertfileZkeyfileZ cert_reqsZ verify_modeZca_certsZload_verify_locationsZciphersZ set_ciphers)r! r8 r r r r- s z%GunicornWebWorker._create_ssl_contextc C s? | | j k r | j St j d | r7 t d n | Sd 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 Z
source_formatr r r r$ s z'GunicornWebWorker._get_valid_log_format)__name__
__module__ __qualname__r Z
LOG_FORMATrM GunicornAccessLogFormatdefaultrL r
r r* r% r5 r; rG r? rE staticmethodr- r$ r r )r r r s c s" e Z d Z f d d Z S)r
c s@ d d l } t j j t j | j t j d S)Nr )uvloopr r r set_event_loop_policyEventLoopPolicyr r )r rW )r r r r s z$GunicornUVLoopWebWorker.init_process)rQ rR rS r r r )r r r
s c s" e Z d Z f d d Z S)r c s@ d d l } t j j t j | j t j d S)Nr )tokior r r rX rY r r )r rZ )r r r r s z#GunicornTokioWebWorker.init_process)rQ rR rS r r r )r r r s )r r
r )__doc__r r/ rN r= r(
contextlibr Zgunicorn.configr rT Zgunicorn.workersr Zaiohttpr Zhelpersr r rJ ImportError__all__ZWorkerr r
r r r r r <module> s(