Y2Y d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZm Z m
Z
d dl m Z m
Z
mZ d dlmZmZ ee Z G d de e
ZdS ) N) getLogger)
inactivity)
Accumulatable MessageType
Splittable) MessageSink
MessageSourceexpect)recurring_checksafe_cancel_taskc & e Zd Zej j ZdZ ee j
dd Z ee j
dd Z
e e
f fd Zd Zd Zd Zd
Z eej d efd Zd
Z xZS )
Accumulate %IMUNIFY360_AGGREGATE_MESSAGES_TIMEOUT< *IMUNIFY360_AGGREGATE_SHUTDOWN_SEND_TIMEOUT2 c t j di | || _ || _ t j t | _ d S )N )super__init___period_shutdown_timeout collections defaultdictlist_data)selfperiodshutdown_timeoutkwargs __class__s W/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/plugins/accumulate.pyr zAccumulate.__init__" sG ""6""" !1 ,T22
c K || _ || _ | j dk rd n<| t | j | j | _ d S )Nr )_loop_sinkr create_taskr _flush_task)r loopsinks r#
create_sourcezAccumulate.create_source- sf
|q
D!!"L"?/$,"?"? "L"L"N"NOO
r$ c K || _ d S )N)r& )r r+ s r# create_sinkzAccumulate.create_sink6 s
r$ c $ K t j | | j d {V d S # t j $ rH t
d| j | j t | j d {V Y d S Y d S w xY w)Nz5Timeout (%ss) sending messages to server on shutdown.) asynciowait_forstopr TimeoutErrorloggererrorr* r r s r# shutdownzAccumulate.shutdown9 s 3"499;;0FGG G G G G G G G G G# 3 3 3 LLG&
z%&tz222222222222 &%%
3s 28 ABBc K t d | j t | j d {V t d | d {V d S )Nz Accumulate.stop cancel _taskzAccumulate.stop wait lock)r5 infor* r r) r7 s r# r3 zAccumulate.stopE s| 2333 : !"4:.. . . . . . . . /000kkmmr$ messagec D K t |j t r|j n|j f}| r`t j d 5 |D ]"}| j | | # d d d d S # 1 swxY w Y d S d S )N
accumulate)
isinstance
LIST_CLASStuple
do_accumulater tracktaskr append)r r;
list_types list_types r# collectzAccumulate.collectM s ',e44
'G $&
" " :!&&|44
:
:!+ : :IJy)009999:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
: : :s &BBBc
K | j }t j t | _ | D ]\ }}t |t r| | n|f}|D ]}t d|j
dt | d | j
|| d {V a# t $ r t d|| w xY wd S )NzPrepare z() for further processing)itemsz%s, %s)r r r r rI
issubclassr batchedr5 r: __name__lenr' process_message TypeErrorr6 )r copy_datarF messagesrK batchs r# r) zAccumulate._flushY sB J ,T22
#,??#4#4 Ix i44! !!(+++ [
!
-y 1 - -3u:: - - - *44YYU5K5K5KLLLLLLLLLL LL9e<<<
s &*C(C9)rL
__module__ __qualname__r ProcessingOrderPOST_PROCESS_MESSAGEPROCESSING_ORDERSHUTDOWN_PRIORITYintosenvirongetDEFAULT_AGGREGATE_TIMEOUTSHUTDOWN_SEND_TIMEOUTr r- r/ r8 r3 r
r r rG r)
__classcell__)r" s @r# r r s) "2G !$
>CC! ! C
CRHH ). 3 3 3 3 3 3
3
3
3 VK %&& :] : : : '& : r$ r )r1 r rZ loggingr defence360agent.apir "defence360agent.contracts.messagesr r r !defence360agent.contracts.pluginsr r r
defence360agent.utilsr r rL r5 r r r$ r# re s * * * * * *
D C C C C C C C 8 Z Z Z Z Zm Z Z Z Z Zr$