관리-도구
편집 파일: mmap_dict.cpython-38.pyc
U .?�f � @ s� d dl Z d dlZd dlZd dlZd dlmZ dZe�d�jZ e�d�jZ e�d�j Z e�d�j Z dd� Zdd � Zdd d �ZG d d � d �Zeeee ee eed�dd�ZdS )� N)�Listi � is ddc C s t ||�| ||d �< d S )N� )�_pack_two_doubles_func)�data�pos�value� timestamp� r �I/opt/hc_python/lib/python3.8/site-packages/prometheus_client/mmap_dict.py�_pack_two_doubles s r c C s t |�| ||d �< d S )N� )�_pack_integer_func)r r r r r r � _pack_integer s r c c s� |dkrt | d�d }d}||k r�t | |�d }|| |krDtd��|d7 }| ||| � }|d|d d }||7 }t| |�\}}|�d�|||fV |d7 }qdS )z<Yield (key, value, timestamp, pos). No locking is performed.r � z2Read beyond file size detected, file is corrupted.r �utf-8r N)�_unpack_integer� RuntimeError�_unpack_two_doubles�decode)r �usedr Z encoded_lenZ encoded_keyZ padded_lenr r r r r �_read_all_values s r c @ sV e Zd ZdZddd�Zedd� �Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dS )� MmapedDicta� A dict of doubles, backed by an mmapped file. The file starts with a 4 byte int, indicating how much of it is used. Then 4 bytes of padding. There's then a number of entries, consisting of a 4 byte int which is the size of the next field, a utf-8 encoded string key, padding to a 8 byte alignment, and then a 8 byte float which is the value and a 8 byte float which is a UNIX timestamp in seconds. Not thread safe. Fc C s� t ||r dnd�| _|| _t�| j�� �j}|dkrD| j�t� t}|| _ t j | j�� | j |rdt j nt j d�| _ i | _t| j d�d | _| jdkr�d| _t| j d| j� n$|s�| �� D ]\}}}}|| j|< q�d S )N�rbza+br )�accessr )�open�_fZ_fname�os�fstat�fileno�st_size�truncate�_INITIAL_MMAP_SIZE� _capacity�mmap� ACCESS_READ� ACCESS_WRITE�_m� _positionsr �_usedr r )�self�filenameZ read_mode�capacity�key�_r r r r �__init__? s$ � zMmapedDict.__init__c C s\ t | d��B}|�tj�}t|d�d }|t|�krH||�|t|� �7 }W 5 Q R X t||�S )Nr r )r �readr$ �PAGESIZEr �lenr )r+ �infpr r r r r �read_all_values_from_fileT s z$MmapedDict.read_all_values_from_filec C s� |� d�}|ddt|�d d }t�dt|�� d�� � t|�|dd�}| jt|� | jkr�| jd9 _| j�| j� t�| j� � | j�| _ qL|| j | j| jt|� �< | jt|�7 _t | j d | j� | jd | j |< d S ) z0Initialize a value. Lock must be held by caller.r � r r �iZsddg � r r N) �encoder2 �struct�packr) r# r r! r$ r r' r r( )r* r- �encoded�paddedr r r r � _init_value_ s &zMmapedDict._init_valuec C s t | j| jd�S )z1Yield (key, value, pos). No locking is performed.)r r )r r' r) �r* r r r r p s zMmapedDict._read_all_valuesc c s& | � � D ]\}}}}|||fV qdS )z7Yield (key, value, timestamp). No locking is performed.N)r )r* �k�v�tsr. r r r �read_all_valuest s zMmapedDict.read_all_valuesc C s* || j kr| �|� | j | }t| j|�S �N)r( r= r r' )r* r- r r r r � read_valuey s zMmapedDict.read_valuec C s2 || j kr| �|� | j | }t| j|||� d S rC )r( r= r r' )r* r- r r r r r r � write_value s zMmapedDict.write_valuec C s* | j r&| j�� d | _| j �� d | _ d S rC )r r' �closer> r r r rF � s zMmapedDict.closeN)F) �__name__� __module__� __qualname__�__doc__r/ � staticmethodr4 r= r rB rD rE rF r r r r r 2 s r )� metric_name�name� labelnames� labelvalues� help_text�returnc C s$ t t||��}tj| |||gdd�S )z&Format a key for use in the mmap file.T)� sort_keys)�dict�zip�json�dumps)rL rM rN rO rP �labelsr r r �mmap_key� s rX )r )rU r$ r r9 �typingr r" �Structr: r r � unpack_fromr r r r r r �strrX r r r r �<module> s [