j) 0 d Z dZddlZddlZddlZddlmZ ddlZddl m
Z
ddl m Z ddl
mZmZmZ d d
lmZ d d lmZ d d lmZ d d
lmZmZ G d de Zedk r7 e ej dej e Z e dS dS )z*
This module contains DecisionMaker class
z ssa.modules N)isfile)setup_database )Common)iter_domains_dataiter_urls_dataget_url_durations )
load_tunables)ssa_tunables_schema) report_path)previous_day_date sentry_initc \ e Zd ZdZd fd Zd Zedefd Ze defd Z
e de fd Z e de
fd Zed
Zde fd Zd edefd
ZdedefdZdedefdZededededefd Zd Zde fdZde fdZdedefd Zd Z d ededefd Z xZ S )!
DecisionMakerz,
SSA Decision maker implementation.
Nc t t j d | _ | j dt |r|n
t | _ d S )Ndecision_makerzDecisionMaker enabled: %s) super__init__logging getLoggerloggerinfo __package__r engine)selfr __class__s Q/opt/cloudlinux/venv/lib64/python3.11/site-packages/ssa/modules/decision_maker.pyr zDecisionMaker.__init__& s_
'(899 4kBBB & ? A A !
!(,(<(<)*:;[=Z)\ )\% 4 8 8 I II ( ))*JCP_L`L`aaa 44 -o >@ @
L 7 78I J J
L 8:wy7I7I4,/ -o >-@ -@)(,==(!((!0;L-01M-N-N(-1B.H.H(I(I*K *K *K L L L
V " 1 1!*d6F$. 2 0 0 y!(($0C z@DecisionMaker.list_handling_considering_time.. s: 4 4 41M!D1H*-.. 4 4 4r r )rr rangera )r rA rr s `@r list_handling_considering_timez,DecisionMaker.list_handling_considering_time sU yB4 4 4 4 4a]++T224 4 4 4r url_slow_reqs_by_timec , |D ]}|| j k r dS dS )z_
This functions will check if any of elements is greater than "request_number"
TF)request_number)r rv rq s r $compare_elements_with_request_numberz2DecisionMaker.compare_elements_with_request_number s2
' AD'''tt (ur r@ rT c b | j sdS t j t j || S )z}
Calculates the correlation coefficient using the "url_total_reqs" and
the "domain_total_reqs" lists
r )r9 rY amincorrcoef)r r@ rT s r rS z DecisionMaker.get_correlation s1
1wr{>3DEEFFFr rE rF rG c T | fdd |r
| d| S | S )a
Will sort the domain list by "slow_reqs", the goal is to leave only
"domains_number" of uppers, also per each domain will sort urls by
"reqs_num", the goal is to leave only "urls_number" of uppers.
leave_top == 0 allows to keep the full list
c | S Nr: )dict_rG s r z.DecisionMaker.report_sorting.. s E/,B r T)keyreverseN)sortrD s `r r_ z DecisionMaker.report_sorting sN B B B B"& ( ( (
+ + r c | j }t | rt | 5 } t j | }| dd dd }n# t j $ r d}Y nw xY wddd n # 1 swxY w Y d| d}| | }t j
|| dS dS ) z#
Rename old report
r4 z
dd.mm.yyyy._unknownNreport__z.json) r2 r openjsonloadrR replaceJSONDecodeErrorr/ r+ rename)r
old_report json_dataddate_from_reportnew_report_name
new_reports r rename_old_reportz DecisionMaker.rename_old_report s@ -
*
.j!!
UYU ),,A ()uuV\'B'B'J'J3PS'T'T$$ + 1 1 1'0$$$1
U
U
U
U
U
U
U
U
U
U
U
U
U
U
U A)9 @ @ @O **?;;J Ij* - - - - -
.
.s3 B
A'+B
'A;8B
:A;;B
BBr' c | t | j dd 5 }t j ||dd ddd dS # 1 swxY w Y dS )z#
Makes json report
wzutf-8)encodingF ) ensure_asciiindentN)r r r2 r dump)r r' fs r r& z DecisionMaker.add_json_report s
$*C'
B
B
B ?a IfaeA > > > > ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?s AAAc d} t | j 5 }t j || }ddd n # 1 swxY w Y n## t t j f$ r
| j }Y nw xY w|S )zT
Return contents of current report or empty report in case of error
N) object_hook)r r2 r r OSErrorr r7 )r _filtering_hookr' report_dicts r get_json_reportz DecisionMaker.get_json_report s -d.//
M6 "iO L L L
M
M
M
M
M
M
M
M
M
M
M
M
M
M
M-. - - - ,KKK -s. A ; A ?A ?A A'&A'ri c * | j sdS || j k S )a
If correlation flag is enabled - we'll compare correlation_coefficient
from configuration with calculated correlation coefficient.
If the calculated value exceeds the configuration value - we return
True otherwise False. At the same time if correlation flag is disabled -
we'll also return "True" since in this case the correlation coefficient
is not checked and its value is specified as zero in final report.
T)r9 correlation_coefficient)r ri s r rX z$DecisionMaker.correlation_conditions s" 4 4#???r c V | | }| | S )z
At least one element from the received list (url_slow_reqs_by_time)
must be greater than request_number
)ru ry )r rA rv s r rW z%DecisionMaker.request_number_exceeded s2
!% C C ! ! 889NOOOr r> c t | t | z dz }| j d| || j dd k S )zY
Check that percent of throttled requests per URL passes given threshold
d z Calculated throttled percent %s allowed_throttling_percentager )r[ r r! r$ rR )r r> r@ throttled_percents r rQ z$DecisionMaker.is_throttling_suitable sp
!!3 4 4s 8 8 " # ;=NOOO D$:$>$> +Q%0 %0 0 0r r ) __name__
__module__ __qualname____doc__r r( staticmethodrV r/ propertyr2 r6 r7 setr; r# r% rL ru boolry rS r] r_ r r& r rX rW rQ
__classcell__)r s @r r r ! s = = = = = = /c / / / \/ 0S 0 0 0 X0 :t : : : X: s X > > \>: : : : :x4D 4T 4 4 4 4DHMQ Gd Gt G G G G
T
c
(+
04
\
. . ."?d ? ? ? ? @ @ @ @ @ @P P P 0 0/3 08< 0 0 0 0 0 0 0 0r r __main__z decision_maker_standalone.log)filenamelevel) r r r r r+ os.pathr numpyrY ssa.dbr commonr storager r r
configurationr configuration.schemesr internal.constantsr
internal.utilsr r r r basicConfigINFOdmr: r r r s ! ! ! ! ! !
* ) ) ) ) ) 7 7 7 7 7 7 , , , , , , ; ; ; ; ; ; ; ;c0 c0 c0 c0 c0F c0 c0 c0L zKMMMG!@ %l, , , , BBDDDDD r