File: //opt/alt/python37/lib64/python3.7/site-packages/aiohttp/__pycache__/worker.cpython-37.pyc
B
�]����� @ s d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z m
Z
m Z m Z m
Z
ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ yddlZej 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� Awaitable�Callable�Optional�Union)�AccessLogFormat)�base)�web� )�
set_result)� Application)� AccessLogger)�GunicornWebWorker�GunicornUVLoopWebWorker�GunicornTokioWebWorkerc s� e Zd ZejZejZe e dd�� fdd� Z
dd�� fdd� Z dd�dd �Z dd�d
d �Z
d d�d
d�Zd ed dd�dd�Zdd�dd�Zeedd�dd�Zeedd�dd�Zee dd�dd ��Zeed �d d �Z� ZS )!r N)�args�kw�returnc s$ t � j||� d | _d| _d | _d S )Nr )�super�__init__�_task� exit_code�_notify_waiter)�selfr r )� __class__� �A/opt/alt/python37/lib64/python3.7/site-packages/aiohttp/worker.pyr % s zGunicornWebWorker.__init__)r c s0 t �� �� t �� | _t �| j� t� �� 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 sr | j �| �� �| _y| j �| j� W n tk
rD | j�d� Y nX | j �| j �� � | j � � t
� | j � d S )Nz Exception in gunicorn worker)
r! Z create_task�_runr Zrun_until_complete� Exception�log� exceptionZshutdown_asyncgensr �sys�exitr )r r r r �run5 s
zGunicornWebWorker.runc � s� d }t | jt�r| j}nHt�| j�rP| �� I d H }t |tj�rJ|}|j}q`|}ntd� | j���|d kr�| j
j rx| j j
nd }tj|| j | j
j|| �| j
j�d�}|�� I d H | j
jr�| �| j
�nd }|}|d k s�t�|j}|d k s�t�x:| jD ]0}tj|||| j
jd d d�}|�� I d H q�W t�� } y�x�| j �r�| � � |j }
| j
j �rt|
| j
j k�rtd| _ | j � d| � nB| t�� k�r�| j!t�"� k�r�d| _ | j � d| � n| �#� I d H �q0W W n t$k
�r� Y nX |�%� I d H d S ) NzUwsgi app should be either Application or async function returning Application, got {})�loggerZkeepalive_timeout�
access_log�access_log_format�d �_ )Z ssl_contextZshutdown_timeoutFz Max requests, shutting down: %sz!Parent changed, shutting down: %s)&�
isinstance�wsgir
r �iscoroutinefunctionr
Z AppRunner�app� RuntimeError�format�cfgZ accesslogr% r+ Z keepalive�_get_valid_log_formatr, �setupZis_ssl�_create_ssl_context�AssertionError�serverZsocketsZSockSiteZgraceful_timeout�start�os�getpid�alive�notifyZrequests_countZ max_requests�infoZppid�getppid�_wait_next_notify�
BaseException�cleanup) r Zrunnerr2 r0 r+ �ctxr: �sock�site�pidZcntr r r r# A sb
zGunicornWebWorker._runzasyncio.Future[bool]c C s> | � � | j}|d k st�|�� | _}| j�d| j |� |S )Ng �?)�_notify_waiter_doner! r9 Z
create_futurer Z
call_later)r r! �waiterr r r rB � s z#GunicornWebWorker._wait_next_notify)rJ r c C s4 |d kr| j }|d k r t|d� || j kr0d | _ d S )NT)r r )r rJ r r r rI � 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 � t�tjd� t�tj
d� tjdk r�t�tjtj� d S )NF)� � )r! Zadd_signal_handler�signal�SIGQUIT� handle_quit�SIGTERMZ handle_exit�SIGINT�SIGWINCHZ handle_winch�SIGUSR1Z handle_usr1�SIGABRT� handle_abort� siginterruptr' � version_info�SIGCHLD�SIG_DFL)r r r r � init_signals� s
z GunicornWebWorker.init_signals)�sig�framer c C s d| _ | j�| � | �� d S )NF)r> r5 Z
worker_intrI )r r[ r\ r r r rO � s z GunicornWebWorker.handle_quitc C s& d| _ d| _| j�| � t�d� d S )NFr )r> r r5 Z worker_abortr' r( )r r[ r\ r r r rU � s z GunicornWebWorker.handle_abort�
SSLContext)r5 r c C s\ t dkrtd��t �| 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.)
�sslr3 r] Z ssl_versionZload_cert_chainZcertfileZkeyfileZ cert_reqsZ verify_modeZca_certsZload_verify_locationsZciphersZ set_ciphers)r5 rE r r r r8 � s z%GunicornWebWorker._create_ssl_context)�
source_formatr c C s. || j kr| jS t�d|�r&td��n|S d 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 r_ r r r r6 � s
z'GunicornWebWorker._get_valid_log_format)N)�__name__�
__module__� __qualname__r Z
LOG_FORMATra �GunicornAccessLogFormat�defaultr` r r r" r) r# rB r rI rZ �intr rO rU � staticmethodr8 �strr6 �
__classcell__r r )r r r s B ( r c s" e Zd Zdd�� fdd� Z� ZS )r N)r c s0 dd l }t�� �� t�|�� � t� �� d S )Nr )�uvloopr r r �set_event_loop_policy�EventLoopPolicyr r" )r rn )r r r r" � s z$GunicornUVLoopWebWorker.init_process)re rf rg r" rm r r )r r r � s r c s" e Zd Zdd�� fdd� Z� ZS )r N)r c s0 dd l }t�� �� t�|�� � t� �� d S )Nr )�tokior r r ro rp r r" )r rq )r r r r" s z#GunicornTokioWebWorker.init_process)re rf rg r" rm r r )r r r s r )$�__doc__r r< rb rM r' �typesr �typingr r r r r Zgunicorn.configr rh Zgunicorn.workersr Zaiohttpr
Zhelpersr Zweb_appr
Zweb_logr r^ r] � ImportError�object�__all__ZWorkerr r r r r r r �<module> s0
Q