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� 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/python38/lib64/python3.8/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 s| | j �| �� �| _z| j �| j� W n tk
rD | j�d� Y nX tj dkrb| j �| j �
� � | j � � t� | j
� d S )Nz Exception in gunicorn worker)� � )r$ Z create_task�_runr Zrun_until_complete� Exception�logZ exception�sys� version_infoZshutdown_asyncgensr# �exitr r& r r r �run5 s
zGunicornWebWorker.runc
� s� t | jt�r| j}n,t�| j�r0| �� I d H }ntd�| j���| jjrP| j j
nd }t j || j | jj
|| �| jj�d�}|�� I d H | jjr�| �| j�nd }|}|d k s�t�|j}|d k s�t�| jD ]0}t j|||| jjd d d�}|�� I d H q�t�� }z�| j�r�| � � |j } | jj �rF| | jj k�rFd| _| j � d| � nB|t�� k�rz| j t�!� k�rzd| _| 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_log�access_log_format�d �_ )Z ssl_contextZshutdown_timeoutFz Max requests, shutting down: %sz!Parent changed, shutting down: %s)%�
isinstanceZwsgir
r! Ziscoroutinefunction� RuntimeError�format�cfgZ accesslogr+ r0 r
Z AppRunnerZ keepalive�_get_valid_log_formatr1 ZsetupZis_ssl�_create_ssl_context�AssertionError�serverZsocketsZSockSiteZgraceful_timeout�start�os�getpid�aliveZnotifyZrequests_countZ max_requests�infoZppid�getppid�_wait_next_notify�
BaseExceptionZcleanup)
r Zappr0 Zrunner�ctxr; ZsockZsite�pidZcntr r r r) B s\ �� �
� zGunicornWebWorker._runzasyncio.Future[bool]c C s> | � � | j}|d k st�|�� | _}| j�d| 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 � t�tjd� t�tj
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� siginterruptr& r r r � init_signals� sL � � � � � �z GunicornWebWorker.init_signals)�sig�framer 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�| � t�d� 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 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.)
�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 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 rY r r r r8 � s
�z'GunicornWebWorker._get_valid_log_format)N)�__name__�
__module__� __qualname__r Z
LOG_FORMATr[ �GunicornAccessLogFormat�defaultrZ 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_policy�EventLoopPolicyr r% )r rh r r r r% � s z$GunicornUVLoopWebWorker.init_process�r_ 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 � ImportError�object�__all__ZWorkerr r r r r r r �<module> s0
@