관리-도구
편집 파일: registry.cpython-38.pyc
U .?�f4 � @ s� d dl mZmZ d dlZd dlmZ d dlmZmZm Z m Z ddl m Z G dd� de�Z G d d � d e �ZG d d � d e �ZG d d� d�Zedd�ZdS )� )�ABC�abstractmethodN)�Lock)�Dict�Iterable�List�Optional� )�Metricc @ s" e Zd Zeee d�dd��ZdS )� Collector��returnc C s d S �N� ��selfr r �H/opt/hc_python/lib/python3.8/site-packages/prometheus_client/registry.py�collect s zCollector.collectN)�__name__� __module__� __qualname__r r r r r r r r r s r c @ s e Zd Zee d�dd�ZdS )�_EmptyCollectorr c C s g S r r r r r r r s z_EmptyCollector.collectN)r r r r r r r r r r r s r c @ s� e Zd ZdZd eeeeef d�dd�Ze dd�dd �Z e dd�d d �Z d d � Z e e d�dd�Ze e dd�dd�Zeeeef dd�dd�Zeeeef d�dd�Zdd� Zd eeeeef ee d �d d �ZdS )!�CollectorRegistryz�Metric collector registry. Collectors must have a no-argument method 'collect' that returns a list of Metric objects. The returned metrics should be consistent with the Prometheus exposition formats. FN)� auto_describe� target_infoc C s. i | _ i | _|| _t� | _i | _| �|� d S r )�_collector_to_names�_names_to_collectors�_auto_describer �_lock� _target_info�set_target_info)r r r r r r �__init__ s zCollectorRegistry.__init__)� collectorr c C s` | j �P | �|�}t| j��|�}|r4td�|���|D ]}|| j|< q8|| j|< W 5 Q R X dS )z Add a collector to the registry.z.Duplicated timeseries in CollectorRegistry: {}N)r � _get_names�setr � intersection� ValueError�formatr )r r" �names� duplicates�namer r r �register% s �� zCollectorRegistry.registerc C s6 | j �& | j| D ] }| j|= q| j|= W 5 Q R X dS )z%Remove a collector from the registry.N)r r r )r r" r* r r r � unregister2 s z CollectorRegistry.unregisterc C s� d}z |j }W n tk r" Y nX |s4| jr4|j}|s<g S g }ddgdddgddddgdddgd gd �}|� D ]4}|�|j� |�|jg �D ]}|�|j| � q�qp|S ) z@Get names of timeseries the collector produces and clashes with.NZ_totalZ_createdZ_sum�_countZ_bucketZ_gsumZ_gcount�_info)�counter�summaryZ histogramZgaugehistogram�info)Zdescribe�AttributeErrorr r �appendr* �get�type)r r" Z desc_func�resultZ type_suffixes�metric�suffixr r r r# 9 s* � z CollectorRegistry._get_namesr c c sZ d}d}| j � t�| j�}| jr*| �� }W 5 Q R X |r>|V |D ]}|�� E dH qBdS )z3Yields metrics from the collectors in the registry.N)r �copyr r �_target_info_metricr )r � collectors�tir" r r r r V s zCollectorRegistry.collect�RestrictedRegistry)r( r c C s t |�}t|| �S )a Returns object that only collects some metrics. Returns an object which upon collect() will return only samples with the given names. Intended usage is: generate_latest(REGISTRY.restricted_registry(['a_timeseries'])) Experimental.)r$ r= )r r( r r r �restricted_registryc s z%CollectorRegistry.restricted_registry)�labelsr c C sZ | j �J |r2| js$d| jkr$td��t� | jd< n| jrF| j�dd � || _W 5 Q R X d S )Nr z7CollectorRegistry already contains a target_info metric)r r r r& r �pop)r r? r r r r p s z!CollectorRegistry.set_target_infoc C s$ | j � | jW 5 Q R � S Q R X d S r )r r r r r r �get_target_infoz s z!CollectorRegistry.get_target_infoc C s t ddd�}|�d| jd� |S )N�targetzTarget metadatar1 r r )r Z add_sampler )r �mr r r r: ~ s z%CollectorRegistry._target_info_metric)r* r? r c C sL |dkr i }| � � D ]2}|jD ]&}|j|kr |j|kr |j S q qdS )z~Returns the sample value, or None if not found. This is inefficient, and intended only for use in unittests. N)r Zsamplesr* r? �value)r r* r? r7 �sr r r �get_sample_value� s z"CollectorRegistry.get_sample_value)FN)N)r r r �__doc__�boolr r �strr! r r+ r, r# r r r r> r rA r: �floatrF r r r r r s r c @ s2 e Zd Zee ed�dd�Zee d�dd�ZdS )r= )r( �registryc C s t |�| _|| _d S r )r$ � _name_set� _registry)r r( rK r r r r! � s zRestrictedRegistry.__init__r c c s� t � }d }| jj�T d| jkr0| jjr0| j�� }| jD ]*}|dkr6|| jjkr6|�| jj| � q6W 5 Q R X |rv|V |D ](}|�� D ]}|� | j�}|r�|V q�qzd S )Nr ) r$ rM r rL r r: r �addr Z_restricted_metric)r r; Ztarget_info_metricr* r" r7 rC r r r r � s zRestrictedRegistry.collectN) r r r r rI r r! r r r r r r r= � s r= T)r )�abcr r r9 � threadingr �typingr r r r Z metrics_corer r r r r= ZREGISTRYr r r r �<module> s |