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_messageNATSSendMessageException) MessageSink)SendToServerClient)Scopec e Zd Zej ZdZ fdZej de
e j ddf fd Z
ddZd fd Z xZS ) SendToServerNATSi c H t j |i | d | _ d S N)super__init__ _nats_api)selfargskwargs __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 : :DNn 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)_pendingempty_shutting_downis_setloggerwarningpop_allinfolenr
send_messagesqsizer publishedput_many
BaseException)r messagesapieunsents r _send_pending_messagesz'SendToServerNATS._send_pending_messages s = 4 % % ' ' NNI J J J4=((** 2CMMBBB
]]__
''111111111=&&((1,,KK8
++-- ,
!!+--0 3M**62225KKK !
&&x000
' s> (F7*A*DF7
F' AE>9F7>)F''F77F;>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~ gg > %.&&(( ( ( ( ( ( ( ( KK0 1 1 1 1 1 & %r )r N)__name__
__module__ __qualname__r IM360SCOPESHUTDOWN_PRIORITYr
contextlibcontextmanagerr r r r r- r/
__classcell__)r s @r r
r
s KE <.d: ; ! ! ! !F2 2 2 2 2 2 2 2 2 2r r
)r7 loggingtypingr defence360agent.api.serverr r !defence360agent.contracts.pluginsr defence360agent.plugins.clientr defence360agent.utilsr getLoggerr1 r r
r r 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