1iP(UddlZddlZddlmZmZddlmZddlmZm Z m Z m Z ddl m Z ddlmZmZddlmZddlmZdd lmZdd lmZd Zd Zd ZdZej:eZGddeZ GddeZ!Gdde!Z"Gdde"Z#Gdde"Z$Gdde"Z%GddeZ&e&jNZ(e&e)d<Gdde Z*Gd d!e Z+y)"N)ABCabstractmethod)Enum)ListOptionalTupleUnion)Redis)DEFAULT_TIMEOUTAsyncHTTPClientWrapper) NoBackoff) HttpClient)UnhealthyDatabaseException)Retryg?ic"eZdZedefdZy) HealthCheckreturnc Kyw)z(Function to determine the health status.N)selfdatabases k/mnt/e/genesis-system/.venvs/voice-bridge/lib/python3.12/site-packages/redis/asyncio/multidb/healthcheck.py check_healthzHealthCheck.check_health  N)__name__ __module__ __qualname__rboolrrrrrs d  r"rcpeZdZdZeedefdZeedefdZ ede e de fdZ y)HealthCheckPolicyz) Health checks execution policy. rcy)z*Number of probes to execute health checks.Nrrs rhealth_check_probesz%HealthCheckPolicy.health_check_probes" r"cy)z"Delay between health check probes.Nrr&s rhealth_check_delayz$HealthCheckPolicy.health_check_delay(r(r" health_checksc Kyw)z8Execute health checks and return database health status.Nrrr+rs rexecutezHealthCheckPolicy.execute.rrN)rrr __doc__propertyrintr'floatr*rrr!r.rr"rr$r$st S   E   4 +< 4  r"r$cheZdZdedefdZedefdZedefdZe de e de fdZ y ) AbstractHealthCheckPolicyr'r*c@|dkr td||_||_y)Nz*health_check_probes must be greater than 0) ValueError_health_check_probes_health_check_delay)rr'r*s r__init__z"AbstractHealthCheckPolicy.__init__5s&  "IJ J$7!#5 r"rc|jSN)r8r&s rr'z-AbstractHealthCheckPolicy.health_check_probes;s(((r"c|jSr<)r9r&s rr*z,AbstractHealthCheckPolicy.health_check_delay?s'''r"r+c Kywr<rr-s rr.z!AbstractHealthCheckPolicy.executeCs  rN)rrr r1r2r:r0r'r*rrrr!r.rr"rr4r44sr6C6U6 )S))(E(( 4 +< 4  r"r4cBeZdZdZdedeffd Zdeede fdZ xZ S)HealthyAllPolicyzM Policy that returns True if all health check probes are successful. r'r*c&t|||yr<superr:rr'r* __class__s rr:zHealthyAllPolicy.__init__M ,.@Ar"r+rcBK|D]t}t|jD]Z} |j|d{sy ||jdz ks4t j |jd{\vy7G#t$r}t d||d}~wwxYw7)w)NFUnhealthy databaser6Tranger'r Exceptionrasynciosleepr9)rr+r health_checkattemptes rr.zHealthyAllPolicy.executePs) BL !9!9: BX!-!:!:8!DDD$E T5599!--(@(@AAA B BE X45I8UVWWXBsIBA?A=A?B"B3B4 B=A?? B BBB rrr r/r1r2r:rrr!r. __classcell__rEs@rr@r@Hs6BCBUB 4 +< 4 r"r@cBeZdZdZdedeffd Zdeede fdZ xZ S)HealthyMajorityPolicyzW Policy that returns True if a majority of health check probes are successful. r'r*c&t|||yr<rBrDs rr:zHealthyMajorityPolicy.__init__crFr"r+rcK|D]}|jdzdk(r|jdz }n|jdzdz }t|jD]c} |j|d{s |dz}|dkry||jdz ks=t j |jd{ey7P#t$r!}|dz}|dkr t d||Yd}~fd}~wwxYw78w)Nrr6FrHT)r'rJrrKrrLrMr9)rr+rrNallowed_unsuccessful_probesrOrPs rr.zHealthyMajorityPolicy.executefs) BL''!+q0.2.F.F.J+/3/G/G!/Kq.P+ !9!9: B !-!:!:8!DDD3q836!;#(T5599!--(@(@AAA B  B*E!/14/2a780(A8BsTAC+B<)B:*B<8C+"C+0C)1 C+:B<< C&C!C+!C&&C+rQrSs@rrUrU^s6BCBUB4 +<4r"rUcBeZdZdZdedeffd Zdeede fdZ xZ S)HealthyAnyPolicyzT Policy that returns True if at least one health check probe is successful. r'r*c&t|||yr<rBrDs rr:zHealthyAnyPolicy.__init__rFr"r+rczKd}|D]}d}t|jD]]} |j|d{rd}n?d} ||jdz ks7t j |jd{_|s|s|cS|r|s||S7Z#t$r}t d||}Yd}~ld}~wwxYw7>w)NFTrHr6rI)rr+r is_healthyrN exceptionrOrPs rr.zHealthyAnyPolicy.executes ) LI !9!9: B )66x@@@%) %* T5599!--(@(@AAA Bi!!I+ .%A ! :,h!I Bsh"B;BBBB;BB;"B;:B9;B; B;B;B B6 B1,B;1B66B;rQrSs@rr[r[s6BCBUB4 +<4r"r[ceZdZeZeZeZy)HealthCheckPoliciesN) rrr r@ HEALTHY_ALLrUHEALTHY_MAJORITYr[ HEALTHY_ANYrr"rraras"K,"Kr"raDEFAULT_HEALTH_CHECK_POLICYceZdZdZdefdZy)PingHealthCheckz- Health check based on PING command. rcKt|jtr#|jjdd{S|jj }|D]'}|j jdd{r'yy7K7 w)NPINGFT) isinstanceclientr execute_command get_nodesredis_connection)rr all_nodesnodes rrzPingHealthCheck.check_healths~ hoou -!88@@ @!113I! !!22BB6JJJ  !A Ks(9B BAB =B >B B  B N)rrr r/r!rrr"rrgrgs d r"rgceZdZdZdeeddddddddf dedededee e e fd e d ee d ee d ee e e fd ee dee dee fdZde fdZy)LagAwareHealthCheckz Health check available for Redis Enterprise deployments. Verify via REST API that the database is healthy based on different lags. i$NT rest_api_portlag_aware_tolerancetimeout auth_basic verify_tlsca_fileca_pathca_dataclient_cert_fileclient_key_fileclient_key_passwordc tt||ttd||||| | |  |_||_||_y)a Initialize LagAwareHealthCheck with the specified parameters. Args: rest_api_port: Port number for Redis Enterprise REST API (default: 9443) lag_aware_tolerance: Tolerance in lag between databases in MS (default: 100) timeout: Request timeout in seconds (default: DEFAULT_TIMEOUT) auth_basic: Tuple of (username, password) for basic authentication verify_tls: Whether to verify TLS certificates (default: True) ca_file: Path to CA certificate file for TLS verification ca_path: Path to CA certificates directory for TLS verification ca_data: CA certificate data as string or bytes client_cert_file: Path to client certificate file for mutual TLS client_key_file: Path to client private key file for mutual TLS client_key_password: Password for encrypted client private key r)retries) rurvretryrwrxryrzr{r|r}N)r rrr _http_client_rest_api_port_lag_aware_tolerance) rrsrtrurvrwrxryrzr{r|r}s rr:zLagAwareHealthCheck.__init__sS@3 %IK3%!1 /$7   ,$7!r"rcK|j tdt|jtr|jj d}n#|jj dj}|jd|j}||jj_ d}|jjdd{D]+}|dD]!}|d|k(r|}|dD] }||k(s |}!#-| tjd td d |d d |j}|jj|dd{y77w)Nz[Database health check url is not set. Please check DatabaseConfig for the current database.hostr:z/v1/bdbs endpointsdns_nameaddrz8LagAwareHealthCheck failed: Couldn't find a matching bdbzCould not find a matching bdbz /v1/bdbs/uidz=/availability?extend_check=lag&availability_lag_tolerance_ms=F) expect_jsonT)health_check_urlr7rjrkr get_connection_kwargs startup_nodesrrrbase_urlgetloggerwarningr) rrdb_hostr matching_bdbbdbendpointrurls rrz LagAwareHealthCheck.check_healths  $ $ ,m  hoou -oo;;=fEGoo33A6;;G//0$2E2E1FG,4  ) **..z:: C , J'72#&L%V,Dw'*      NNU V<= = U+,-??C?X?X>Y [ ##CU#;;;/;( (04!%!%/3*.)--1/8/8!/8 /8 U38_- /8  /8#/8#/8%U +,/8#3-/8"#/8&c]/8b'd'r"rr),rLloggingabcrrenumrtypingrrrr redis.asyncior redis.asyncio.http.http_clientr r redis.backoffr redis.http.http_clientrredis.multidb.exceptionr redis.retryrDEFAULT_HEALTH_CHECK_PROBESDEFAULT_HEALTH_CHECK_INTERVALDEFAULT_HEALTH_CHECK_DELAYr getLoggerrrrr$r4r@rUr[rarbre__annotations__rgrrrr"rrs#//R#-> ! "   8 $ #   . 1 (0,5B"0"J#$# 4G3R3R0Rk$^+^r"