관리-도구
편집 파일: local_counters.cpython-311.pyc
� b��fz � �N � U d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZm Z d dl m Z e j e � � Ze G d� d� � � � Z e� � Zeeef ed< ej � � Zedd ed e eddf fd �� � Zd � Zd� Zdd�Zd edefd�Zd efd�ZdS )� N)�contextmanager)� dataclass)�Dict� Generator)�request_data_storagec �H � e Zd ZU dZej ed< eed< ed� � � Z dS )�TaskCounterStoragezH Typing that has lock and value storage with some syntax sugar. �lock�next_request_idc � � | j dz S )N� )r )�selfs �S/opt/cloudlinux/venv/lib64/python3.11/site-packages/xray/internal/local_counters.py�processed_requestsz%TaskCounterStorage.processed_requests s � ��#�a�'�'� N) �__name__� __module__� __qualname__�__doc__� threading�Lock�__annotations__�int�propertyr � r r r r sU � � � � � � �� � �.�������� �(� (� �X�(� (� (r r �_request_id_storageF� fake_task_id�returnc # � K � t � d| � � t | � � }|j 5 |V � |r+t � d| � � t | |� � ddd� � dS # 1 swxY w Y 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 %sz)Updating task %s requests counter in fileN)�logger�debug�_get_or_create_recordr �info�_save_data_to_file)r �flush�storages r �open_local_storager' $ s� � � � � �L�L�%�|�4�4�4�#�L�1�1�G� � � 6� 6�� � � � � 6� �K�K�C�\� R� R� R� �|�W� 5� 5� 5� 6� 6� 6� 6� 6� 6� 6� 6� 6� 6� 6� 6���� 6� 6� 6� 6� 6� 6s �2A3�3A7�:A7c � � t � d| � � t 5 | t v rt | = ddd� � dS # 1 swxY w Y 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 s r �remove_local_storager* 7 s� � � �K�K�5�|�D�D�D� � 2� 2� �.� .� .�#�L�1�2� 2� 2� 2� 2� 2� 2� 2� 2� 2� 2� 2���� 2� 2� 2� 2� 2� 2s �A�A� Ac �* � t t � � S )�8 List all fake task ids saved in local storage. )�listr r r r � get_task_idsr. C s � � �#� $� $�$r Tc � � t t � � � � � D ]^}t � d|� � t |� � 5 }t ||� � | rt |= ddd� � n # 1 swxY w Y �_dS )r, zFlushing task id %s on diskN)r- r �keysr r# r' r$ )�remover r& s r �flush_memory_storager2 J s� � � � 0� 5� 5� 7� 7�8�8� 6� 6� �� � �1�<�@�@�@� � � -� -� 6�� �|�W� 5� 5� 5�� 6�'� �5� 6� 6� 6� 6� 6� 6� 6� 6� 6� 6� 6���� 6� 6� 6� 6��6� 6s �A:�:A> �A> r& c �� � t j � t | � � }t |d� � 5 }|� t |j � � � � ddd� � dS # 1 swxY w Y dS )z1 Saves storage data from memory to file. �wN)�os�path�joinr �open�write�strr �r r& � req_id_file�fs r r$ r$ V s� � � �'�,�,� 3�\�B�B�K� �k�3� � � .�1� ����G�+�,�,�-�-�-�.� .� .� .� .� .� .� .� .� .� .� .���� .� .� .� .� .� .s �(A+�+A/�2A/c �� � t 5 | t vr�t t j � � d�� � t | < t | }t j � t | � � } t |d� � 5 }t |� � � � � |_ ddd� � n # 1 swxY w Y n # t $ r Y nw xY wt | }ddd� � n # 1 swxY w Y |S )zO Takes record from local storage or creates new one and returns object r )r r �rN)r) r r r r r5 r6 r7 r r8 r �readr �FileNotFoundErrorr; s r r" r" _ sf � � � 8� 8� � 2� 2� 2�0B� �^�%�%� !�1� 1� 1� � � -� *�,�7�G� �'�,�,�';�\�J�J�K� ��+�s�+�+� <�q�.1�!�&�&�(�(�m�m�G�+�<� <� <� <� <� <� <� <� <� <� <���� <� <� <� <���$� � � ��� ���� *�,�7�G�!8� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8���� 8� 8� 8� 8�" �NsZ �A&C'�/B>�?'B2�& B>�2B6 �6B>�9B6 �:B>�=C'�> C �C'� C � C'�'C+�.C+)F)T) �loggingr5 r � contextlibr � dataclassesr �typingr r �xray.internal.constantsr � getLoggerr r r �dictr r: r r r) r' r* r. r2 r$ r" r r r �<module>rI s� �� ����� � � � � � � � � %� %� %� %� %� %� !� !� !� !� !� !� "� "� "� "� "� "� "� "� 8� 8� 8� 8� 8� 8� �� �8� $� $�� � (� (� (� (� (� (� (� �� (� 6:�T�V�V� �T�#� 1�1�2� ;� ;� ;�%�y�~�'�'� � �6� 6�S� 6�)�DV�X\�^b�Db�:c� 6� 6� 6� � �6�$ 2� 2� 2�%� %� %� 6� 6� 6� 6�.�S� .�3E� .� .� .� .��+=� � � � � � r