hjkhjkjhkhjkgjghjhgjhgjghjhghjghjhgjjnbvnvbnvk
bnmbnmbnhjkhjkhjkhjktyutyutyuytutyutyutyhgjgjghjgjghjguytutyry
/
opt
/
alt
/
python37
/
lib
/
python3.7
/
site-packages
/
xray
/
internal
/
__pycache__
/
Upload FileeE
HOME
B �'Ud� � @ sj d Z ddlZddlZddlmZmZ ddlmZmZ ddlm Z m Z ddl m Z m Z mZ G dd � d �ZdS ) z[ This module contains FaultDetector class, aimed to track throttling for incoming requests � N)�datetime� timedelta)�current_thread�RLock)�Tuple�Any� )� drop_after� check_period�throttling_thresholdc @ s e Zd ZdZdd� Zeeeeef d�dd�Z e ed�dd ��Z e e d�d d ��Z e e d�d d ��Ze ed�dd��Zeee d�dd��Zee d�dd�Zeedd�dd�Zedd�dd �Zeeeeef d �d d �Zdd�d d!�Zdd�d"d#�Zed�d$d%�ZdS )&� FaultDetectorz Fault Detector class c C s& t �d�| _t� | _t� | _d| _d S )N�fault_detectorz$/usr/share/alt-php-xray/flush.latest)�logging� getLogger�logger�dict�throttling_statr �lock�time_marker_file)�self� r �M/opt/alt/python37/lib/python3.7/site-packages/xray/internal/fault_detector.py�__init__ s zFaultDetector.__init__)�t_key� cpu_value�returnc C sd | j �P y| j| \}}W n tk r6 d\}}Y n X | �||�\}}| �|� W dQ R X ||fS )z) Perform fault detection )Fr N)r r �KeyError�_was_throttled_since� _flush_entry)r r r Z saved_cpu�_�hitting_limits�throttled_timer r r �__call__ s zFaultDetector.__call__)r c C s t t�� �S )z( Current Unix timestamp )�int�time)r r r r � timestamp/ s zFaultDetector.timestampc C s | � t�S )z8 Drop after value as datetime.timedelta )�timedelta_in_minutesr )r r r r � _drop_after6 s zFaultDetector._drop_afterc C s | � t�S )z: Check period value as datetime.timedelta )r& r )r r r r � _check_period= s zFaultDetector._check_periodc C sn y6t | j��}t|�� �� �}W dQ R X | �|�| jkS ttfk rh | j � | � � W dQ R X dS X dS )ze If throttling_stat should be flushed right now regarding check period value NF) �openr r# �read�strip� current_deltar( �OSError� ValueErrorr �_update_latest_flush)r � latest_infoZlatest_flush_timer r r � should_flushD s zFaultDetector.should_flush)�minsr c C s t | d�S )zB Get timedelta object for given number of minutes )Zminutes)r )r2 r r r r& S s z"FaultDetector.timedelta_in_minutes)� time_stampr c C s" t td�dd�}|| j�||� S )zQ Calculate timedelta for given time_stamp from current timestamp )�tsr c S s t �| �S )z> Make a datetime object from ts (int) )r Z fromtimestamp)r4 r r r �_cast_ s z*FaultDetector.current_delta.<locals>._cast)r# r r% )r r3 r5 r r r r, Z s zFaultDetector.current_deltaN)�key�cpur c C s\ | j }| j�F | j�dt� j|||� ||f| j|< | j�dt� jt| j�� W dQ R X dS )z5 Save throttling entry for given key z-[%s] Adding new throttling entry %s: (%i, %i)z [%s] Throttling entries total %iN)r% r r �debugr �namer �len)r r6 r7 Zstampr r r �saveg s zFaultDetector.save)r6 r c C sR | j �B | j�dt� j|� | j�|d� | j�dt� jt| j�� W dQ R X dS )z6 Remove throttling entry by given key z [%s] Abandon throttling entry %sr z [%s] Throttling entries total %iN)r r r8 r r9 r �popr: )r r6 r r r r t s zFaultDetector._flush_entry)� initial_value� current_valuer c C s@ || d }|t k}| j�dt� jt ||� ||dkr:|ndfS )z� Check current throttling time of given lve_id vs given initial_value Return: tuple(fact of throttling, throttling time in ms) i@B z'[%s] Throttling checked with %i: %s, %ir )r r r8 r r9 )r r= r> Z spent_timeZthrottling_factr r r r � s z"FaultDetector._was_throttled_sincec C st | j rp| j�^ | j�dt� jt| j�� x| �� D ]}| � |� q2W | � � | j�dt� jt| j�� W dQ R X dS )z, Clear throttling stat dict z$[%s] Flush started, total entries %iz%[%s] Flush finished, total entries %iN) r1 r r r8 r r9 r: r �_expired_entriesr r/ )r r6 r r r �flush� s zFaultDetector.flushc C sR y,t | jd��}|�t| j�� W dQ R X W n tk rL | j�d� Y nX dS )z? Write current timestamp into time marker file �wNz(Failed to save timestamp of latest flush)r) r �write�strr% r- r �error)r r0 r r r r/ � s z"FaultDetector._update_latest_flushc C s^ t � }xR| j�� D ]D\}}|\}}| �|�}|| jkr| j�dt� j||� |� |� qW |S )ze Collect expired throttling stat entries keys regarding the drop after value z*[%s] Expired entry detected %i: %s elapsed) �listr �itemsr, r' r r8 r r9 �append)r Zexpiredr6 �valuer r% Z_deltar r r r? � s z FaultDetector._expired_entries)�__name__� __module__� __qualname__�__doc__r r r# r �boolr" �propertyr% r r' r( r1 � staticmethodr& r, r; r r r@ r/ rE r? r r r r r s* r )rL r r$ r r � threadingr r �typingr r � constantsr r r r r r r r �<module> s