%IIr d dl mZ d dlZd dlZd dlZd dlmZ d dlmZm Z m
Z
d dl m Z m Z d dl
mZ d dlmZmZmZmZmZmZmZmZ d dlmZ d d lmZ d d
l m Z d d l m Z m Z m!Z! d
Z" dZ# dZ$dZ%dZ& ej' d Z( ej' d Z) ej' d Z* G d de+e Z,e G d d Z-e G d d Z.e G d d Z/ G d d Z0e G d d Z1 G d! d"e Z2 G d# d$e Z3 G d% d&e3e2 Z4ee.gef Z5 e G d' d( Z6 G d) d*e Z7e G d+ d,e7 Z8e G d- d.e7 Z9e G d/ d0 Z: G d1 d2e Z;dde#fd:d9Z]*[>]?$c e Zd ZdZdZdZdS ) ServiceVerbPINGSTATSINFON)__name__
__module__ __qualname__r r r r/builddir/build/BUILD/imunify360-venv-2.6.2/opt/imunify360/venv/lib/python3.11/site-packages/nats/micro/service.pyr r , s D E DDDr% r c ` e Zd ZU ded< ded< dZded< dZded< dZd ed
< dd
ZdS )EndpointConfigstrnamer handlerN
Optional[str] queue_groupsubjectOptional[Dict[str, str]]metadatareturnNonec f | j st d t | j st d | j r.t
| j st d | j r.t
| j st d d S d S )NName cannot be empty.zWInvalid name. Name must contain only alphanumeric characters, underscores, and hyphens.zSInvalid subject. Subject must not contain spaces, and can only have '>' at the end.z9Invalid queue group. Queue group must not contain spaces.)r*
ValueError
NAME_REGEXmatchr.
SUBJECT_REGEXr- selfs r&
__post_init__z EndpointConfig.__post_init__F s y 6 455 5 ** x vww w < x &&t|44
x !vwww ^ &&t'788
^ !\]]] ^ ^
^
^r% r1 r2 )r! r" r# __annotations__r- r. r0 r; r$ r% r& r( r( 2 s
III
&!%K%%%% G !G!!!!h)-H----'
^
^
^
^
^
^r% r( c e Zd ZU dZded< ded< ded< ded< ded< ded < ded
< d Zd ed
< d Zded< edd ZddZ d S )
EndpointStatsz6
Statistics about a specific service endpoint
r) r* r. r- int num_requests
num_errorsprocessing_timeaverage_processing_timeNr,
last_errorz
Optional[Any]dataDict[str, Any]r1 c | |d |d |d |d |d |d |d |d |d
S ) Nr* r. r- rA rB rE rC rD rF r* r. r- rA rB rE rC rD rF r$ clsrF s r& from_dictzEndpointStats.from_dict sb sf O ]+ n-L)L) !2 3$()B$Cf
r% c
x | j | j | j | j | j | j pd| j | j | j d S )N rI rI r9 s r& to_dictzEndpointStats.to_dict sG I| + - / //R # 3'+'CI
r% )rF rG r1 r? r1 rG )
r! r" r# __doc__r= rE rF classmethodrL rO r$ r% r& r? r? V s III LLL OOO ! !%J$$$$ D
[
r% r? c | e Zd ZU dZded< ded< ded< ee Zded< edd Z dd Z
d
S ) EndpointInfoz The information of an endpoint.r) r* r. r- default_factoryrG r0 rF r1 c P | |d |d |d |d S )Nr* r. r- r0 r* r. r- r0 r$ rJ s r& rL zEndpointInfo.from_dict s9 sf O ]+*%
r% c 8 | j | j | j | j dS )NrX rX r9 s r& rO zEndpointInfo.to_dict s& I| +
r% N)rF rG r1 rT rP ) r! r" r# rQ r= r dictr0 rR rL rO r$ r% r& rT rT s ))
III LLL %uT : : :H::::
[
r% rT c : e Zd ZdZddZdd Zdd
Zdd ZddZdS )Endpointz$Endpoint manages a service endpoint.serviceServiceconfigr( r1 r2 c || _ |j | _ |j p|j | _ |j pt | _ |j | _ |j
| _ d| _ d| _
d| _ d| _ d | _ d | _ d S )Nr )_servicer* _namer. _subjectr- DEFAULT_QUEUE_GROUP _queue_groupr+ _handlerr0 _metadata
_num_requests _num_errors_processing_time_average_processing_time _last_error
_subscriptionr: r] r_ s r& __init__zEndpoint.__init__ s{
[
5&+
" . E2E
! ()% 59r% c K | j rJ | j j | j | j | j d {V | _ d S )N)r. queuecb)rm ra _client subscriberc re _handle_requestr9 s r& _startzEndpoint._start sm %%%%#'=#8#B#BM## $C $
$
r% c h K | j sJ | j d {V d | _ d S N)rm unsubscriber9 s r& _stopzEndpoint._stop sJ !!!! ,,......... !r% c V d| _ d| _ d| _ t | _ dS )z Reset the endpoint statistics.r N)rh ri rj r rk r9 s r& _resetzEndpoint._reset s, ! (1 %%%r% msgr c z K t j }| xj dz
c_ t | } | | d{V n# t
$ r}| xj dz
c_ t | | _ t |t r'| |j |j
d{V n)| dt | d{V Y d}~nd}~ww xY wt j }||z
}| xj |z
c_ t | j | j z | _ dS )zHandle an endpoint message.r N500)timeperf_counter_nsrh r rf Exceptionri reprrl
isinstancer
respond_errorcode descriptionrj r@ rk )r: r}
start_timerequesterror current_time elapsed_times r& ru zEndpoint._handle_request sj )++
a #,, @--(( ( ( ( ( ( ( ( ( @ @ @ ! #E{{D % ..
@ ++EJ8IJJJJJJJJJJ ++E4;;????????? @ +-- #j0 - (+D,ADDV,V(W(W%%%s A
C/ B C**C/N)r] r^ r_ r( r1 r2 r< r} r r1 r2 ) r! r" r# rQ ro rv rz r| ru r$ r% r& r\ r\ s ..: : : :
" " " "
4 4 4 4X X X X X Xr% r\ c . e Zd ZU dZded< dZded< dS ) GroupConfigz The configuration of a group.r) r* Nr, r- )r! r" r# rQ r= r- r$ r% r& r r s5 ''
III !%K%%%%//r% r c V e Zd ZdZedd Zedddddd ZdddZdS )EndpointManagerz-
Manages the endpoints of a service.
r_ r( r1 r2 c
K d S rx r$ r: r_ s r& add_endpointz EndpointManager.add_endpoint BE#r% Nr- r. r0 r* r) r+ r r- r, r. r0 r/ c
K d S rx r$ r: r* r+ r- r. r0 s r& r z EndpointManager.add_endpoint#
sr% Optional[EndpointConfig]c
K d S rx r$ r: r_ kwargss r& r z EndpointManager.add_endpoint. s
r% r_ r( r1 r2 r* r) r+ r r- r, r. r, r0 r/ r1 r2 rx r_ r r1 r2 )r! r" r# rQ r r r$ r% r& r r sw EEE XE
&*!%-1 X a``````r% r c R e Zd ZdZedddd
Zedd
ZdddZdS ) GroupManagerz*
Manages the groups of a service.
Nr- r* r) r- r, r1 Groupc d S rx r$ r: r* r- s r& add_groupzGroupManager.add_group6 SVSVr% r_ r c d S rx r$ r s r& r zGroupManager.add_group9 7:sr% Optional[GroupConfig]c d S rx r$ r s r& r zGroupManager.add_group< s r% r* r) r- r, r1 r r_ r r1 r rx r_ r r1 r )r! r" r# rQ r r r$ r% r& r r 1 s` CGVVVVV XV
::: X:UUUUUUUr% r c e Zd Zd dZed d Zed
d
d
d d d Zd d!dZed
dd"d Zed#d Zd d$d Zd
S )%r r] 'Service'r_ r r1 r2 c D || _ |j | _ |j | _ d S rx )ra r* _prefixr- re rn s r& ro zGroup.__init__@ s#
{ " .r% r( c
K d S rx r$ r s r& r zGroup.add_endpointE r r% Nr r* r) r+ r r- r, r. r0 r/ c
K d S rx r$ r s r& r zGroup.add_endpointH r r% r c 6 K |
t di |}n
t |fi |}t || j d d|j p|j d |j p| j }| j | d {V d S )N.)r. r- r$ )
r( r r stripr. r* r- re ra r r s r& r zGroup.add_endpointS s >#--f--FF V..v..F |))#..PP 1N6;PPVVWZ[[ *?d.?
m((00000000000r% r c d S rx r$ r s r& r zGroup.add_groupa r r% c d S rx r$ r s r& r zGroup.add_groupd r r% r c |rt |fi |}n t di |}t || j d|j |j p| j }t
| j | S )Nr )r* r- r$ )r r r r* r- re r ra r s r& r zGroup.add_groupg s + V..v..FF **6**F L006;00 *?d.?
T]F+++r% )r] r r_ r r1 r2 r r rx r r r r )r! r" r# ro r r r r$ r% r& r r ? s / / / /
EEE XE
&*!%-1 X 1 1 1 1 1 CGVVVVV XV
::: X: , , , , , , ,r% r c t e Zd ZU dZded< ded< dZded< dZded < dZded
< dZd ed < ddZ dS )
ServiceConfigz The configuration of a service.r) r* versionNr, r r/ r0 r- zOptional[StatsHandler]
stats_handlerr1 r2 c | j st d t | j st d | j st d t
| j st d | j r.t | j st d d S d S )Nr4 zTInvalid name. It must contain only alphanumeric characters, dashes, and underscores.zVersion cannot be empty.zQInvalid version. It must follow semantic versioning (e.g., 1.0.0, 2.1.3-alpha.1).z[Invalid queue group. It must contain only alphanumeric characters, dashes, and underscores.)r* r5 r6 r7 r SEMVER_REGEXr- r8 r9 s r& r; zServiceConfig.__post_init__ s y 6 455 5 ** u stt t| 9 788 8!!$,// r pqq q &&t'788
q
r% r< )
r! r" r# rQ r= r r0 r- r r; r$ r% r& r r | s ))
III@ LLL6 "&K%%%%))-H----&!%K%%%%1,0M0000 r% r c < e Zd ZU dZded< ded< ded< ded< dS ) ServiceIdentityz@
Defines fields helping to identity a service instance.
r) idr* r Dict[str, str]r0 N)r! r" r# rQ r= r$ r% r& r r sE GGG
IIILLL r% r c e Zd ZU dZded< ded< ded< ee Zded< eZ ded < e
dd
Z ddZ dS ) ServicePingz The response to a ping message.r) r r* r rU r r0 typerF rG r1 c l | |d |d |d | di S )Nr r* r r0 r r* r r0 getrJ s r& rL zServicePing.from_dict sA sDzf OXXj"--
r% c D | j | j | j | j | j dS )Nr r r* r r0 r r9 s r& rO zServicePing.to_dict s* I'I|
r% N)rF rG r1 r rP )
r! r" r# rQ r= r rZ r0 PING_RESPONSE_TYPEr rR rL rO r$ r% r& r r s )) GGG
IIILLL $uT : : :H::::"D""""
[
r% r c e Zd ZU dZded< ded< ded< ded< ee Zd ed
< ee Z d ed < e
Z ded
< e dd Z
ddZdS ) ServiceStatsz The statistics of a service.r) r r* r r startedrU zList[EndpointStats] endpointsr r0 r rF rG r1 c | |d |d |d t j |d d d |d D |d }|S )
Nr r* r r z%Y-%m-%dT%H:%M:%S.%fZc B g | ] }t | S r$ )r? rL .0endpoints r&
z*ServiceStats.from_dict.. s& [[[X}..x88[[[r% r r0 )r r* r r r r0 )r strptime)rK rF results r& rL zServiceStats.from_dict sg Dzf O %d9o7NOO[[kIZ[[[*%
r% c | j | j | j | j | j d d d dz d | j D | j dS )Nz%Y-%m-%dT%H:%M:%S.%fZc 6 g | ]}| S r$ rO r s r& r z(ServiceStats.to_dict.. $ LLL(**,,LLLr% )r r* r r r r r0 )r r* r r r strftimer r0 r9 s r& rO zServiceStats.to_dict s_ II'||,,-CDDSbSICOLLT^LLL
r% N)rF rG r1 r rP )r! r" r# rQ r= r listr rZ r0 STATS_RESPONSE_TYPEr rR rL rO r$ r% r& r r s && GGG III LLL &+U4%@%@%@I@@@@ %uT : : :H::::#D####
[
r% r c e Zd ZU dZded< ded< ded< dZded< ee Zd
ed < ee Z
d ed
< e Z ded< e
dd ZddZdS ) ServiceInfoz The information of a service.r) r* r r Nr, r rU zList[EndpointInfo]r r r0 r rF rG r1 c | |d |d |d | d d |d D |d | dd
S ) zu
Create a `ServiceInfo` from a dictionary.
Notes:
- Unknown fields are ignored.
r r* r r c B g | ] }t | S r$ )rT rL r s r& r z)ServiceInfo.from_dict..E s& ZZZH|--h77ZZZr% r r0 r r )r r* r r r r0 r r rJ s r& rL zServiceInfo.from_dict7 sq sDzf O //ZZ[HYZZZ*%&"BCC
r% c p | j | j | j | j | j d | j D | j dS )z<
Convert the `ServiceInfo` to a dictionary.
c 6 g | ]}| S r$ r r s r& r z'ServiceInfo.to_dict..U r r% r r r* r r r r0 r r9 s r& rO zServiceInfo.to_dictJ sE I'I| +LLT^LLL
r% )rF rG r1 r rP )r! r" r# rQ r= r r r r rZ r0 INFO_RESPONSE_TYPEr rR rL rO r$ r% r& r r s ''
III GGG LLL "&K%%%% %*E$$?$?$?I???? %uT : : :H:::: #D""""
[
$
r% r c e Zd Zd1dZed2d Zd3d
Zed4d Zed
d
d
dd5d Zd6d7dZed
dd8d Z ed9d Z d6d:d Z d;d"Z
dd*Zd3d+Zd?d.Zd?d/Zd?d0Zd
S )@r^ clientr r_ r r1 r2 c |j | _ |j | _ |j | _ |j | _ |j
pi | _ |j | _
|j | _ || _ i | _ g | _ t' j | _ t- | _ t0 | _ d S rx )_nuidnextdecode_idr* rb r _versionr _descriptionr0 rg r- re r _stats_handlerrs _subscriptions
_endpointsr utcnow_startedr _stoppedDEFAULT_PREFIXr )r: r r_ s r& ro zService.__init__[ s <$$&&--//[
" . .B " . $ 2 ))
% r% r) c | j S )z?
Returns the unique identifier of the service.
)r r9 s r& r z
Service.idk s
xr% c
P K | j rd S t j | j t j | j t j | j i}| D ]\ }}| dt |d d | j
f| dt || j d | j
f|t || j | j | j
fg}|D ]/\ }}| j
|| d {V | j |< 0t j | _ | j
d {V d S )Nz-all)r* r prefixz-kind)rr )r r r _handle_ping_requestr _handle_info_requestr _handle_stats_requestitemscontrol_subjectr rb r rs rt r r r flush)r: verb_request_handlersverb verb_handler
verb_subjectskeyr. s r& startz
Service.startr sp F
d 7 d 7 t 9!
#8"="="?"? b b D, MMM#DtT\RRR
NNN#Dtzd4<XXX
#Dtzdht|\\\
M !.
b
b W15 1G1GT`1G1a1a+a+a+a+a+a+a#C((
b !))
l """""""""""r% r( c
K d S rx r$ r s r& r zService.add_endpoint r r% Nr r* r+ r r- r, r. r0 r/ c
K d S rx r$ r s r& r zService.add_endpoint r r% r c K |
t di |}n
t |fi |}t ||j p| j }t | | }| d {V | j | d S Nr r$ )r( r r- re r\ rv r append)r: r_ r r s r& r zService.add_endpoint s >#--f--FF V..v..FV-?-T4CTUUUD&))oo x(((((r% r r c d S rx r$ r s r& r zService.add_group r r% r c d S rx r$ r s r& r zService.add_group r r% r c |rt |fi |}n t di |}t ||j p| j }t | | S r )r r r- re r r s r& r zService.add_group sa + V..v..FF **6**FV-?-T4CTUUUT6"""r% r c t | j | j | j | j d | j pg D | j }| j r$|j D ] }| | |_ |S )z9
Returns the statistics for the service.
c g | ]A}t |j |j |j |j |j |j |j |j BS ))r* r. r- rA rB rE rC rD ) r? rb rc re rh ri rl rj rk r s r& r z!Service.stats.. sh ! $ - ( 5!)!7 ' 3 ' 3$,$=,4,M r% )r r* r r0 r r )
r r rb r rg r r r r rF )r: statsendpoint_statss r& r z
Service.stats s xM^ "& !6B M%
* J"'/
J
J&*&9&9.&I&I## r% r c | t | j | j | j | j | j d | j D S )Nc \ g | ])}t |j |j |j |j *S )rX )rT rb rc re rg r s r& r z Service.info.. sP
! $ - ( 5 % / r% )r r* r r0 r r )r r rb r rg r r r9 s r& infoz Service.info sS xM^ ) !%
r% c : K t j | _ dS )zP
Resets all statistics for all endpoints on a service instance.
N)r r r r9 s r& resetz
Service.reset s !))
r% c B K | j rd S | j D ] }| d {V | j D ] }| d {V g | _ i | _ | j d S rx )r is_setr rz r valuesdrainset)r: r subscriptions r& stopz Service.stop s = ! ! F # #H.."" " " " " " " " " / 6 6 8 8 ' 'L $$&& & & & & & & & &
r% r c | j S )zG
Indicates whether `stop` was executed on the service.
)r r9 s r& stoppedzService.stopped s
} r% r c > K | d {V | S rx )r r9 s r&
__aenter__zService.__aenter__ s+ jjll r% c > K | d {V d S rx )r# )r: exc_typeexctbs r& __aexit__zService.__aexit__
s, iikkr% r} r c K t | j | j | j | j }| t j | d{V dS )zHandle a ping message.r rF N)
r r rb r rg rO respondjsondumpsencode)r: r} pings r& r z Service._handle_ping_request s xM^
'))
kktz$ / / 6 6 8 8k99999999999r% c K | }| t j | d{V dS )zHandle an info message.r. N)r rO r/ r0 r1 r2 )r: r} r s r& r z Service._handle_info_request sa yy{{""$$kktz$ / / 6 6 8 8k99999999999r% c K | }| t j | d{V dS )zHandle a stats message.r. N)r rO r/ r0 r1 r2 )r: r} r s r& r z Service._handle_stats_request! sa
$$&&kktz% 0 0 7 7 9 9k:::::::::::r% )r r r_ r r1 r2 )r1 r) r< r r rx r r r r )r1 r )r1 r )r1 r )r1 r r )r! r" r# ro propertyr r r r r r r r r# r% r' r, r r r r$ r% r& r^ r^ Z s & & & & X # # # #@ EEE XE
&*!%-1 X
)
)
)
)
) CGVVVVV XV
::: X:# # # # # >
$* * * *
X : : : :: : : : ; ; ; ; ; ;r% r^ r r* r, r r1 r) c h || | d| j S || d| j d| S | d| j d| d| S )Nr )value)r r* r r s r& r r ' sl |
''4:''' ..4:.....334:3333r333r% )r r r* r, r r, r1 r) )=
__future__r r0 rer asyncior dataclassesr r r r r enumr
typingr r r
r r r r r nats.aio.clientr nats.aio.msgr nats.aio.subscriptionr r r r r rd r r r r compiler r6 r8 r) r r( r? rT r\ r r r r StatsHandlerr r r r r r^ r r$ r% r& rD s " " " " " " 1 1 1 1 1 1 1 1 1 1 ( ( ( ( ( ( ( ( # " " " " " . . . . . . 3 3 3 3 3 3 3 3 3 3 0 '5 5 7 rz { RZ,
-
-
?++
#t ^ ^ ^ ^ ^ ^ ^ ^F K
K
K
K
K
K
K
K
\ &
&
&
&
&
&
&
&
R>X >X >X >X >X >X >X >XB 0 0 0 0 0 0 0 0a a a a ah a a a, V V V V V8 V V V 4, 4, 4, 4, 4,L/ 4, 4, 4,n ,-
* * * * * * * *Z h
/
8 :
:
:
:
:
? :
:
:
z F
F
F
F
F
F
F
F
RJ; J; J; J; J;! J; J; J;^ 4 4 4 4 4 4 4r%