���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_result�GunicornWebWorker�GunicornUVLoopWebWorker�GunicornTokioWebWorkerc 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)�self�args�kw)� __class__� �
/worker.pyr
" s
zGunicornWebWorker.__init__c s@ t j � j � t j � | _ t j | j � t � j � d S)N)�asyncio�get_event_loop�closeZnew_event_loop�loopZset_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)N�loggerZkeepalive_timeout�
access_log�access_log_format�shutdown_asyncgens)�cfgZ accesslog�logr �dictZ keepalive�_get_valid_log_formatr r Z AppRunnerZwsgir r Zrun_until_completeZsetupZ create_task�_runr r � Exception�hasattrr r �sys�exitr )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_timeout�d �_ Fz Max requests, shutting down: %sz!Parent changed, shutting down: %s)r! Zis_ssl�_create_ssl_contextZsocketsr ZSockSiter Zgraceful_timeout�start�os�getpid�alive�notifyZserverZrequests_countZ max_requestsr" �infoZppid�getppid�_wait_next_notify�
BaseException�cleanup)r �ctx�sockZsite�pidZcntr 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_handler�signal�SIGQUIT� handle_quit�SIGTERMZ handle_exit�SIGINT�SIGWINCHZ handle_winch�SIGUSR1Z handle_usr1�SIGABRT� 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 �sig�framer 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 %\([^\)]+\)z�Gunicorn'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_FORMAT�DEFAULT_AIOHTTP_LOG_FORMAT�re�search�
ValueError)r Z
source_formatr r r r$ � s z'GunicornWebWorker._get_valid_log_format)�__name__�
__module__� __qualname__r Z
LOG_FORMATrM �GunicornAccessLogFormat�defaultrL 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_policy�EventLoopPolicyr 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(
�