File: //usr/local/lib/python3.10/dist-packages/aiohttp/__pycache__/worker.cpython-310.pyc
o
���gh � @ s d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z m
Z
m Z ddl m
Z ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ e rTddlZejZnz ddlZejZW n
eyj dZe ZY nw d Z G d
d� dej �Z G dd� de �Z dS )z%Async gunicorn worker for aiohttp.web� N)� FrameType)�
TYPE_CHECKING�Any�Optional)�AccessLogFormat)�base)�web� )�
set_result)� Application)� AccessLogger)�GunicornWebWorker�GunicornUVLoopWebWorkerc s� e Zd ZejZejZde de ddf� fdd� Z
d"� fdd� Z d"d d
�Z d"d d �Z
d#dd�Z d$ded
ddfdd�Zd"dd�Zdedee ddfdd�Zdedee ddfdd�Zede dd fd d ��Zd edefd d!�Z� ZS )%r
�args�kw�returnNc s( t � j|i |�� d | _d| _d | _d S �Nr )�super�__init__�_task� exit_code�_notify_waiter)�selfr r �� __class__� �9/usr/local/lib/python3.10/dist-packages/aiohttp/worker.pyr * s
zGunicornWebWorker.__init__c s0 t �� �� t �� | _t �| j� t� �� d S �N)�asyncio�get_event_loop�close�new_event_loop�loop�set_event_loopr � init_process�r r r r r$ 1 s
z GunicornWebWorker.init_processc C sp | j �| �� �| _z | j �| j� W n ty! | j�d� Y nw | j �| j �� � | j � � t
� | j � d S )Nz Exception in gunicorn worker)
r" � create_task�_runr �run_until_complete� Exception�log� exception�shutdown_asyncgensr �sys�exitr r% r r r �run: s �
zGunicornWebWorker.runc � s� �d }t | jt�r
| j}n$t�| j�r)| �� I d H }t |tj�r&|}|j}n |}ntd� | j���|d u rX| j
j r=| j j
nd }tj|| j | j
j|| �| j
j�| j
jd d d�}|�� I d H | j
jri| �| j
�nd }|}|d ussJ �|j}|d us|J �| jD ]}tj|||d�}|�� I d H qt�� } zA| jr�| � � |j }
| j r�|
| j kr�d| _| j � d| � n | t�� kr�| j t�!� kr�d| _| j � d| � n| �"� I d H | js�W n t#y� Y nw |�$� I d H d S ) NzUwsgi app should be either Application or async function returning Application, got {}�d �_ )�logger�keepalive_timeout�
access_log�access_log_format�shutdown_timeout)� ssl_contextFz Max requests, shutting down: %sz!Parent changed, shutting down: %s)%�
isinstance�wsgir r �iscoroutinefunctionr � AppRunner�app� RuntimeError�format�cfg� accesslogr* r4 � keepalive�_get_valid_log_formatr5 �graceful_timeout�setup�is_ssl�_create_ssl_context�server�sockets�SockSite�start�os�getpid�alive�notify�requests_count� max_requests�info�ppid�getppid�_wait_next_notify�
BaseException�cleanup) r �runnerr< r9 r4 �ctxrG �sock�site�pid�cntr r r r' F sr � ���
���
�zGunicornWebWorker._run�asyncio.Future[bool]c C s> | � � | j}|d us
J �|�� | _}| j�d| j |� |S )Ng �?)�_notify_waiter_doner" �
create_futurer �
call_later)r r" �waiterr r r rT � s z#GunicornWebWorker._wait_next_notifyra c C s8 |d u r| j }|d urt|d� || j u rd | _ d S d S )NT)r r
)r ra r r r r^ � 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" �add_signal_handler�signal�SIGQUIT� handle_quit�SIGTERM� handle_exit�SIGINT�SIGWINCH� handle_winch�SIGUSR1� handle_usr1�SIGABRT� handle_abort� siginterruptr% r r r � init_signals� s( ������z GunicornWebWorker.init_signals�sig�framec C s d| _ | j�| � | �� d S rb )rM r? �
worker_intr^ �r rr rs r r r rf � s z GunicornWebWorker.handle_quitc C s& d| _ d| _| j�| � t�d� d S )NFr )rM r r? � worker_abortr- r. ru r r r ro � s z GunicornWebWorker.handle_abortr? �
SSLContextc C s\ t du rtd��t �| j�}|�| j| j� | j|_| j r#|�
| j � | j r,|� | j � |S )z~Creates SSLContext instance for usage in asyncio.create_server.
See ssl.SSLSocket.__init__ for more details.
NzSSL is not supported.)
�sslr= rw � ssl_version�load_cert_chain�certfile�keyfile� cert_reqs� verify_mode�ca_certs�load_verify_locations�ciphers� set_ciphers)r? rX r r r rF � s z%GunicornWebWorker._create_ssl_context�
source_formatc C s( || j kr| jS t�d|�rtd��|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 rB � s
�z'GunicornWebWorker._get_valid_log_format�r N)r r] r )�__name__�
__module__� __qualname__r �
LOG_FORMATr� �GunicornAccessLogFormat�defaultr� r r r$ r/ r' rT r r^ rq �intr rf ro � staticmethodrF �strrB �
__classcell__r r r r r
% s(
B ��
�
" r
c s e Zd Zd� fdd� Z� ZS )r r Nc s0 dd l }t�� �� t�|�� � t� �� d S r )�uvloopr r r �set_event_loop_policy�EventLoopPolicyr r$ )r r� r r r r$ � s z$GunicornUVLoopWebWorker.init_processr� )r� r� r� r$ r� r r r r r � s r )!�__doc__r rK r� rd r- �typesr �typingr r r �gunicorn.configr r� �gunicorn.workersr �aiohttpr �helpersr
�web_appr �web_logr rx rw � ImportError�object�__all__�Workerr
r r r r r �<module> s8
� K