관리-도구
편집 파일: continuous_profiler.cpython-38.pyc
U �MgC � @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZ d dlm Z d dl m Z d dl m Z d dlmZmZ d dlmZmZmZmZmZ d dlmZ e�rDd d lmZ d d lmZ d d lmZ d d lm Z d d lm Z d dlm Z d dlm Z d dl m!Z! d dl"m#Z#m$Z$ d dlm%Z%m&Z&m'Z'm(Z(m)Z)m*Z* e!de+e(e,d��Z-z*d dl.m/Z/ d dl0m1Z2 e2Z1e/dd�Z3W n e4k �r� ej5Z3dZ1Y nX da6dd� Z7dd � Z8d d � Z9d d � Z:d!d"� Z;d#d$� Z<G d%d&� d&�Z=G d'd(� d(e=�Z>G d)d*� d*e=�Z?d+Z@G d,d-� d-�ZAG d.d/� d/�ZBdS )0� N)�datetime�timezone)�VERSION)�Envelope)�LRUCache)�DEFAULT_SAMPLING_FREQUENCY� extract_stack)�capture_internal_exception� is_gevent�logger�now�set_in_app_in_frames)� TYPE_CHECKING)�Any)�Callable)�Dict)�List)�Optional)�Type)�Union)� TypedDict)�ContinuousProfilerMode�SDKInfo)�ExtractedSample�FrameId�StackId�ThreadId�ProcessedFrame�ProcessedStack�ProcessedSample�� timestampZ thread_id�stack_id)� get_original)� ThreadPool�time�sleepc C s� t d k rt�d� dS t� r$tj}ntj}| �di �}|�d�pB|}t}|tjkrbt|| ||�a n(|tjkr|t|| ||�a nt d� |���t�dj t jd�� t � t � dS ) Nz0[Profiling] Continuous Profiler is already setupF� _experimentsZcontinuous_profiling_modez$Unknown continuous profiler mode: {}z9[Profiling] Setting up continuous profiler in {mode} mode)�modeT)� _schedulerr �debugr �GeventContinuousSchedulerr( �ThreadContinuousScheduler�getr � ValueError�format�atexit�register� teardown_continuous_profiler)�options�sdk_info� capture_funcZdefault_profiler_mode� experimentsZ profiler_mode� frequency� r8 �U/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/profiler/continuous_profiler.py�setup_continuous_profilerE s@ � � ��� r: c C s$ t d kr d S t �� sd S t �� d S �N)r) �should_autostart�ensure_runningr8 r8 r8 r9 �!try_autostart_continuous_profilert s r> c C s t d kr d S t �� d S r; )r) r= r8 r8 r8 r9 �start_profiler� s r? c C s t d kr d S t �� d S r; )r) �teardownr8 r8 r8 r9 � stop_profiler� s rA c C s t � d ad S r; )rA r) r8 r8 r8 r9 r2 � s r2 c C s t d kr d S t jS r; )r) � profiler_idr8 r8 r8 r9 �get_profiler_id� s rC c @ s\ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z e dd� �Z dd� Z dd� Z dS )�ContinuousScheduler�unknownc C s6 d| | _ || _|| _|| _| �� | _d | _d| _d S )Ng �?F)�intervalr3 r4 r5 � make_sampler�sampler�buffer�running��selfr7 r3 r4 r5 r8 r8 r9 �__init__� s z ContinuousScheduler.__init__c C s | j �d�}|sdS |�d�S )Nr' FZ continuous_profiling_auto_start)r3 r- )rL r6 r8 r8 r9 r< � s z$ContinuousScheduler.should_autostartc C s t �d S r; ��NotImplementedError�rL r8 r8 r9 r= � s z"ContinuousScheduler.ensure_runningc C s t �d S r; rN rP r8 r8 r9 r@ � s z ContinuousScheduler.teardownc C s t �d S r; rN rP r8 r8 r9 �pause� s zContinuousScheduler.pausec C s t | j| jt| j�| _d S r; )� ProfileBufferr3 r4 �PROFILE_BUFFER_SECONDSr5 rI rP r8 r8 r9 � reset_buffer� s �z ContinuousScheduler.reset_bufferc C s | j d krd S | j jS r; )rI rB rP r8 r8 r9 rB � s z ContinuousScheduler.profiler_idc s& t �� �tdd�� � ��fdd�}|S )N� )�max_sizec sf t � }z � �fdd�t�� �� D �}W n" tk rH tt�� � Y dS X �jdk rb�j�||� dS )z� Take a sample of the stack on all the threads in the process. This should be called at a regular interval to collect samples. c s$ g | ] \}}t |�t|� ��f�qS r8 )�strr )�.0�tid�frame)�cache�cwdr8 r9 � <listcomp>� s �zKContinuousScheduler.make_sampler.<locals>._sample_stack.<locals>.<listcomp>N) r �sys�_current_frames�items�AttributeErrorr �exc_inforI �write)�args�kwargs�ts�sample�r[ r\ rL r8 r9 � _sample_stack� s � z7ContinuousScheduler.make_sampler.<locals>._sample_stack)�os�getcwdr )rL ri r8 rh r9 rG � s z ContinuousScheduler.make_samplerc C s\ t �� }| jrD| �� t �� | }|| jk r:t| j| � t �� }q| jd k rX| j�� d S r; )r% � perf_counterrJ rH rF � thread_sleeprI �flush)rL �last�elapsedr8 r8 r9 �run� s zContinuousScheduler.runN)�__name__� __module__� __qualname__r( rM r< r= r@ rQ rT �propertyrB rG rq r8 r8 r8 r9 rD � s rD c sD e Zd ZdZdZdZ� fdd�Z� fdd�Zdd � Zd d � Z � Z S ) r, zr This scheduler is based on running a daemon thread that will call the sampler at a regular interval. �threadz)sentry.profiler.ThreadContinuousSchedulerc s, t � �||||� d | _d | _t�� | _d S r; )�superrM rv �pid� threading�Lock�lockrK �� __class__r8 r9 rM s z"ThreadContinuousScheduler.__init__c s t � �� o| jt�� kS r; �rw r< rx rj �getpidrP r| r8 r9 r< s z*ThreadContinuousScheduler.should_autostartc C s� t �� }| jr | j|kr d S | j�~ | jrB| j|krBW 5 Q R � d S || _d| _| �� tj| j| j dd�| _ z| j � � W n t k r� d| _d | _ Y nX W 5 Q R X d S )NT)�name�target�daemonF) rj r rJ rx r{ rT ry �Threadr� rq rv �start� RuntimeError�rL rx r8 r8 r9 r= s z(ThreadContinuousScheduler.ensure_runningc C s0 | j r d| _ | jd k r&| j�� d | _d | _d S �NF�rJ rv �joinrI rP r8 r8 r9 r@ B s z"ThreadContinuousScheduler.teardown) rr rs rt �__doc__r( r� rM r< r= r@ � __classcell__r8 r8 r| r9 r, s #r, c s@ e Zd ZdZdZ� fdd�Z� fdd�Zdd� Zd d � Z� Z S ) r+ as This scheduler is based on the thread scheduler but adapted to work with gevent. When using gevent, it may monkey patch the threading modules (`threading` and `_thread`). This results in the use of greenlets instead of native threads. This is an issue because the sampler CANNOT run in a greenlet because 1. Other greenlets doing sync work will prevent the sampler from running 2. The greenlet runs in the same thread as other greenlets so when taking a sample, other greenlets will have been evicted from the thread. This results in a sample containing only the sampler's code. Zgeventc sD t d krtd�| j���t� �||||� d | _d | _t� � | _ d S )Nz"Profiler mode: {} is not available) r$ r. r/ r( rw rM rv rx ry rz r{ rK r| r8 r9 rM ^ s z"GeventContinuousScheduler.__init__c s t � �� o| jt�� kS r; r~ rP r| r8 r9 r< j s z*GeventContinuousScheduler.should_autostartc C s� t �� }| jr | j|kr d S | j�� | jrB| j|krBW 5 Q R � d S || _d| _| �� td�| _z| j�| j � W n, t k r� d| _d | _Y W 5 Q R � d S X W 5 Q R X d S )NT� F) rj r rJ rx r{ rT r$ rv �spawnrq r� r� r8 r8 r9 r= n s z(GeventContinuousScheduler.ensure_runningc C s0 | j r d| _ | jd k r&| j�� d | _d | _d S r� r� rP r8 r8 r9 r@ � s z"GeventContinuousScheduler.teardown) rr rs rt r� r( rM r< r= r@ r� r8 r8 r| r9 r+ N s r+ � c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) rR c C sP || _ || _|| _|| _t�� j| _t� | _ t � | _ t � t j��� | j | _d S r; )r3 r4 � buffer_sizer5 �uuid�uuid4�hexrB � ProfileChunk�chunkr �start_monotonic_timer r �utcr! �start_timestamp)rL r3 r4 r� r5 r8 r8 r9 rM � s �zProfileBuffer.__init__c C s: | � |�r"| �� t� | _t� | _| j�| j| |� d S r; )� should_flushrn r� r� r r� rc r� )rL �monotonic_timerg r8 r8 r9 rc � s zProfileBuffer.writec C s || j | jkS r; )r� r� )rL r� r8 r8 r9 r� � s zProfileBuffer.should_flushc C s4 | j �| j| j| j�}t� }|�|� | �|� d S r; )r� �to_jsonrB r3 r4 r Zadd_profile_chunkr5 )rL r� �enveloper8 r8 r9 rn � s zProfileBuffer.flushN)rr rs rt rM rc r� rn r8 r8 r8 r9 rR � s rR c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )r� c C s. t �� j| _i | _i | _g | _g | _g | _d S r; ) r� r� r� �chunk_id�indexed_frames�indexed_stacks�frames�stacks�samplesrP r8 r8 r9 rM � s zProfileChunk.__init__c s� |D ]�\}\}}}z�|� j kr�t|�D ]2\}}|� jkr&t� j�� j|< � j�|| � q&t� j �� j |< � j�� fdd�|D �� � j�||� j | d�� W q tk r� t t � � � Y qX qd S )Nc s g | ]}� j | �qS r8 )r� )rX �frame_idrP r8 r9 r] � s z&ProfileChunk.write.<locals>.<listcomp>r ) r� � enumerater� �lenr� �appendr� r� ra r r^ rb ) rL rf rg rY r" Z frame_idsr� �ir� r8 rP r9 rc � s& ���zProfileChunk.writec C s� | j | j| jdd� t�� D �d�}t|d |d |d |d � | j|d td �d ||d d �}d D ]$}|| d k rbt|| �� � ||< qb|S )Nc S s" i | ]}t |j�d t |j�i�qS )r� )rW �identr� )rX rv r8 r8 r9 � <dictcomp>� s � �z(ProfileChunk.to_json.<locals>.<dictcomp>)r� r� r� Zthread_metadatar� Zin_app_excludeZin_app_includeZ project_rootr� )r� �version�python�2)r� Z client_sdk�platform�profilerB r� )�release� environment�dist) r� r� r� ry r� r r� r rW �strip)rL rB r3 r4 r� �payload�keyr8 r8 r9 r� � s4 �� ��� zProfileChunk.to_jsonN)rr rs rt rM rc r� r8 r8 r8 r9 r� � s r� )Cr0 rj r^ ry r% r� r r Zsentry_sdk.constsr Zsentry_sdk.enveloper Zsentry_sdk._lru_cacher Zsentry_sdk.profiler.utilsr r Zsentry_sdk.utilsr r r r r �typingr r r r r r r r �typing_extensionsr Zsentry_sdk._typesr r r r r r r r �float�intr Z gevent.monkeyr# Zgevent.threadpoolr$ Z _ThreadPoolrm � ImportErrorr& r) r: r> r? rA r2 rC rD r, r+ rS rR r� r8 r8 r8 r9 �<module> sf �� /cDL0