Current File : //lib/python3.9/site-packages/cockpit/channels/__pycache__/metrics.cpython-39.pyc
a
��h� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZm Z m
Z
m Z m Z m
Z
mZmZ ddlmZmZ ddlmZmZmZ ddlmZmZmZmZ e� e �Z G dd � d e
�Z G d
d � d e�Z dS ) � N)� defaultdict)�Dict�List�
NamedTuple�Optional�Set�Tuple�Type�Union� )� AsyncChannel� ChannelError)�JsonList�
JsonObject�get_int)�SAMPLERS�SampleDescription�Sampler�Samplesc @ s" e Zd ZU ee ed< eed< dS )�
MetricInfo�derive�descN)�__name__�
__module__� __qualname__r �str�__annotations__r � r r �</usr/lib/python3.9/site-packages/cockpit/channels/metrics.pyr ! s
r c @ s e Zd ZU dZdgZee ed< ee ed< dZ
e e e
eee ef f ed< dZeed< d Zeed
< d Zeed < d Zeed
< edd�dd��Zedd�dd�Ze edd�dd�Z e d�dd�Z ee e e eef d�dd�Z e e dd �d d �Z!edd�d d �Z"dS )!�InternalMetricsChannelZmetrics1)�source�internal�metrics�samplersN�samplers_cache�� �intervalT� need_metar �last_timestamp�next_timestamp)�returnc C s | j d u rdd� tD �| _ d S )Nc S s$ i | ] }|j D ]}|j||f�qqS r )Z descriptions�name)�.0�samplerr r r r �
<dictcomp>6 � z:InternalMetricsChannel.ensure_samplers.<locals>.<dictcomp>)r$ r )�clsr r r �ensure_samplers3 s
z&InternalMetricsChannel.ensure_samplers)�optionsr* c C sx t �d|| j� t|d| j�}|dks0|tjkrBtdd|� �d��|| _|�d�}t |t
�rht |�dkr�t � d|� tdd d��| j
s�J d
��t� }|D ]�}|�d �}|�d �}|�d
�}z| j
| \} }
W nF t�y } z,t � d|� tdd|� �d�| �W Y d } ~ n
d } ~ 0 0 |�rD||
jk�rDtd|� d|
j� d|� �d��|�| � | j�t||
d�� q�dd� |D �| _d S )Nz&metrics internal open: %s, channel: %sr& r zprotocol-errorzinvalid "interval" value: )�messager" zinvalid "metrics" value: %sz5invalid "metrics" option was specified (not an array)zensure_samples not calledr+ �unitsr zunsupported metric: %sz
not-supportedzunsupported metric: z has units z, not )r r c S s h | ]
}|� �qS r r )r, r0 r r r � <setcomp>Z r/ z7InternalMetricsChannel.parse_options.<locals>.<setcomp>)�logger�debug�channelr r& �sys�maxsizer
�get�
isinstance�list�len�errorr$ �set�KeyErrorr4 �addr" �appendr r# ) �selfr2 r& r" Zsampler_classesZmetricr+ r4 r r- r �excr r r �
parse_options8 s2
*
z$InternalMetricsChannel.parse_options)�samples� timestampr* c C s� g }| j D ]`}|jjrH|�|jj|jjt||jj �� �|jjd�� q
|�|jj|j |jj|jjd�� q
t
t � � �d }| j d| j
|d ||d� d| _d S )N)r+ r4 � instances� semantics)r+ r r4 rJ r% r! )r r& rH �nowr" F)r" r � instancedrC r+ r4 r= �keysrJ r �int�timeZ send_jsonr& r' )rD rG rH r" �
metricinforK r r r � send_meta\ s(
�
�
�z InternalMetricsChannel.send_metac C s" t t�}| jD ]}|�|� q|S )N)r �dictr# �sample)rD rG r- r r r rS s s
z InternalMetricsChannel.sample)�value� old_valuer* c C s$ |d ur || | j | j S dS d S )NF)r) r( )rD rT rU r r r �calculate_sample_ratey s z,InternalMetricsChannel.calculate_sample_rate)rG � last_samplesr* c
C sP g }t � � }|| _| jD �]}||jj }|jjr�||jj }t|t�sLJ �t|t�sZJ �|�� |�� krpd| _ |j
dkr�g }|� � D ] \} }
|� | �
|
|�| ��� q�|� |� n|� t|�� �� q|�|jj�}t|t�r�J �t|t�r�J �|j
dk�r|� | �
||�� q|� |� q| j �r2| �||� | j| _| �t�|g�� d S )NTZrate)rO r) r" r r+ rL r< rR rM r' r �itemsrC rV r; r= �valuesrQ r( Z send_text�json�dumps) rD rG rW �datarH rP rT rU rI �key�valr r r � send_updates s6
z#InternalMetricsChannel.send_updatesc � sd g | _ t� | _t�� | �|� | �� tt�}| � � }| �
||� |}t � | j
d �I d H q0d S )Nr% )r" r@ r# r r1 rF Zreadyr rR rS r_ �asyncio�sleepr& )rD r2 rW rG r r r �run� s
zInternalMetricsChannel.run)#r r r ZpayloadZ restrictionsr r r r r r$ r r r r r r r& rN r' �boolr( �floatr) � classmethodr1 r rF r rQ rS r
rV r_ rb r r r r r & s"
$ $ )r )!r` rZ Zloggingr9 rO � collectionsr �typingr r r r r r r r
r8 r r
Zjsonutilr r r rG r r r r Z getLoggerr r6 r r r r r r �<module> s (
Mini Shell