/usr/share/cagefs-skeleton/opt/imunify360/venv/lib/python3.11/site-packages/im360/__pycache__
� ��BS1z� � � d dl Z d dlZd dlmZ d dlmZmZ d dlmZ d dl m Z d dl m Z ej e� � Z G d� de e� � ZdS ) � N)� Generator)� send_message�NATSSendMessageException)� MessageSink)�SendToServerClient)�Scopec � � � e Zd Zej ZdZ� fd�Zej de e j ddf fd�� � Z dd�Zd� fd� Z� xZS ) �SendToServerNATSi� c �H �� t � � j |i |�� d | _ d S �N)�super�__init__� _nats_api)�self�args�kwargs� __class__s ��G/opt/imunify360/venv/lib/python3.11/site-packages/im360/nats_gateway.pyr zSendToServerNATS.__init__ s* �� ������$�)�&�)�)�)� ����� �returnNc # �Z K � | j �t j � � | _ | j V � d S r )r r �NATSGatewayAPI)r s r �_get_apizSendToServerNATS._get_api s4 � � � � �>� !� )� 8� :� :�D�N��n� � � � � r c � � K � | j � � � rd S | j � � � r t � d� � d S | j � � � }t � dt |� � � � | � � � 5 } |� |� � � d {V �� | j � � � dk r2t � d| j � � � � � n�# t $ rd}||j d � }|r| j � |� � t � d|j t |� � |� � Y d }~n,d }~wt $ r | j � |� � � w xY wd d d � � d S # 1 swxY w Y d S )Nz,Shutdown signal received, skipping NATS sendz Sending %s messages via NATSr z Still need to send %s messagesz@Failed to send messages via NATS: %d published, %d re-queued: %s)�_pending�empty�_shutting_down�is_set�logger�warning�pop_all�info�lenr � send_messages�qsizer � published�put_many� BaseException)r �messages�api�e�unsents r �_send_pending_messagesz'SendToServerNATS._send_pending_messages s � � � � �=� � � � � ��4� � � %� %� '� '� � �N�N�I� J� J� J��4��=�(�(�*�*��� � �2�C��M�M�B�B�B� �]�]�_�_� �� ��'�'��1�1�1�1�1�1�1�1�1��=�&�&�(�(�1�,�,��K�K�8� � �+�+�-�-�� � ��� ,� � � �!�!�+�-�-�0��� 3��M�*�*�6�2�2�2����5��K���K�K�� � � � � � � � ����� !� � � � � �&�&�x�0�0�0�� ����' � � � � � � � � � � � ���� � � � � � s> �(F7�*A*D�F7� F'� AE>�9F7�>)F'�'F7�7F;�>F;c � �� �K � t � � � � � � d {V �� | j �;| j � � � � d {V �� t � d� � d S d S )NzNATS connection closed)r �shutdownr �closer r" )r r s �r r/ zSendToServerNATS.shutdownA s~ �� � � ��g�g� � � � � � � � � � � � �>� %��.�&�&�(�(� (� (� (� (� (� (� (� �K�K�0� 1� 1� 1� 1� 1� &� %r )r N)�__name__� __module__� __qualname__r �IM360�SCOPE�SHUTDOWN_PRIORITYr � contextlib�contextmanagerr r r r r- r/ � __classcell__)r s @r r r s� �� � � � � � �K�E��� � � � � � � � � �<�.��d�:� ;� � � � � � �!� !� !� !�F2� 2� 2� 2� 2� 2� 2� 2� 2� 2r r )r7 �logging�typingr �defence360agent.api.serverr r �!defence360agent.contracts.pluginsr � defence360agent.plugins.clientr �defence360agent.utilsr � getLoggerr1 r r � r r �<module>rB s� �� � � � � ���� � � � � � � M� M� M� M� M� M� M� M� 9� 9� 9� 9� 9� 9� =� =� =� =� =� =� '� '� '� '� '� '� �� �8� $� $��82� 82� 82� 82� 82�)�;� 82� 82� 82� 82� 82r
.
Edit
..
Edit
__init__.cpython-311.opt-1.pyc
Edit
__init__.cpython-311.pyc
Edit
__main__.cpython-311.opt-1.pyc
Edit
__main__.cpython-311.pyc
Edit
_version.cpython-311.opt-1.pyc
Edit
_version.cpython-311.pyc
Edit
aibolit_job.cpython-311.opt-1.pyc
Edit
aibolit_job.cpython-311.pyc
Edit
cli.cpython-311.opt-1.pyc
Edit
cli.cpython-311.pyc
Edit
nats_gateway.cpython-311.opt-1.pyc
Edit
nats_gateway.cpython-311.pyc
Edit
rpc_handlers.cpython-311.opt-1.pyc
Edit
rpc_handlers.cpython-311.pyc
Edit
run.cpython-311.opt-1.pyc
Edit
run.cpython-311.pyc
Edit