File: //opt/alt/python37/lib/python3.7/site-packages/xray/continuous/__pycache__/common.cpython-37.pyc
B
�V�d�
� @ 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