�
8Heh
� �� � d dl Z d dlZd dlZd dlmZmZmZ ddlmZm Z ddl
m Z e j d� � Z
ded dfd
�Zded dfd �Zded ed dfd
�Zd� Zdefdeeef ded dfd�ZdS )� N)�Union�Iterable�Callable� )�switch_schedstats�no_xray_active_tasks� )�RequestProcessor�tools�signum�returnc �> � t � d| | � � dS )z/
Signal handler to log received signal
zReceived sig: %i (%s)N)�logger�info�r �frames �I/opt/cloudlinux/venv/lib64/python3.11/site-packages/ssa/internal/tools.py�log_on_receiver s! � � �K�K�'���8�8�8�8�8� c �J � t � d| � � t � � � � � t � d| � � t � � r*t � d� � t d�� � t
j d� � dS )zO
Signal handler to dump collected statistics and switch off schedstats
zReceived sig: %iz/Dumped successfully after %i signal, exiting...z0No X-Ray tasks running, switching schedstats offF)�enabledr N)r r r
� flush_bufferr r �sys�exitr s r �sigterm_receivedr s� � � �K�K�"�F�+�+�+����#�#�%�%�%�
�K�K�A�6�J�J�J� � � � )�� � �F�G�G�G��%�(�(�(�(��H�Q�K�K�K�K�Kr �
sighandlerc �� � t j | |� � t � d| � � dS # t t f$ r'}t � d| |� � Y d}~dS d}~ww xY w)z4
Register given sighandler for given signum
z
Registered %szSkipped %s because of %sN)�signalr r �OSError�
ValueError)r r �es r �register_a_signalr" ) s� � �;��
�f�j�)�)�)�� � �O�V�,�,�,�,�,�� �Z� � ;� ;� ;�� � �.���:�:�:�:�:�:�:�:�:�����;���s �04 �A,� A'�'A,c �D � t t j t � � dS )z4
Register a dump handler for SIGTERM signal
N)r" r �SIGTERMr � r r �register_sigtermr& 4 s � � �f�n�&6�7�7�7�7�7r �signals�handlerc �� � | �!t j D ]}t ||� � �dS t | t � � rt | |� � dS | D ]}t ||� � �dS )z�
Register given handler for given signal or range of signals
If no signals given, add given handler for all available signals
N)r �Signalsr" �
isinstance�int)r' r( �sigs r �register_signalsr. ; s� � � ���>� ,� ,�C� �c�7� +� +� +� +� ,� ,� �g�s� #� #� 0� �g�w� /� /� /� /� /� �
0�
0��!�#�w�/�/�/�/�
0�
0r )�loggingr r �typingr r r �utilsr r �modules.processorr
� getLoggerr r, r r r" r& r. r% r r �<module>r4 sC �� ����
�
�
�
�
�
�
�
� ,� ,� ,� ,� ,� ,� ,� ,� ,� ,� :� :� :� :� :� :� :� :� 0� 0� 0� 0� 0� 0� �� �7� #� #��9�3� 9�$� 9� 9� 9� 9� �S� �D� � � � � ;�c� ;�x� ;�D� ;� ;� ;� ;�8� 8� 8� 6:�)7�0� 0�e�C��M� 2� 0� &�0�<@�0� 0� 0� 0� 0� 0r |