keJiP(UddlZddlZddlmZmZddlmZddlmZm Z m Z m Z ddl m Z ddlmZmZddlmZddlmZdd lmZdd lmZd Zd Zd ZdZejeZGddeZ GddeZ!Gdde!Z"Gdde"Z#Gdde"Z$Gdde"Z%GddeZ&e&j'Z(e&e)d<Gdde Z*Gd d!e Z+dS)"N)ABCabstractmethod)Enum)ListOptionalTupleUnion)Redis)DEFAULT_TIMEOUTAsyncHTTPClientWrapper) NoBackoff) HttpClient)UnhealthyDatabaseException)Retryg?ic*eZdZedefdZdS) HealthCheckreturnc KdS)z(Function to determine the health status.N)selfdatabases g/var/www/html/movieo_spanner_bot/venv/lib/python3.11/site-packages/redis/asyncio/multidb/healthcheck.py check_healthzHealthCheck.check_health  N)__name__ __module__ __qualname__rboolrrrrrrs9 d   ^   rrceZdZdZeedefdZeedefdZ ede e de fdZ dS)HealthCheckPolicyz) Health checks execution policy. rcdS)z*Number of probes to execute health checks.Nrrs rhealth_check_probesz%HealthCheckPolicy.health_check_probes"  rcdS)z"Delay between health check probes.Nrr%s rhealth_check_delayz$HealthCheckPolicy.health_check_delay(r'r health_checksc KdS)z8Execute health checks and return database health status.Nrrr*rs rexecutezHealthCheckPolicy.execute.rrN)rrr __doc__propertyrintr&floatr)rrr!r-rrrr#r#s S   ^X  E   ^X  4 +< 4   ^   rr#ceZdZdedefdZedefdZedefdZe de e de fdZ d S) AbstractHealthCheckPolicyr&r)cL|dkrtd||_||_dS)Nz*health_check_probes must be greater than 0) ValueError_health_check_probes_health_check_delay)rr&r)s r__init__z"AbstractHealthCheckPolicy.__init__5s2  " "IJJ J$7!#5   rrc|jSN)r7r%s rr&z-AbstractHealthCheckPolicy.health_check_probes;s ((rc|jSr;)r8r%s rr)z,AbstractHealthCheckPolicy.health_check_delay?s ''rr*c KdSr;rr,s rr-z!AbstractHealthCheckPolicy.executeCs  rN)rrr r0r1r9r/r&r)rrrr!r-rrrr3r34s6C6U6666 )S)))X)(E(((X( 4 +< 4   ^   rr3cHeZdZdZdedeffd Zdeede fdZ xZ S)HealthyAllPolicyzM Policy that returns True if all health check probes are successful. r&r)cLt||dSr;superr9rr&r) __class__s rr9zHealthyAllPolicy.__init__M% ,.@AAAAArr*rc$K|D]}t|jD]s} ||d{VsdSn##t$r}t d||d}~wwxYw||jdz krt j|jd{VtdS)NFUnhealthy databaser5Tranger&r Exceptionrasynciosleepr8)rr*r health_checkattemptes rr-zHealthyAllPolicy.executePs) B BL !9:: B BX!-!:!:8!D!DDDDDDD%$uuu% XXX45I8UVWWWXT5999!-(@AAAAAAAAA Bts> AAA rrr r.r0r1r9rrr!r- __classcell__rDs@rr?r?HsBCBUBBBBBB 4 +< 4        rr?cHeZdZdZdedeffd Zdeede fdZ xZ S)HealthyMajorityPolicyzW Policy that returns True if a majority of health check probes are successful. r&r)cLt||dSr;rArCs rr9zHealthyMajorityPolicy.__init__crErr*rcK|D]}|jdzdkr |jdz }n |jdzdz }t|jD]} ||d{Vs|dz}|dkrdSn3#t$r&}|dz}|dkrt d||Yd}~nd}~wwxYw||jdz krt j|jd{VdS)Nrr5FrGT)r&rIrrJrrKrLr8)rr*rrMallowed_unsuccessful_probesrNrOs rr-zHealthyMajorityPolicy.executefs^) B BL'!+q00.2.F.J++/3/G!/Kq.P+ !9:: B B !-!:!:8!D!DDDDDDD)3q836!;;#(555 /14/2a7780(A87777T5999!-(@AAAAAAAAA Bts&A// B9BBrPrRs@rrTrT^sBCBUBBBBBB4 +<4rrTcHeZdZdZdedeffd Zdeede fdZ xZ S)HealthyAnyPolicyzT Policy that returns True if at least one health check probe is successful. r&r)cLt||dSr;rArCs rr9zHealthyAnyPolicy.__init__rErr*rcVKd}|D]}d}t|jD]z} ||d{Vrd}nYd}n(#t$r}t d||}Yd}~nd}~wwxYw||jdz krt j|jd{V{|s|s|cS|s|r||S)NFTrGr5rH)rr*r is_healthyrM exceptionrNrOs rr-zHealthyAnyPolicy.executes) )  LI !9:: B B )66x@@@@@@@@+%) %*   :,h!!IIIIII T5999!-(@AAAAAAAAA i !!!! I sAA A)A$$A)rPrRs@rrZrZsBCBUBBBBBB4 +<4rrZceZdZeZeZeZdS)HealthCheckPoliciesN) rrr r? HEALTHY_ALLrTHEALTHY_MAJORITYrZ HEALTHY_ANYrrrr`r`s"K,"KKKrr`DEFAULT_HEALTH_CHECK_POLICYceZdZdZdefdZdS)PingHealthCheckz- Health check based on PING command. rcKt|jtr |jdd{VS|j}|D]%}|jdd{VsdS&dS)NPINGFT) isinstanceclientr execute_command get_nodesredis_connection)rr all_nodesnodes rrzPingHealthCheck.check_healths hou - - !88@@@@@@@@ @!1133I! ! !!2BB6JJJJJJJJ! 55!4rN)rrr r.r!rrrrrfrfs9 d      rrfceZdZdZdeeddddddddf dedededee e e fd e d ee d ee d ee e e fd ee dee dee fdZde fdZdS)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||||| | |  |_||_||_dS)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) rtruretryrvrwrxryrzr{r|N)r rrr _http_client_rest_api_port_lag_aware_tolerance) rrrrsrtrurvrwrxryrzr{r|s rr9zLagAwareHealthCheck.__init__sp@3 %IKK333%!1 /$7      ,$7!!!rrcK|jtdt|jtr |jd}n|jjdj}|jd|j}||j j_ d}|j dd{VD]2}|dD]'}|d|kr|}n|dD] }||kr|}n (3|)t d td d |d d |j}|j |dd{VdS)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_urlr6rirjr get_connection_kwargs startup_nodesrrrbase_urlgetloggerwarningr) rrdb_hostr matching_bdbbdbendpointrurls rrz LagAwareHealthCheck.check_healths  $ ,m  hou - - <o;;==fEGGo3A6;G/GG$2EGG,4 ) *..z::::::::  C ,  J'722#&LE%V,Dw'* '   NNU V V V<== = [ U+ [ [?C?X [ [ ##CU#;;;;;;;;;tr)rrr r.DEFAULT_LAG_AWARE_TOLERANCEr r0r1rrstrr!r bytesr9rrrrrqrqs"#>(04!%!%/3*.)--1/8/8/8!/8 /8 U38_- /8  /8#/8#/8%U +,/8#3-/8"#/8&c]/8/8/8/8b'd''''''rrq),rKloggingabcrrenumrtypingrrrr 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#r3r?rTrZr`rard__annotations__rfrqrrrrs########////////////RRRRRRRR######------>>>>>> ! "  8 $ $     #           .     1   (0,5B"""""0"""J#####$### 4G3R0RRRk$^^^^^+^^^^^r