Current File : //opt/alt/python37/lib64/python3.7/site-packages/aiohttp/__pycache__/worker.cpython-37.pyc
B
�Z�_V � @ 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 s| | j �| �� �| _y| 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�x:| jD ]0}t j|||| jjd d d�}|�� I d H q�W t�� }y�x�| j�r�| � � |j } | jj �rL| | jj k�rLd| _| j � d| � nB|t�� k�r�| j t�!� k�r�d| _| j � d| � n| �"� I d H �qW W 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' r, r
Z AppRunnerZ keepalive�_get_valid_log_formatr- ZsetupZis_ssl�_create_ssl_context�AssertionError�serverZsocketsZSockSiteZgraceful_timeout�start�os�getpid�aliveZnotifyZrequests_countZ max_requests�infoZppid�getppid�_wait_next_notify�
BaseExceptionZcleanup)
r Zappr, Zrunner�ctxr7 ZsockZsite�pidZcntr r r r% B sT
zGunicornWebWorker._runzasyncio.Future[bool]c C s> | � � | j}|d k st�|�� | _}| j�d| j |� |S )Ng �?)�_notify_waiter_doner! r6 Z
create_futurer Z
call_later)r r! �waiterr r r r> z s z#GunicornWebWorker._wait_next_notify)rC r c C s4 |d kr| j }|d k r t|d� || j kr0d | _ d S )NT)r r )r rC r r r rB � 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� siginterrupt)r r r r � init_signals� s z GunicornWebWorker.init_signals)�sig�framer c C s d| _ | j�| � | �� d S )NF)r; r3 Z
worker_intrB )r rO rP r r r rF � s z GunicornWebWorker.handle_quitc C s& d| _ d| _| j�| � t�d� d S )NFr )r; r r3 Z worker_abortr( r* )r rO rP r r r rL � s z GunicornWebWorker.handle_abort�
SSLContext)r3 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.)
�sslr1 rQ Z ssl_versionZload_cert_chainZcertfileZkeyfileZ cert_reqsZ verify_modeZca_certsZload_verify_locationsZciphersZ set_ciphers)r3 r@ r r r r5 � 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 rS r r r r4 � s
z'GunicornWebWorker._get_valid_log_format)N)�__name__�
__module__� __qualname__r Z
LOG_FORMATrU �GunicornAccessLogFormat�defaultrT r r r" r+ r% r> r rB rN �intr rF rL � staticmethodr5 �strr4 �
__classcell__r r )r r r s
8 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 rb )r r r r" � s z$GunicornUVLoopWebWorker.init_process)rY rZ r[ r" ra 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 rc rd r r" )r re )r r r r" � s z#GunicornTokioWebWorker.init_process)rY rZ r[ r" ra r r )r r r � s r )$�__doc__r r9 rV rD r( �typesr �typingr r r r r Zgunicorn.configr r\ Zgunicorn.workersr Zaiohttpr
Zhelpersr Zweb_appr
Zweb_logr rR rQ � ImportError�object�__all__ZWorkerr r r r r r r �<module> s0
@
Mini Shell