File: //opt/cloudlinux/venv/lib64/python3.11/site-packages/ssa/__pycache__/agent.cpython-311.pyc
�
��si6
� � � d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z ddl
m Z ddl m
Z
dd lmZ d
Z G d � d � � ZdS )
zH
This module contains contains classes implementing SSA Agent behaviour
� N)�ThreadPoolExecutor)�current_thread� )�
agent_sock)�SSAError)�
create_socket)�RequestProcessor�2 c �> � e Zd ZdZd� Zd� Zd
d�Zdd �Zd
d defd �Z dS )� SimpleAgentz
SSA Simple Agent class
c �� � t j d� � | _ t � � | _ t t �� � | _ t j | j
� � | � � � d S )N�agent)� max_workers) �logging� getLogger�loggerr �request_processorr � MAX_WORKERS�executor�atexit�register� _shutdown�listen��selfs �@/opt/cloudlinux/venv/lib64/python3.11/site-packages/ssa/agent.py�__init__zSimpleAgent.__init__ sW � � �'��0�0�� �!1�!3�!3�� �*�{�C�C�C��
�����'�'�'� � � �
�
�
�
�
� c �< � | j � d�� � dS )z-Gracefully shutdown the thread pool executor.F)�waitN)r �shutdownr s r r zSimpleAgent._shutdown' s! � � �
� � �E� �*�*�*�*�*r �returnNc �� � t t � � } |� � � \ }}| j � | j |� � | j � d� � �R)z(
Start listening socket
Tz[ThreadPool] Submitted task)r r �acceptr �submit�handler �debug)r �_socket�
connection�addresss r r zSimpleAgent.listen+ s` � � �
�+�+�� =�")�.�.�"2�"2� �J�� �M� � ���j� 9� 9� 9� �K� � � ;� <� <� <� =r r) �
socket objectc �^ � |� d�� � } | � |� � }| j � |� � nQ# t $ rD}| j � dt � � j t |� � � � Y d}~nd}~ww xY w|�
� � dS )z�
Handle incoming connection
:param connection: socket object usable to
send and receive data on the connection
�ignore)�errorsz Handled exception in [%s]: %sN) �makefile�
read_inputr r& r r �errorr �name�str�close)r r) �fileobj�
input_data�es r r& zSimpleAgent.handle5 s� � � �%�%�X�%�6�6�� � � � ��1�1�J� � "� )� )�*� 5� 5� 5� 5��� � � � �K� � � =� ,� .� .� 3�S��V�V�
=�
=�
=� �D�D�D�D����� ���� �������s �/A �
B�:B�B�fileioz file objectc � � |� � � }| j � dt � � j t |� � � � � |� � |r(t j |� � � d�� � S i S )zy
Read input data and return decoded json
:param fileio: a file-like object providing read method
z [%s] I received %i bytes: %sF)�strict)
�readr �infor r2 �len�encode�json�loads�strip)r r8 �datas r r0 zSimpleAgent.read_inputE s{ � �
�{�{�}�}�� � ���7�'�)�)�.��D�K�K�M�M�0B�0B� � � � � � ��:�d�j�j�l�l�5�9�9�9� 9��Ir )r" N)r) r+ r" N)
�__name__�
__module__� __qualname__�__doc__r r r r& �dictr0 � r r r r s� � � � � � �� �� � �+� +� +�=� =� =� =�� � � � � � �4� � � � � � r r )rF r r? r �concurrent.futuresr � threadingr �internal.constantsr �internal.exceptionsr �internal.utilsr �modules.processorr r r rH r r �<module>rO s� ��� � �
�
�
� � � � � ���� 1� 1� 1� 1� 1� 1� $� $� $� $� $� $� *� *� *� *� *� *� )� )� )� )� )� )� )� )� )� )� )� )� /� /� /� /� /� /� � �7� 7� 7� 7� 7� 7� 7� 7� 7� 7r