관리-도구
편집 파일: multiprocess.cpython-38.pyc
U .?�fs � @ s� d dl mZ d dlZd dlZd dlZd dlZddlmZ ddlm Z ddl m Z ddl m Z ddlmZ ze W n ek r� eZY nX G d d � d �Zd d � ejD �Zdd d�ZdS )� )� defaultdictN� )�Gauge)�Metric)� MmapedDict)�Sample)�floatToGoStringc @ sH e Zd ZdZddd�Zeddd��Zedd � �Zed d � �Zd d � Z dS )�MultiProcessCollectorz+Collector for files for multi-process mode.Nc C st |d krDdt jkr8dt jkr8t jd t jd< t�dt� t j�d�}|rTt j�|�s\td��|| _ |rp|� | � d S )N�prometheus_multiproc_dir�PROMETHEUS_MULTIPROC_DIRzpprometheus_multiproc_dir variable has been deprecated in favor of the upper case naming PROMETHEUS_MULTIPROC_DIRz:env PROMETHEUS_MULTIPROC_DIR is not set or not a directory) �os�environ�warnings�warn�DeprecationWarning�get�path�isdir� ValueError�_path�register)�self�registryr � r �L/opt/hc_python/lib/python3.8/site-packages/prometheus_client/multiprocess.py�__init__ s z MultiProcessCollector.__init__Tc C s t �| �}t �||�S )z�Merge metrics from given mmap files. By default, histograms are accumulated, as per prometheus wire format. But if writing the merged data back to mmap files, use accumulate=False to avoid compound accumulation. )r � _read_metrics�_accumulate_metrics)�files� accumulate�metricsr r r �merge# s zMultiProcessCollector.mergec s i }i � � fdd�}| D ]�}t j�|��d�}|d }zt�|�}W n0 tk rt |dkrn|d �d�rnY q� Y nX |D ]�\}}} } ||�\} } } }}|�| �}|d kr�t | ||�}||| < |dkr�|d d d � }|d |_ |� | |d |ff || � qz|� | ||� qzq|S ) Nc sJ � � | �}|sFt�| �\}}}}tt|�� ��}|||||f }� | < |S �N)r �json�loads�tuple�sorted�items)�key�val� metric_name�name�labels� help_text� labels_key�Z key_cacher r � _parse_key3 s z7MultiProcessCollector._read_metrics.<locals>._parse_key�_r �gauger �live� ����pid) r r �basename�splitr Zread_all_values_from_file�FileNotFoundError� startswithr r �_multiprocess_modeZ add_sample)r r r0 �f�parts�typZ file_valuesr( �value� timestampr1 r* r+ r, r. r- �metricr6 r r/ r r . s0 z#MultiProcessCollector._read_metricsc C sV | � � D �]B}tt�}tt�}tdd� �}|j}|jD �]n}|\}} } } } |jdk�r |tdd� | D ��f} |jdkr�|| | �}| |k r�| || < n�|jdkr�|| | �}| |kr�| || < nd|jdkr�|| | 7 < nH|jd k�r|| }t| p�d �} || k �r | || < | || < n | ||| f< q6|jd k�r�| D ]L}|d d k�r.t|d �}tdd� | D ��}|| | | 7 < �q��q.||| f | 7 < q6||| f | 7 < q6|jd k�r8|�� D ]z\} }d}t |�� �D ]F\}} |j d | d t |�ff f}|�r|| 7 }|||< n| ||< �q�|�r�|||j d | f< �q�dd� |�� D �|_q| � � S )Nc S s t t�S r" )r �floatr r r r �<lambda>] � z;MultiProcessCollector._accumulate_metrics.<locals>.<lambda>r2 c s s | ]}|d dkr|V qdS )r r6 Nr ��.0�lr r r � <genexpr>b s z<MultiProcessCollector._accumulate_metrics.<locals>.<genexpr>)�minZlivemin)�maxZlivemax)�sumZlivesum)Z mostrecentZlivemostrecentr Z histogram�ler c s s | ]}|d dkr|V qdS )r rL Nr rE r r r rH } s g Z_bucket�_countc S s$ g | ] \\}}}t |t|�|��qS r )r �dict)rF Zname_r, r? r r r � <listcomp>� s z=MultiProcessCollector._accumulate_metrics.<locals>.<listcomp>) �valuesr rB � setdefault�samples�typer% r; r' r& r+ r )r r rA rR Zsample_timestampsZbucketsZsamples_setdefault�sr+ r, r? r@ ZexemplarZwithout_pid_key�currentZcurrent_timestamprG Z bucket_valueZ without_lerP �accZbucketZ sample_keyr r r r X sd � z)MultiProcessCollector._accumulate_metricsc C s$ t � tj�| jd��}| j|dd�S )Nz*.dbT)r )�globr r �joinr r! )r r r r r �collect� s z MultiProcessCollector.collect)N)T) �__name__� __module__� __qualname__�__doc__r � staticmethodr! r r rY r r r r r s ) Cr c C s h | ]}|� d �r|�qS )r3 )r: )rF �mr r r � <setcomp>� s r` c C s^ |dkr t j�dt j�d��}tD ]6}t�t j�|d|� d| � d���D ]}t �|� qHq"dS )zBDo bookkeeping for when one process dies in a multi-process setup.Nr r Zgauge_r1 z.db)r r r � _LIVE_GAUGE_MULTIPROCESS_MODESrW r rX �remove)r6 r �moder< r r r �mark_process_dead� s &rd )N)� collectionsr rW r# r r r r Z metrics_corer Z mmap_dictr rR r �utilsr r9 � NameError�IOErrorr Z_MULTIPROC_MODESra rd r r r r �<module> s"