File: //usr/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< eed< dZ e
e e e
ee ef f ed< dZeed< d Zeed
< d Zeed < d Zeed
< edd� �Zedd�dd�Ze ed�dd�Z e d�dd�Z ee
e e eef d�dd�Z e e d �d d �Z!edd�d d �Z"dS )!�InternalMetricsChannelZmetrics1)�source�internal�metrics�samplersN�samplers_cache�� �intervalT� need_metar �last_timestamp�next_timestampc 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)�options�returnc 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+ r/ 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# ) �selfr1 r& r" Zsampler_classesZmetricr* r4 r r, r �excr r r �
parse_options8 s2
*
z$InternalMetricsChannel.parse_options)�samples� timestampc 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_meta)r2 c 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_valuer2 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_samplesc
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 r0 rF Zreadyr rR rS r_ �asyncio�sleepr& )rD r1 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) � classmethodr0 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 (