hjkhjkjhkhjkgjghjhgjhgjghjhghjghjhgjjnbvnvbnvk
bnmbnmbnhjkhjkhjkhjktyutyutyuytutyutyutyhgjgjghjgjghjguytutyry
/
opt
/
alt
/
python37
/
lib
/
python3.7
/
site-packages
/
xray
/
continuous
/
__pycache__
/
Upload FileeE
HOME
B �'Ud� � @ sV d Z ddlZddlZddlmZ ddlmZ ddlmZ ddl m Z G dd � d �Z dS ) zG This module contains basic part of continuous tracingg implementation � N)�List� )�continuous_storage)� XRayError)�ContinuousTaskc @ sZ e Zd ZdZdd� Zed�dd�Zee d�dd�Z ee d�d d �Z e d d �d d�Z d S )�ContinuousCommonz+ Base class for continuous tracing c C s t | _t�d�| _| �� | _d S )N� continuous)r �storage�logging� getLogger�logger�load_tracing_configuration� tracing_conf)�self� r �G/opt/alt/python37/lib/python3.7/site-packages/xray/continuous/common.py�__init__ s zContinuousCommon.__init__)�returnc s| | j �d� y.t�| j��� � fdd�� �� D �S Q R X W n< tk rv } z | j jddt|�id� t � S d}~X Y nX dS )zS Load full continuous tracing configuration from the local storage z(Loading continuous tracing configurationc s i | ]}� | |�qS r r )�.0�item)�dbr r � <dictcomp>& s z?ContinuousCommon.load_tracing_configuration.<locals>.<dictcomp>Nz/Failed to load continuous tracing configuration�err)�extra) r �info�shelve�openr �keys�OSError�error�str�dict)r �er )r r r s z+ContinuousCommon.load_tracing_configurationc C s t dd� | j�� D ��S )zC Return existing configuration in the form of list c S s h | ]}t f |��qS r )r )r r r r r � <setcomp>2 s z8ContinuousCommon.get_continuous_tasks.<locals>.<setcomp>)�sortedr �values)r r r r �get_continuous_tasks- s z%ContinuousCommon.get_continuous_tasksc s� yLt �| j��6� x | jD ]}| j| � |< qW � fdd�� �� D �S Q R X W n4 tk r� } ztdt|�� ���W dd}~X Y nX dS )zJ Dump continuous tracing configuration into local storage c s g | ] }� | �qS r r )r r )r r r � <listcomp>< s z?ContinuousCommon.dump_tracing_configuration.<locals>.<listcomp>Nz)Failed to dump continuous configuration: )r r r r r r r r )r r r" r )r r �dump_tracing_configuration4 s z+ContinuousCommon.dump_tracing_configurationN)�domainr c C s� y*| j |= t�| j�� }||= W dQ R X W nt tk rf | jjd|| j d�d� td|� d���Y n: tk r� } z td|� dt |�� ���W dd}~X Y nX dS ) zD Remove continuous tracing configuration for domain Nz$Continuous monitoring is not enabled)r) �tracing_entries)r zContinuous monitoring for z is not enabledz.Failed to remove continuous configuration for z: ) r r r r �KeyErrorr r r r r )r r) r r" r r r � remove_tracing_configurationA s z-ContinuousCommon.remove_tracing_configuration) �__name__� __module__� __qualname__�__doc__r r! r r r r& r( r r, r r r r r s r ) r0 r r �typingr Zinternal.constantsr �internal.exceptionsr �internal.typesr r r r r r �<module> s