File: //opt/saltstack/salt/lib/python3.10/site-packages/aiohttp/__pycache__/worker.cpython-310.pyc
o
;j:! � @ s d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl m
Z
m Z m Z ddl
mZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ e
rXddlZejZnz ddlZejZW n
e yn 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�ZdededB ddfdd�ZdededB 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__� �B/opt/saltstack/salt/lib/python3.10/site-packages/aiohttp/worker.pyr * s
zGunicornWebWorker.__init__c sH zt �� �� W n ty Y nw t �� | _t �| j� t� �� d S �N) �asyncio�get_event_loop�close� RuntimeErrorZnew_event_loop�loopZset_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" Z create_task�_runr Zrun_until_complete� Exception�log� exceptionZshutdown_asyncgensr �sys�exitr r$ r r r �run> s �
zGunicornWebWorker.runc � s� �d }t | jt�r
| j}n/t�| j�s tjdk r4t�| j�r4| �� I d H }t |tj �r1|}|j
}n |}nt d| j� ���|d u rc| j j
rH| jjnd }tj || j| j j|| �| j j�| j jd d d�}|�� I d H | j jrt| �| j �nd }|}|d us~J �|j}|d us�J �| jD ]}tj|||d�}|�� I d H q�t� � } zA| j r�| � � |j }
| j r�|
| j kr�d| _ | j�!d| � n | t� � kr�| j"t�#� kr�d| _ | j�!d | � n| �$� I d H | j s�W n t%y� Y nw |�&� I d H d S )
N)� � zSwsgi app should be either Application or async function returning Application, got �d �_ )�loggerZkeepalive_timeout�
access_log�access_log_formatZshutdown_timeout)Z ssl_contextFz Max requests, shutting down: %sz!Parent changed, shutting down: %s)'�
isinstance�wsgir �inspectZiscoroutinefunctionr) � version_infor r Z AppRunner�appr! �cfgZ accesslogr' r1 Z keepalive�_get_valid_log_formatr2 Zgraceful_timeout�setupZis_ssl�_create_ssl_context�serverZsocketsZSockSite�start�os�getpid�alive�notifyZrequests_countZ max_requests�infoZppid�getppid�_wait_next_notifyr& Zcleanup) r Zrunnerr7 r4 r1 �ctxr< �sock�site�pidZcntr r r r% J sv � ����
���
�zGunicornWebWorker._run�asyncio.Future[bool]c C s> | � � | j}|d us
J �|�� | _}| j�d| j |� |S )Ng �?)�_notify_waiter_doner" Z
create_futurer Z
call_later)r r" �waiterr r r rD � s z#GunicornWebWorker._wait_next_notifyrK c C s8 |d u r| j }|d urt|d� || j u rd | _ d S d S )NT)r r
)r rK r r r rJ � 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� 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� siginterrupt�SIGCHLD�SIG_DFLr$ r r r � init_signals� s* ������z GunicornWebWorker.init_signals�sig�framec C s d| _ | j�| � | �� d S rL )r@ r8 Z
worker_intrJ �r rZ r[ r r r rO � s z GunicornWebWorker.handle_quitc C s& d| _ d| _| j�| � t�d� d S )NFr )r@ r r8 Z worker_abortr) r* r\ r r r rU � s z GunicornWebWorker.handle_abortr8 �
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! r] � ssl_version�load_cert_chain�certfile�keyfile� cert_reqs� verify_mode�ca_certs�load_verify_locations�ciphers� set_ciphers)r8 rE r r r r; � 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 ri r r r r9 � s
�z'GunicornWebWorker._get_valid_log_format�r N)r rI r )�__name__�
__module__� __qualname__r Z
LOG_FORMATrk �GunicornAccessLogFormat�defaultrj r r r# r+ r% rD r rJ rY �intr rO rU � staticmethodr; �strr9 �
__classcell__r r r r r
& s(
D ��
�
& r
c s e Zd Zd� fdd� Z� ZS )r r Nc sH dd l }zt�� �� W n ty Y nw t�|�� � t� �� d S r ) �uvloopr r r r! Zset_event_loop_policyZEventLoopPolicyr r# )r ry r r r r# � s �z$GunicornUVLoopWebWorker.init_processro )rp rq rr r# rx r r r r r � s r )"�__doc__r r5 r> rl rM r) �typesr �typingr r r Zgunicorn.configr rs Zgunicorn.workersr Zaiohttpr Zhelpersr
Zweb_appr Zweb_logr r^ r] � ImportError�object�__all__ZWorkerr
r r r r r �<module> s:
� T