hjkhjkjhkhjkgjghjhgjhgjghjhghjghjhgjjnbvnvbnvk
bnmbnmbnhjkhjkhjkhjktyutyutyuytutyutyutyhgjgjghjgjghjguytutyry
/
opt
/
alt
/
python37
/
lib
/
python3.7
/
site-packages
/
xray
/
internal
/
__pycache__
/
Upload FileeE
HOME
B �'Ud5 � @ s� U d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZm Z d dl m Z eG dd� d��Z e � Zeee f ed< e�� Zedee e ddf d �d d ��Zd d� Zdd� Zddd�Zee d�dd�Ze d�dd�ZdS ) � N)�contextmanager)� dataclass)�Dict� Generator)�request_data_storagec @ s0 e Zd ZU dZejed< eed< edd� �Z dS )�TaskCounterStoragezH Typing that has lock and value storage with some syntax sugar. �lock�next_request_idc C s | j d S )N� )r )�self� r �M/opt/alt/python37/lib/python3.7/site-packages/xray/internal/local_counters.py�processed_requests s z%TaskCounterStorage.processed_requestsN) �__name__� __module__� __qualname__�__doc__� threading�Lock�__annotations__�int�propertyr r r r r r s r �_request_id_storageF)� fake_task_id�returnc c sH t �d| � t| �}|j�$ |V |r:t �d� t| |� W dQ R X dS )z� Open local task information storage. @param fake_task_id: unique string, usually obtained as task.fake_id @param flush: whether to save data to file right after update zOpening storage %szFlushing task to fileN)�logging�info�_get_or_create_recordr �_save_data_to_file)r �flush�storager r r �open_local_storage" s r! c C s. t �d| � t� | tkr t| = W dQ R X dS )zw Remove local storage record. @param fake_task_id: unique string, usually obtained as task.fake_id z#Removing memory storage for task %sN)r r �_global_storage_lockr )r r r r �remove_local_storage5 s r# c C s t t�S )z8 List all fake task ids saved in local storage. )�listr r r r r � get_task_idsA s r% Tc C sN xHt t�� �D ]8}t�d|� t|��}t||� | r<t|= W dQ R X qW dS )z8 List all fake task ids saved in local storage. zFlushing task id %s on diskN)r$ r �keysr r r! r )�remover r r r r �flush_memory_storageH s r( )r r c C s8 t j�t| �}t|d��}|�t|j�� W dQ R X dS )z1 Saves storage data from memory to file. �wN)�os�path�joinr �open�write�strr )r r � req_id_file�fr r r r T s r )r c C s� t �� | tkrxtt�� dd�t| < t| }tj�t| �}y(t |d��}t |� � �|_ W dQ R X W q� t k rt Y q�X nt| }W dQ R X |S )zO Takes record from local storage or creates new one and returns object r )r r �rN)r" r r r r r* r+ r, r r- r �readr �FileNotFoundError)r r r0 r1 r r r r ] s r )F)T)r r* r � contextlibr � dataclassesr �typingr r Zxray.internal.constantsr r �dictr r/ r r r" r! r# r% r( r r r r r r �<module> s"