관리-도구
편집 파일: metrics.cpython-38.pyc
U �MgIu � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m Z m Z d dl m Z d dlmZmZ d dlmZmZ d dlZd dlmZmZmZmZmZmZ d dlm Z m Z d dl m Z m Z m!Z!m"Z" d d l#m$Z$ e$�r�d d l#m%Z% d d l#m&Z& d d l#m'Z' d d l#m(Z( d dl#m)Z) d dl#m*Z* d dl#m+Z+ d dl#m,Z, d dl#m-Z- d dl#m.Z. d dl/m0Z0 d dl/m1Z1 d dl/m2Z2 d dl/m3Z3 d dl/m4Z4 d dl/m5Z5 d dl/m6Z6 d dl/m7Z7 d d l/m8Z8 d d l/m9Z9 ej:d e;d d � ed!d"d#�Z<e=Z>e?e e e!e"g�Z@ee�Ad$�jBd%�ZCee�Ad&�jBd'�ZDee�Ad(�jBd%�ZEd)d*� ZFd+d,� ZGe d-d.� �ZHd/d0� ZIG d1d2� d2e �ZJG d3d4� d4eJ�ZKG d5d6� d6eJ�ZLG d7d8� d8eJ�ZMG d9d:� d:eJ�ZNd;d<� ZOd=d>� ZPeKeLeMeNd?�ZQed@dA� dBdA� edCdA� dDdA� dEdA� dFdA� dG�ZRG dHdI� dI�ZSG dJdK� dK�ZTdLdM� ZUdNdO� ZVdPdQ� ZWdRdS� ZXdcdVdW�ZYeYZZG dXdY� dY�Z[ddd[d\�Z\ded]d^�Z]dfd_d`�Z=dgdadb�Z^dS )h� N)�ABC�abstractmethod)�contextmanager)�datetime�timezone)�wraps�partial)� ContextVar�now�nanosecond_time� to_timestamp�serialize_frame� json_dumps)�Envelope�Item)�TRANSACTION_SOURCE_ROUTE�TRANSACTION_SOURCE_VIEW� TRANSACTION_SOURCE_COMPONENT�TRANSACTION_SOURCE_TASK)� TYPE_CHECKING)�Any)�Callable)�Dict)� Generator)�Iterable)�List)�Optional)�Set)�Tuple)�Union)� BucketKey)� DurationUnit)�FlushedMetricValue)�MeasurementUnit)� MetricMetaKey)�MetricTagValue)� MetricTags)�MetricTagsInternal)� MetricType)� MetricValuez�The sentry_sdk.metrics module is deprecated and will be removed in the next major release. Sentry will reject all metrics sent after October 7, 2024. Learn more: https://sentry.zendesk.com/hc/en-us/articles/26369339769883-Upcoming-API-Changes-to-Metrics� )� stacklevel� in_metricsF)�defaultz[^a-zA-Z0-9_]+� z[^a-zA-Z0-9_\-.]+�_z[^a-zA-Z0-9_\-.\/]+c C s" t �ddddddd��}| �|�S )Nz\nz\rz\tz\\z\u{7c}z\u{2c})� � � �\�|�,)�str� maketrans� translate)�value�table� r; �@/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/metrics.py�_sanitize_tag_valueS s �� r= c C s4 zt �| �}W n tk r$ Y d S X t|ddd�S )NFT)Zinclude_local_variablesZinclude_source_context)�sys� _getframe� Exceptionr )r+ Zfrmr; r; r< �get_code_locationb s �rA c c s. t �� } t �d� z | V W 5 t �| � X dS )z5Enters recursion protection and returns the old flag.TN)� _in_metrics�get�set)Zold_in_metricsr; r; r< �recursion_protectionn s rE c s t � �� fdd��}|S )zYConvenient decorator that uses `recursion_protection` to make a function a noop. c s0 t � � }|s"� | |�W 5 Q R � S W 5 Q R X d S �N)rE )�args�kwargsr, ��funcr; r< �new_func� s z metrics_noop.<locals>.new_func)r )rJ rK r; rI r< � metrics_noopz s rL c @ sD e Zd ZdZedd� �Zeedd� ��Zedd� �Zedd � �Z d S ) �Metricr; c C s d S rF r; ��self�firstr; r; r< �__init__� s zMetric.__init__c C s d S rF r; �rO r; r; r< �weight� s z Metric.weightc C s d S rF r; �rO r9 r; r; r< �add� s z Metric.addc C s d S rF r; rR r; r; r< �serialize_value� s zMetric.serialize_valueN) �__name__� __module__� __qualname__� __slots__r rQ �propertyrS rU rV r; r; r; r< rM � s rM c @ s4 e Zd ZdZdd� Zedd� �Zdd� Zdd � Zd S ) � CounterMetric�r9 c C s t |�| _d S rF ��floatr9 rN r; r; r< rQ � s zCounterMetric.__init__c C s dS �N� r; rR r; r; r< rS � s zCounterMetric.weightc C s | j t|�7 _ d S rF )r9 r_ rT r; r; r< rU � s zCounterMetric.addc C s | j fS rF r] rR r; r; r< rV � s z CounterMetric.serialize_valueN� rW rX rY rZ rQ r[ rS rU rV r; r; r; r< r\ � s r\ c @ s4 e Zd ZdZdd� Zedd� �Zdd� Zdd � Zd S ) � GaugeMetric��last�min�max�sum�countc C s* t |�}|| _|| _|| _|| _d| _d S r` �r_ re rf rg rh ri rN r; r; r< rQ � s zGaugeMetric.__init__c C s dS )N� r; rR r; r; r< rS � s zGaugeMetric.weightc C sJ t |�}|| _t| j|�| _t| j|�| _| j|7 _| jd7 _d S r` rj rT r; r; r< rU � s zGaugeMetric.addc C s | j | j| j| j| jfS rF rd rR r; r; r< rV � s �zGaugeMetric.serialize_valueNrb r; r; r; r< rc � s rc c @ s4 e Zd ZdZdd� Zedd� �Zdd� Zdd � Zd S ) �DistributionMetricr] c C s t |�g| _d S rF r^ rN r; r; r< rQ � s zDistributionMetric.__init__c C s t | j�S rF ��lenr9 rR r; r; r< rS � s zDistributionMetric.weightc C s | j �t|�� d S rF )r9 �appendr_ rT r; r; r< rU � s zDistributionMetric.addc C s | j S rF r] rR r; r; r< rV s z"DistributionMetric.serialize_valueNrb r; r; r; r< rl � s rl c @ s4 e Zd ZdZdd� Zedd� �Zdd� Zdd � Zd S ) � SetMetricr] c C s |h| _ d S rF r] rN r; r; r< rQ s zSetMetric.__init__c C s t | j�S rF rm rR r; r; r< rS s zSetMetric.weightc C s | j �|� d S rF )r9 rU rT r; r; r< rU s z SetMetric.addc s dd� � � fdd�| j D �S )Nc S s& t | t�r t�| �d��d@ S t| �S )N�utf-8l �� )� isinstancer6 �zlib�crc32�encode�int)�xr; r; r< �_hash s z(SetMetric.serialize_value.<locals>._hashc 3 s | ]}� |�V qd S rF r; )�.0r9 �rx r; r< � <genexpr>! s z,SetMetric.serialize_value.<locals>.<genexpr>r] rR r; rz r< rV s zSetMetric.serialize_valueNrb r; r; r; r< rp s rp c C sB t �� }|j}| D �]$\}}|�� D �]\}}|\}}} } t|�}t| �} ||�d�� |d� || �d�� |�� D ] } |d� |t| ��d�� qv|d� ||�d�� | �r|d� d} | D ]P\} }t | �} | s�q�| r�d} n|d � || �d�� |d� |t |��d�� q�|d � |t|��d�� |d � q$q|� � S ) Nrq � @� :� |�asciis |#TF� ,s |T� ) �io�BytesIO�write�items�_sanitize_metric_key�_sanitize_unitru rV r6 �_sanitize_tag_keyr= �getvalue)�flushable_buckets�out�_write� timestamp�buckets� bucket_key�metric� metric_typeZ metric_nameZ metric_unitZ metric_tagsZserialized_valuerP Ztag_key� tag_valuer; r; r< �_encode_metrics$ s@ r� c C sZ i }|D ]B\}}|\}}}d� |t|�t|��}d|d< |�|g ��|� qt| |d��S )Nz{}:{}@{}�location�type)r� �mapping)�formatr� r� � setdefaultro r ) r� �code_locationsr� �key�locr� �name�unitZmrir; r; r< �_encode_locationsT s �r� )�c�g�d�sc C s t � d S )Ng @�@�r r; r; r; r< �<lambda>n � r� c C s t � d S )Ng ��.Ar� r; r; r; r< r� o r� c C s t � d S )Ng N@�r r; r; r; r< r� q r� c C s t � d S )N� �@r� r; r; r; r< r� r r� c C s t � d d S )Nr� � 8@r� r; r; r; r< r� s r� c C s t � d d d S )Nr� r� g @r� r; r; r; r< r� t r� )Z nanosecond� microsecondZ millisecond�second�minute�hour�day�weekc @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �LocalAggregator�� _measurementsc C s i | _ d S rF r� rR r; r; r< rQ { s �zLocalAggregator.__init__c C s� d|||f }||f}| j �|�}|d k r\|\} } } } t| |�} t| |�} | d7 } | |7 } n| } } } d} | | | | f| j |< d S )Nz%s:%s@%sra )r� rC rf rg ) rO �tyr� r9 r� �tags� export_keyr� �old�v_min�v_max�v_count�v_sumr; r; r< rU � s zLocalAggregator.addc C sJ i }| j �� D ]6\\}}\}}}}|�|g ��t|�||||d�� q|S )N)r� rf rg ri rh )r� r� r� ro � _tags_to_dict)rO �rvr� r� r� r� r� r� r; r; r< �to_json� s � �� zLocalAggregator.to_jsonN)rW rX rY rZ rQ rU r� r; r; r; r< r� x s r� c @ s� e Zd ZdZdZdZd!dd�Zdd� Zd d � Zd d � Z d d� Z dd� Z e d"dd��Z d#dd�Ze dd� �Zdd� Ze dd � �Zd d � Zd d � ZdS )$�MetricsAggregatorg $@i�� g @Fc C sf i | _ || _t� | _i | _d| _|| _d| _t� � | _ t� � | _ d| _ t�� | j | _d | _d | _d S )Nr TF)r� �_enable_code_locations�_set�_seen_locations�_pending_locations�_buckets_total_weight� _capture_func�_running� threading�Lock�_lock�Event� _flush_event� _force_flush�random�ROLLUP_IN_SECONDS� _flush_shift�_flusher� _flusher_pid)rO Z capture_funcZenable_code_locationsr; r; r< rQ � s zMetricsAggregator.__init__c C s� | j s dS t�� }| j|kr dS | j�r | j|kr@W 5 Q R � dS || _tj| jd�| _d| j_ z| j� � W n& t k r� d| _ Y W 5 Q R � dS X W 5 Q R X dS )z�For forking processes we might need to restart this thread. This ensures that our process actually has that thread running. FT)�target) r� �os�getpidr� r� r� �Thread� _flush_loopr� �daemon�start� RuntimeError)rO �pidr; r; r< �_ensure_thread� s" z MetricsAggregator._ensure_threadc C s8 t �d� | js| jr4| jr*| j�| j� | �� q d S �NT)rB rD r� r� r� �wait�FLUSHER_SLEEP_TIME�_flushrR r; r; r< r� � s z MetricsAggregator._flush_loopc C s | � | �� | �� � d S rF )�_emit�_flushable_buckets�_flushable_locationsrR r; r; r< r� � s zMetricsAggregator._flushc C s� | j �� | j}t�� | j | j }d}d}|rL| j�� }i | _d| _d| _njg }| j�� D ] \}}||krZ|�||f� qZ|D ](\}}|� � D ]}||j 7 }q�| j|= q~| j|8 _W 5 Q R X |S )Nr; r F) r� r� �timer� r� r� r� r� ro �valuesrS )rO Z force_flush�cutoffr� Zweight_to_removeZbuckets_timestampr� r� r; r; r< r� � s( z$MetricsAggregator._flushable_bucketsc C s" | j � | j}i | _W 5 Q R X |S rF )r� r� )rO � locationsr; r; r< r� s z&MetricsAggregator._flushable_locationsNr c C s$ | � � r| jd krd S |d kr(t�� }nt|t�r:t|�}t|| j | j �} t|�} |||| f} | j �x | j � | i �} | � | �} | d k r�| j }| �|� nt| |� } | | < d}| j | }|d k r�| �||||d |� W 5 Q R X | �� |d k �r t|dk�r|n|�}|�||||| � d S )Nr r* r� )r� r� r� rr r r rv r� �_serialize_tagsr� r� r� rC rS rU � METRIC_TYPES�record_code_location�_consider_force_flushr_ )rO r� r� r9 r� r� r� �local_aggregatorr+ Zbucket_timestampZserialized_tagsr� Z local_bucketsr� Zprevious_weight�addedZ local_valuer; r; r< rU " s< �� zMetricsAggregator.addc C s� | j s d S |d krt�� }|||f}t�|tj�jddddd d�}tt|��}||f| j kr�| j � ||f� t |d �}|d k r�| j � |g ��||f� d S )Nr �r� r� r� r� �tzinfo� )r� r� r � fromtimestampr �utc�replacerv r r� rU rA r� r� ro ) rO r� r� r� r+ r� �meta_key� start_of_dayr� r; r; r< r� Z s( � �z&MetricsAggregator.record_code_locationc C sL | j r dS |||f}t�|tj�jddddd d�}tt|��}||f| jkS )NFr r� ) r� r r� r r� r� rv r r� )rO r� r� r� r� r� r� r; r; r< �need_code_locationw s � z$MetricsAggregator.need_code_locationc C s( | j d krd S d| _| j�� d | _ d S )NF)r� r� r� rD rR r; r; r< �kill� s zMetricsAggregator.killc C s d| _ | �� d S r� )r� r� rR r; r; r< �flush� s zMetricsAggregator.flushc C s. t | j�| j }|| jkr*d| _| j�� d S r� )rn r� r� � MAX_WEIGHTr� r� rD )rO Z total_weightr; r; r< r� � s z'MetricsAggregator._consider_force_flushc C sj t � }|r$t|�}|�t|dd�� |�� D ]$\}}t||�}|�t|dd�� q,|jrf| �|� |S d S )NZstatsd)�payloadr� Z metric_meta)r r� Zadd_itemr r� r� r� )rO r� r� �envelopeZencoded_metricsr� r� Zencoded_locationsr; r; r< r� � s zMetricsAggregator._emit)F)NNr )N)rW rX rY r� r� r� rQ r� r� r� r� r� rL rU r� r� r� r� r� r� r; r; r; r< r� � s0 � ! � = � r� c C sx | sdS g }| � � D ]V\}}t|ttf�rP|D ] }|d k r.|�|t|�f� q.q|d k r|�|t|�f� qtt|��S )Nr; )r� rr �list�tuplero r6 �sorted)r� r� r� r9 Z inner_valuer; r; r< r� � s r� c C sT i }| D ]F\}}|� |�}|d k rFt|t�r8|�|� qN||g||< q|||< q|S rF )rC rr r� ro )r� r� Ztag_namer� � old_valuer; r; r< r� � s r� c C s$ t �� } | �� r | jd k r | jS d S rF )� sentry_sdk� get_client� is_active�metrics_aggregator)�clientr; r; r< �_get_aggregator� s ���r c C s t �� }|�� r|jd kr$d d |fS t|p,d�}|�d|jd � |�d|jd � t �� }d }|j� d�}|t kr�|j } | r�|�d| � |j d k r�|j � � }|j� di �} | � d�} | d k r�t� �.} | s�| | |||�s�d d |fW 5 Q R � S W 5 Q R X |j||fS )Nr; �release� environment�sourceZ transactionZ _experimentsZbefore_emit_metric)r� r� r� r �dictr� �optionsZget_current_scopeZ_transaction_inforC �GOOD_TRANSACTION_SOURCESZ _transaction�_spanZ_get_local_aggregatorrE ) r� r9 r� r� r Z updated_tags�scoper� Ztransaction_sourceZtransaction_nameZ experimentsZbefore_emit_callbackr, r; r; r< � _get_aggregator_and_update_tags� s. r � �?�nonec C s8 t | |||�\}}}|dk r4|�d| ||||||� dS )zIncrements a counter.Nr� �r rU �r� r9 r� r� r� r+ � aggregatorr� r; r; r< � increment s" � �r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �_Timingc C s4 || _ || _|| _|| _|| _d | _d | _|| _d S rF )r� r� r� r9 r� �enteredr r+ )rO r� r� r� r9 r� r+ r; r; r< rQ s z_Timing.__init__c C s | j d k rtd| ��d S )Nz0cannot use timing as %s when a value is provided)r9 � TypeError)rO �contextr; r; r< �_validate_invocation2 s �z _Timing._validate_invocationc C s� t | j � | _| �d� tjd| jd�| _| jrt| j� � D ]8\}}t |t t f�rdd� ttt|���}| j�||� q:| j�� t� }|d k r�|�d| j| j| j� | S )Nzcontext-managerz metric.timing)�opr� r5 r� )�TIMING_FUNCTIONSr� r r r� Z start_spanr� r r� r� rr r� r� �joinr� �mapr6 Zset_tag� __enter__r r� r+ )rO r� r9 r r; r; r< r 9 s z_Timing.__enter__c C s| | j std��t| j| j| j| j�\}}}|d k rbt| j � | j }|� d| j|| j|| j |d � | j � |||� d | _ d S )Nz did not enterr� ) r �AssertionErrorr r� r9 r� r� r r rU r� �__exit__)rO �exc_type� exc_value�tbr r� r� �elapsedr; r; r< r L s* � � z_Timing.__exit__c s$ �� d� t� �� �fdd��}|S )N� decoratorc sB t �j�j�j�j�jd d�� � | |�W 5 Q R � S Q R X d S )Nra )r� r� r� r� r+ )�timingr� r� r� r� r+ )rG rH ��frO r; r< � timed_funci s �z$_Timing.__call__.<locals>.timed_func)r r )rO r% r& r; r$ r< �__call__e s z_Timing.__call__N)rW rX rY rQ r r r r' r; r; r; r< r s r r� c C sN |dk r<t | |||�\}}}|dk r<|�d| ||||||� t| |||||�S )a# Emits a distribution with the time it takes to run the given code block. This method supports three forms of invocation: - when a `value` is provided, it functions similar to `distribution` but with - it can be used as a context manager - it can be used as a decorator Nr� )r rU r r r; r; r< r# x s& � �r# c C s8 t | |||�\}}}|dk r4|�d| ||||||� dS )zEmits a distribution.Nr� r r r; r; r< � distribution� s" � �r( c C s8 t | |||�\}}}|dk r4|�d| ||||||� dS )z Emits a set.Nr� r r r; r; r< rD � s" � �rD c C s8 t | |||�\}}}|dk r4|�d| ||||||� dS )zEmits a gauge.Nr� r r r; r; r< �gauge� s" � �r) )r r NNr )Nr� NNr )r NNr )r NNr )r NNr )_r� r� r� �rer> r� r� �warningsrs �abcr r � contextlibr r r � functoolsr r r� Zsentry_sdk.utilsr r r r r r Zsentry_sdk.enveloper r Zsentry_sdk.tracingr r r r �typingr r r r r r r r r r r Zsentry_sdk._typesr r! r" r# r$ r% r& r' r( r) �warn�DeprecationWarningrB rD r� � frozensetr �compile�subr� r� r� r= rA rE rL rM r\ rc rl rp r� r� r� r r� r� r� r� r r r �incrr r# r( r) r; r; r; r< �<module> s� � �� 0 0� � 6 $ � \ � � � �