Jy@*> d dl Z d dlmZmZ d dlmZ d dlmZ d dlm Z m
Z
d dl m Z
d dlmZ d dlmZ G d d
e Z G d d e ZdS )
N)insertselect) sessionmaker)LveStatsPlugin)
get_chunks reboot_lock)user)history)serversc ( e Zd Zd Zd Zd ZddZdS )DBSaverc p t j d | _ d| _ d | _ d| _ d| _ d S )Nzplugin.DBSaverr < localhost)logging getLoggerlognowconfigperiod server_idselfs /builddir/build/BUILDROOT/lve-stats-4.3.2-1.el9.x86_64/opt/cloudlinux/venv/lib/python3.11/site-packages/lvestats/plugins/generic/dbsaver.py__init__zDBSaver.__init__ s5 $%566 $ c t | d| j | _ | d| j | _ d S )N
db_timeoutr )intgetr r r r s r
set_configzDBSaver.set_config s< &**\4;??@@ K @@r c > t t }t t g t j | j k }t t }t j t j | j k }t 5 | j
}| } | | }|j
r}| } | C| d |d k r0| | d|d i nG| || j |d n#| || j |d g }
| di D ]\ } } |
i d| dt' t) | j d| j d| j d | j d
| j d t' t) | j d | j d
t' t) | j d| j d| j d| j dt' t) | j d| j d| j! dt' t) | j" d| j# | j$ t' t) | j% | j& | j' | j( | j) t' t) | j* d tW |
D ]?}
| ||
# tX $ r | - |||
w xY w|. n## t^ $ r |0 w xY w |1 n# |1 w xY w d d d d S # 1 swxY w Y d S )N lve_version LVE_VERSION)r r$ lve_usageidcpu cpu_limit cpu_faultmep mep_limitioio_limitmem mem_limit mem_fault mep_faultcreatedr lmemphymemphy memphy_fault)lnprocnproc nproc_faultio_fault
iops_faultliopsiops)2r r
r r wherer __table__updater engineconnectbeginexecute returns_rowsfetchonevaluesr itemsappendr round cpu_usagelcpur* r+ lepio_usager- mem_usagelmemr1 r2 r r4 r5 r6 r7 r8 r9 r: r; r< r= r
OverflowError_try_executing_small_chunkscommit Exceptionrollbackclose)r lve_datasql_insert_querysql_select_serverssql_insert_serverssql_update_serversconntxrowressql_insert_listlve_idvchunks r rD zDBSaver.execute sk !'?? #WI . . 4 4W5F$.5X Y Y #G__ $ . 5 5 7 7 = =g>OSWSa>a b b
]] 9 9 ;&&((D B6
ll#566# t ,,..C }--1HHH LL);)B)BMS[\iSjCk)l)l m m m %74>_ghu_v wwwwLL!3t~[cdq[rLsss"$!) k2!>!>!D!D!F!F IFA#** & !3uQ['9'9#:#: ( (
"15 (
!#eAJ&7&7"8"8 ' "3uQ['9'9#:#: ( ( ( &s5??';'; ( &qy %c%//&:&:!" +AN#$ '(h%(qw %8%8+,=()
*+,%&W$'af
$6$61 : ( 8 8 E %5u====( 88?OQVWWW
s9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 sU 6.P%I:N2 M76N27#NN21O+2 OO+P+PPPP
c t || D ]F} | || # t $ r | j d| Y dS w xY wdS )a Try executing smaller chunks so that it will be feasible to log the one causing error.
In case when data contains value which is bigger than column type can handle, OveflowError is raised.
:param sqlalchemy.engine.Connection conn: db engine
:param sqlalchemy.sql.expression.Insert query: query to execute
:param typing.List[typing.Dict] insert_params: list of params to insert
:param int chunk_length: length of small chunks
zOverflow detected in %sN)r rD rQ r error)r r\ query
insert_paramssmall_chunk_size small_chunks r rR z#DBSaver._try_executing_small_chunksb s &m5EFF K
UK0000
8+FFF
s +%AAN)rd )__name__
__module__ __qualname__r r" rD rR r r r
r
s\ % % %A A A@ @ @ D r r
c & e Zd Zd Zd Zd Zd ZdS )DbUsernamesSaverc 0 d| _ d | _ d| _ d S )Ni T)r rA enabledr s r r zDbUsernamesSaver.__init__t s r c h | dd dk | _ d S )Ncollect_usernamesfalsetrue)r lowerrr r! s r r" zDbUsernamesSaver.set_configy s- zz"5w??EEGG6Q r c || _ d S )N)rA )r rA s r
set_db_enginez DbUsernamesSaver.set_db_engine| s
r c | j r t | j }|d D ]s}t }|\ |_ |_ |_ | | | P# t $ r |
Y pw xY w| | d S d S )N)bindusers)
rr r rA
user_classuid user_namer mergeflushrT rU rS rV )r rW session user_infor s r rD zDbUsernamesSaver.execute s <
4l 44466G %g .
'
' !||;D8$.$.'MM$'''MMOOOO ' ' '$$&&&&&'
NN MMOOOOO
s )A== B B N)rk rl rm r r" ry rD rn r r rp rp s sS
R R R r rp )r
sqlalchemyr r sqlalchemy.ormr lvestats.core.pluginr lvestats.lib.commons.funcr r lvestats.ormr r} lvestats.orm.historyr
lvestats.orm.serversr r
rp rn r r r s % % % % % % % % ' ' ' ' ' ' / / / / / / = = = = = = = = + + + + + + ( ( ( ( ( ( ( ( ( ( ( (\ \ \ \ \n \ \ \~ ~ r