관리-도구
편집 파일: transaction_profiler.cpython-38.pyc
U �Mg�l � @ sV d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z ddl mZ ddlZddlmZ ddlmZmZ ddlmZmZmZmZmZmZm Z ddl m Z e �r�dd l m Z dd l m Z dd l m!Z! dd l m"Z" dd l m#Z# ddl m$Z$ ddl m%Z% ddl m&Z& ddl'm(Z( ddlm)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/ ddl0m1Z1m2Z2m3Z3 e(de4e.e5d��Z6e(de#e* e#e) e#e6 e"e.e+f d��Z7z*ddl8m9Z9 ddl:m;Z< e<Z;e9dd�Z=W n e>k �r� ej?Z=dZ;Y nX da@d ZAd d � ZBd d � ZCd!d"� ZDe5d#�ZEG d$d%� d%�ZFG d&d'� d'e �ZGG d(d)� d)eG�ZHG d*d+� d+eG�ZIdS ),a� This file is originally based on code from https://github.com/nylas/nylas-perftools, which is published under the following license: The MIT License (MIT) Copyright (c) 2014 Nylas Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions: The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. � N)�ABC�abstractmethod)�deque)�LRUCache)�DEFAULT_SAMPLING_FREQUENCY� extract_stack)�capture_internal_exception�get_current_thread_meta� is_gevent�is_valid_sample_rate�logger�nanosecond_time�set_in_app_in_frames)� TYPE_CHECKING)�Any)�Callable)�Deque)�Dict)�List)�Optional)�Set)�Type)� TypedDict)�ProcessedStack�ProcessedFrame�ProcessedThreadMetadata�FrameId�StackId�ThreadId�ExtractedSample)�Event�SamplingContext� ProfilerMode�ProcessedSample��elapsed_since_start_nsZ thread_id�stack_id�ProcessedProfile��frames�stacks�samples�thread_metadata)� get_original)� ThreadPool�time�sleep� c C s` | d }|d k rdS | d }|d k r0|dkr0dS | d � d�}|d k r\t�d� |dkr\dS dS )N�profiles_samplerT�profiles_sample_rater � _experimentszy_experiments['profiles_sample_rate'] is deprecated. Please use the non-experimental profiles_sample_rate option directly.F)�getr �warning)�optionsr2 r3 � r8 �V/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/profiler/transaction_profiler.py�has_profiling_enabledx s �r: c C s� t d k rt�d� dS t}t� r(tj}ntj}| �d�d k rF| d }n,| �di ��d�}|d k rjt� d� |pp|}|tjks�|dkr�t|d�a n$|tjkr�t|d�a nt d� |���t�d j t jd �� t � � t �t� d S ) Nz%[Profiling] Profiler is already setupF� profiler_moder4 zk_experiments['profiler_mode'] is deprecated. Please use the non-experimental profiler_mode option directly.r0 �� frequencyzUnknown profiler mode: {}z.[Profiling] Setting up profiler in {mode} mode)�modeT)� _schedulerr �debugr r �GeventSchedulerr> �ThreadSchedulerr5 r6 � ValueError�format�setup�atexit�register�teardown_profiler)r7 r= Zdefault_profiler_moder; r8 r8 r9 �setup_profiler� s: ��� � rI c C s t d k rt �� d a d S �N)r? �teardownr8 r8 r8 r9 rH � s rH g ���Bc @ s� e Zd Zddd�Zdd� Zdd� Zdd � Zd d � Zd d � Zdd� Z dd� Z dd� Z dd� Z dd� Z edd� �Zejdd� �ZdS ) �ProfileNc C s� |d kr t n|| _t�� j| _|| _t� d p0d| _d | _ z || _ W n t k r^ d| _ Y nX d| _ d| _ i | _i | _g | _g | _g | _d| _d | _|d k r�|| _tjdtdd� d S )Nr Fz8The `hub` parameter is deprecated. Please do not use it.r1 �� stacklevel)r? � scheduler�uuid�uuid4�hex�event_id�sampledr �_default_active_thread_id�active_thread_id�start_ns�AttributeError�stop_ns�active�indexed_frames�indexed_stacksr) r* r+ �unique_samples�_hub�warnings�warn�DeprecationWarning)�selfrT rW �hubrO r8 r8 r9 �__init__� s2 �zProfile.__init__c C s$ t � d | _t�dj| jd�� d S )Nr z.[Profiling] updating active thread id to {tid})�tid)r rV r r@ rD �rb r8 r8 r9 �update_active_thread_id� s ��z Profile.update_active_thread_idc C s | j st�d� d| _ dS | jdkr8t�d� d| _ dS t�� }|�� sRd| _ dS |j}t|� d��rt|d |�}n$|d dk r�|d }n|d � d�}|dkr�t�d� d| _ dS t |d d �s�t� d � d| _ dS t � � t |�k | _ | j r�t�d � nt�d jt |�d�� dS )a Sets the profile's sampling decision according to the following precedence rules: 1. If the transaction to be profiled is not sampled, that decision will be used, regardless of anything else. 2. Use `profiles_sample_rate` to decide. z@[Profiling] Discarding profile because transaction is discarded.FNz@[Profiling] Discarding profile because profiler was not started.r2 r3 r4 zA[Profiling] Discarding profile because profiling was not enabled.Z Profiling)�sourcez>[Profiling] Discarding profile because of invalid sample rate.z [Profiling] Initializing profilezk[Profiling] Discarding profile because it's not included in the random sample (sample rate = {sample_rate}))� sample_rate)rT r r@ rO � sentry_sdk� get_client� is_activer7 �callabler5 r r6 �random�floatrD )rb Zsampling_context�clientr7 ri r8 r8 r9 � _set_initial_sampling_decision sT � � � � ��z&Profile._set_initial_sampling_decisionc C sL | j r | jrd S | js td��t�d� d| _| js<t� | _| j�| � d S )N�No scheduler specifiedz [Profiling] Starting profileT) rT rZ rO �AssertionErrorr r@ rW r �start_profilingrf r8 r8 r9 �startP s z Profile.startc C s: | j r | jsd S | js td��t�d� d| _t� | _d S )Nrr z [Profiling] Stopping profileF)rT rZ rO rs r r@ r rY rf r8 r8 r9 �stop\ s z Profile.stopc C s* t �� }|j}| |_||f| _| �� | S rJ )rj Zget_isolation_scope�profile�_context_manager_stateru )rb �scope� old_profiler8 r8 r9 � __enter__f s zProfile.__enter__c C s | � � | j\}}| `||_d S rJ )rv rx rw )rb �ty�value�tbry rz r8 r8 r9 �__exit__r s zProfile.__exit__c s � j s d S |� jk rd S |� j }|tkr6� �� d S � jd7 _t|�}|D ]�\}\}}}z�|� jkr�t|�D ]2\} } | � jkrrt � j�� j| < � j � || � qrt � j�� j|< � j � � fdd�|D �� � j � ||� j| d�� W qP tk �r tt�� � Y qPX qPd S )N� c s g | ]}� j | �qS r8 )r[ )�.0�frame_idrf r8 r9 � <listcomp>� s z!Profile.write.<locals>.<listcomp>r$ )rZ rW �MAX_PROFILE_DURATION_NSrv r] �strr\ � enumerater[ �lenr) �appendr* r+ rX r �sys�exc_info) rb �ts�sample�offsetr% re r& Z frame_idsr) �ir� r8 rf r9 �write{ s: ���z Profile.writec C s& dd� t �� D �}| j| j| j|d�S )Nc S s" i | ]}t |j�d t |j�i�qS )�name)r� �identr� )r� �threadr8 r8 r9 � <dictcomp>� s � �z#Profile.process.<locals>.<dictcomp>r( )� threadingr� r) r* r+ )rb r, r8 r8 r9 �process� s ��zProfile.processc C s� | � � }t|d |d |d |d � |�d�| jd||�dd�|d d d t�� it�� t�� d �t�� t� � d �|d |d dt | j | j �|d d d t | j d kr�| jn| j �d�gd� S )Nr) Zin_app_excludeZin_app_includeZ project_root� environment�python�release� Zstart_timestamp�1� architecture)r� �versionrS Z transaction�0Zcontexts�trace�trace_id)�idr� Zrelative_start_nsZrelative_end_nsr� rV ) r� rS �platformrw r� � timestampr� Zdevice�os�runtimeZ transactions)r� r r5 rS r� �machine�systemr� �python_implementation�python_versionr� rY rW rV rU )rb Z event_optr7 rw r8 r8 r9 �to_json� sH � ��������zProfile.to_jsonc C s~ t �� }|�� sdS t|j�s"dS | jd ks2| jsL|jrH|jjddd� dS | jt k rz|jrl|jjddd� t � d� dS dS )NFri rw )Z data_categoryZinsufficient_dataz<[Profiling] Discarding profile because insufficient samples.T) rj rk rl r: r7 rT � transportZrecord_lost_eventr] �PROFILE_MINIMUM_SAMPLESr r@ )rb rp r8 r8 r9 �valid� s* � � z Profile.validc C s t jdtdd� | jS )Nz;The `hub` attribute is deprecated. Please do not access it.r1 rM �r_ r` ra r^ rf r8 r8 r9 rc s �z Profile.hubc C s t jdtdd� || _d S )Nz8The `hub` attribute is deprecated. Please do not set it.r1 rM r� )rb r} r8 r8 r9 rc s �)NN)�__name__� __module__� __qualname__rd rg rq ru rv r{ r r� r� r� r� �propertyrc �setterr8 r8 r8 r9 rL � s" � - I ,3 rL c @ sX e Zd ZdZdd� Zdd� Zdd� Zedd � �Zed d � �Z d d � Z dd� Z dd� Z dS )� Scheduler�unknownc C s, d| | _ | �� | _tdd�| _t� | _d S )Ng �?� )�maxlen)�interval� make_sampler�samplerr � new_profiles�set�active_profiles�rb r= r8 r8 r9 rd s zScheduler.__init__c C s | � � | S rJ )rE rf r8 r8 r9 r{ * s zScheduler.__enter__c C s | � � d S rJ )rK )rb r| r} r~ r8 r8 r9 r / s zScheduler.__exit__c C s d S rJ r8 rf r8 r8 r9 rE 3 s zScheduler.setupc C s d S rJ r8 rf r8 r8 r9 rK 8 s zScheduler.teardownc C s dS )z� Ensure the scheduler is running. By default, this method is a no-op. The method should be overridden by any implementation for which it is relevant. Nr8 rf r8 r8 r9 �ensure_running= s zScheduler.ensure_runningc C s | � � | j�|� d S rJ )r� r� r� )rb rw r8 r8 r9 rt F s zScheduler.start_profilingc s& t �� �tdd�� � ��fdd�}|S )N� )�max_sizec s� �j s�jsdS t�j �}t� }z � �fdd�t�� �� D �}W n" tk rb tt� � � Y dS X t |�D ]}�j� �j � � � qlg }�jD ]"}|j r�|�||� q�|�|� q�|D ]}�j�|� q�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. Nc s$ g | ] \}}t |�t|� ��f�qS r8 )r� r )r� re �frame)�cache�cwdr8 r9 r� j s �zAScheduler.make_sampler.<locals>._sample_stack.<locals>.<listcomp>)r� r� r� r r� �_current_frames�itemsrX r r� �range�add�popleftrZ r� r� �remove)�args�kwargsr� �nowr� �_Zinactive_profilesrw �r� r� rb r8 r9 � _sample_stackQ s( � z-Scheduler.make_sampler.<locals>._sample_stack)r� �getcwdr )rb r� r8 r� r9 r� K s ?zScheduler.make_samplerN) r� r� r� r> rd r{ r r rE rK r� rt r� r8 r8 r8 r9 r� s r� c sH e Zd ZdZdZdZ� fdd�Zdd� Zdd � Zd d � Z d d � Z � Z S )rB zr This scheduler is based on running a daemon thread that will call the sampler at a regular interval. r� z sentry.profiler.ThreadSchedulerc s. t � j|d� d| _d | _d | _t�� | _d S )Nr< F)�superrd �runningr� �pidr� �Lock�lockr� �� __class__r8 r9 rd � s zThreadScheduler.__init__c C s d S rJ r8 rf r8 r8 r9 rE � s zThreadScheduler.setupc C s$ | j r d| _ | jd k r | j�� d S �NF�r� r� �joinrf r8 r8 r9 rK � s zThreadScheduler.teardownc C s� t �� }| jr | j|kr dS | j�� | jrB| j|krBW 5 Q R � dS || _d| _tj| j| jdd�| _ z| j � � W n, t k r� d| _d| _ Y W 5 Q R � dS X W 5 Q R X dS )aE Check that the profiler has an active thread to run in, and start one if that's not the case. Note that this might fail (e.g. in Python 3.12 it's not possible to spawn new threads at interpreter shutdown). In that case self.running will be False after running this function. NT)r� �target�daemonF) r� �getpidr� r� r� r� �Threadr� �runr� ru � RuntimeError�rb r� r8 r8 r9 r� � s z ThreadScheduler.ensure_runningc C sH t �� }| jrD| �� t �� | }|| jk r:t| j| � t �� }qd S rJ �r/ � perf_counterr� r� r� � thread_sleep�rb �last�elapsedr8 r8 r9 r� � s zThreadScheduler.run� r� r� r� �__doc__r> r� rd rE rK r� r� � __classcell__r8 r8 r� r9 rB � s 'rB c sH e Zd ZdZdZdZ� fdd�Zdd� Zdd � Zd d � Z d d � Z � Z S )rA 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. Zgeventz sentry.profiler.GeventSchedulerc sF t d krtd�| j���t� j|d� d| _d | _d | _t � � | _ d S )Nz"Profiler mode: {} is not availabler< F) r. rC rD r> r� rd r� r� r� r� r� r� r� r� r8 r9 rd � s zGeventScheduler.__init__c C s d S rJ r8 rf r8 r8 r9 rE s zGeventScheduler.setupc C s$ | j r d| _ | jd k r | j�� d S r� r� rf r8 r8 r9 rK s zGeventScheduler.teardownc C s� t �� }| jr | j|kr d S | j�z | 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 )NTr� F) r� r� r� r� r� r. r� �spawnr� r� r� r8 r8 r9 r� s z GeventScheduler.ensure_runningc C sH t �� }| jrD| �� t �� | }|| jk r:t| j| � t �� }qd S rJ r� r� r8 r8 r9 r� 5 s zGeventScheduler.runr� r8 r8 r� r9 rA � s rA )Jr� rF r� r� rn r� r� r/ rP r_ �abcr r � collectionsr rj Zsentry_sdk._lru_cacher Zsentry_sdk.profiler.utilsr r Zsentry_sdk.utilsr r r r r r r �typingr r r r r r r r r �typing_extensionsr r r r r r r r Zsentry_sdk._typesr r! r" r� �intr# r' Z gevent.monkeyr- Zgevent.threadpoolr. Z _ThreadPoolr� � ImportErrorr0 r? r� r: rI rH r� rL r� rB rA r8 r8 r8 r9 �<module> s~ $ $ �� �� 3 OvX