관리-도구
편집 파일: ray.cpython-38.pyc
U �Mg� � @ s� d dl Z d dlZd dlZd dlmZmZ d dlmZmZ d dl m Z d dl m Z m Z mZmZmZ z d dlZW n ek r� ed��Y nX d dlZd dlmZ er�d dlmZ d d lmZmZ d d l mZ d d � Zd d� Z dd� Z G dd� de�Z dS )� N)�OP� SPANSTATUS)� DidNotEnable� Integration)�TRANSACTION_SOURCE_TASK)�event_from_exception�logger�package_version�qualname_from_function�reraisezRay not installed.)� TYPE_CHECKING)�Callable)�Any�Optional)�ExcInfoc C s t �� �� rd S t�d� d S )Nz[[Tracing] Sentry not initialized in ray cluster worker, performance data will be discarded.)� sentry_sdk� get_clientZ is_activer �debug� r r �I/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/ray.py�_check_sentry_initialized s �r c s& t j� t�� �� fdd��} | t _d S )Nc sZ t �� �r�� f||�� S d d�� fdd� }�|f||�� }|j�� �fdd�}||_|S )N)�_tracingc s� t � tj| pi tjt� �tjtd�}t� |��f}z� ||�}|� t j � W n8 t k r� |� t j� t�� }t|� t|� Y nX |W 5 Q R � S Q R X dS )z$ Ray Worker )�op�name�origin�sourceN)r r Zcontinue_tracer ZQUEUE_TASK_RAYr �RayIntegrationr r Zstart_transaction� set_statusr �OK� Exception�INTERNAL_ERROR�sys�exc_info�_capture_exceptionr )r Zf_argsZf_kwargsZ transaction�resultr" )�fr r �_f6 s$ � z1_patch_ray_remote.<locals>.new_remote.<locals>._fc s� t jtjt� �tjd���}dd� t �� �� D �}z"�| |d|i��}|� t j � W n8 t k r� |� t j � t�� }t|� t|� Y nX |W 5 Q R � S Q R X dS )z$ Ray Client )r r r c S s i | ]\}}||�qS r r )�.0�k�vr r r � <dictcomp>^ s � zi_patch_ray_remote.<locals>.new_remote.<locals>._remote_method_with_header_propagation.<locals>.<dictcomp>r N)r Z start_spanr ZQUEUE_SUBMIT_RAYr r r Zget_current_scopeZ iter_trace_propagation_headersr r r r r r! r" r# r )�args�kwargs�span�tracingr$ r" �r% Zold_remote_methodr r �&_remote_method_with_header_propagationT s$ � � zU_patch_ray_remote.<locals>.new_remote.<locals>._remote_method_with_header_propagation)�inspect�isclass�remote)r% r+ r, r& �rvr0 �Z old_remoter/ r � new_remote, s z%_patch_ray_remote.<locals>.new_remote)�rayr3 � functools�wraps)r6 r r5 r �_patch_ray_remote( s Dr: c K s6 t �� }t| |jdtjd�d�\}}t j||d� d S )NF)Zhandled�type)Zclient_optionsZ mechanism)�hint)r r r �optionsr � identifierZ capture_event)r" r, �client�eventr< r r r r# t s �� r# c @ s& e Zd ZdZde� �Zedd� �ZdS )r r7 z auto.queue.c C s8 t d�} | d kr td�| ���| dk r.td��t� d S )Nr7 zUnparsable ray version: {})� � r zRay 2.7.0 or newer required)r r �formatr: )�versionr r r � setup_once� s zRayIntegration.setup_onceN)�__name__� __module__� __qualname__r> r � staticmethodrE r r r r r � s r ) r1 r! r Zsentry_sdk.constsr r Zsentry_sdk.integrationsr r Zsentry_sdk.tracingr Zsentry_sdk.utilsr r r r r r7 � ImportErrorr8 �typingr �collections.abcr r r r r r: r# r r r r r �<module> s( L