Edit File: persistent_message.cpython-311.pyc
� �*vh � �H � d dl Z d dlZd dlmZ d dlmZ G d� d� � ZdS )� N)�db)� MessageToSendc �� � e Zd ZdZdd�Zdd�Zdefd�Zdefd �Z de fd �Z e de fd �� � Z e de fd �� � Zdd efd�Zdeeeef ddfd�Zdefd�ZdS )�PersistentMessagesQueuea� The queue to store messages sent to the server if it is unavailable. - stores more recent data; if a limit is exceeded, older messages are deleted. - no duplicate messages are sent NOTE: it is worth remembering that when writing a large number of messages, the amount of memory used may increase by the size of the sqlite cache (this may not be immediately obvious). https://www.sqlite.org/pragma.html#pragma_cache_size � �� Nc �L � || _ || _ g | _ |pt | _ d S �N)� _buffer_limit�_storage_limit�_bufferr �_model)�self� buffer_limit� storage_limit�models �a/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/internals/persistent_message.py�__init__z PersistentMessagesQueue.__init__ s) � � )��� +����� ��,�}�� � � � �returnc � � | j r�t j � � 5 | j � | j � � | j | j z }|dk r| j � |� � g | _ d d d � � d S # 1 swxY w Y d S d S �Nr )r r �atomicr � insert_many� storage_sizer � delete_old)r �need_to_removes r �push_buffer_to_storagez.PersistentMessagesQueue.push_buffer_to_storage s� � � �<� "����� "� "� � �'�'�� �5�5�5�!%�!2�T�5H�!H��!�A�%�%��K�*�*�>�:�:�:� !�� � "� "� "� "� "� "� "� "� "� "� "� "���� "� "� "� "� "� "� "� "s �AA>�>B�Bc � � g }t j � � 5 |t | j � | j j | j j � � � � � � � z }| j � � � � � � d d d � � n # 1 swxY w Y || j z }g | _ t |� � S r ) r r �listr �select� timestamp�message�tuples�delete�executer �sorted)r �itemss r �pop_allz PersistentMessagesQueue.pop_all' s� � ��� �Y�[�[� +� +� �T�� �"�"��K�)�4�;�+>�� ��&�(�(�� � �E� �K� � � � � (� (� *� *� *� +� +� +� +� +� +� +� +� +� +� +���� +� +� +� +� �� � ���� ��e�}�}� s �A=B � B#�&B#c �2 � | � � � dk S r )�qsize�r s r �emptyz PersistentMessagesQueue.empty4 s � ��z�z�|�|�q� � r c �: � | j t | j � � z S r )r �lenr r, s r r+ z PersistentMessagesQueue.qsize7 s � �� �3�t�|�#4�#4�4�4r c �* � t | j � � S r )r/ r r, s r � buffer_sizez#PersistentMessagesQueue.buffer_size: s � ��4�<� � � r c �X � | j � � � � � � S r )r r! �countr, s r r z$PersistentMessagesQueue.storage_size> s" � ��{�!�!�#�#�)�)�+�+�+r r# c � � |�t j � � }| j � ||f� � | j | j k r| � � � d S d S r )�timer �appendr1 r r )r r# r" s r �putzPersistentMessagesQueue.putB s_ � � � � � � � �I� � ���Y�� 0�1�1�1� � �t� 1� 1� 1� � '� '� )� )� )� )� )� 2� 1r �messagesc � � | j � |� � | j | j k r| � � � d S d S r )r �extendr1 r r )r r8 s r �put_manyz PersistentMessagesQueue.put_manyI sI � � � ���H�%�%�%� � �t� 1� 1� 1� � '� '� )� )� )� )� )� 2� 1r c �d � | j sq| j dk rf| j � | j �� � }| xj d� |� � � D � � z c_ | j � |� � | j r | j � d� � \ }}|S t j � � �)Nr )�limitc �" � g | ] }|d d� �� S )� N� )�.0�items r � <listcomp>z/PersistentMessagesQueue.get.<locals>.<listcomp>S s � � A� A� A�$�T�!�"�"�X� A� A� Ar ) r r r � get_oldestr r$ � delete_in�pop�queue�Empty)r r( �_r# s r �getzPersistentMessagesQueue.getN s� � ��|� )�� 1�Q� 6� 6��K�*�*��1C�*�D�D�E� �L�L� A� A�%�,�,�.�.� A� A� A� A�L�L� �K� !� !�%� (� (� (� �<� � � �)�)�!�,�,�J�A�w��N��k�m�m�r )r r N)r Nr )�__name__� __module__� __qualname__�__doc__r r r r) �boolr- �intr+ �propertyr1 r �bytesr7 �tuple�floatr; rJ r@ r r r r sP � � � � � � � �-� -� -� -� "� "� "� "� �� � � � �!�t� !� !� !� !�5�s� 5� 5� 5� 5� �!�S� !� !� !� �X�!� �,�c� ,� ,� ,� �X�,�*� *�5� *� *� *� *�*��e�E�5�L�&9�!:� *�t� *� *� *� *� �U� � � � � � r r )rG r5 � defence360agent.model.instancer �&defence360agent.model.messages_to_sendr r r@ r r �<module>rW sw �� � � � � � � � � -� -� -� -� -� -� @� @� @� @� @� @�Q � Q � Q � Q � Q � Q � Q � Q � Q � Q r
Back to File Manager