KinE.UddlZddlZddlZddlmZmZddlmZddlm Z m Z m Z m Z m Z ddlmZddlmZddlmZmZddlmZddlmZddlmZdd lmZdd lm Z dd l!m"Z"e eefZ#d e d e$fdZ%de&d e d e&fdZ'dZ(dZ)dZ*dZ+dZ,ejZe.Z/GddeZ0GddeZ1Gdde1Z2Gdde2Z3Gdde2Z4Gdd e2Z5Gd!d"eZ6e6jnZ8e6e9d#<Gd$d%e0Z:Gd&d'e:Z;Gd(d)e:Zz#_get_init_params..sR  D% 6> JJ    3 3    * *    sAA)r signature__init__ frozenset parametersitems)rsigs r _get_init_paramsr(s=   CLL )C  >>//1   kwargscvt|}|jDcic] \}}||vs ||c}}Scc}}w)zJFilter kwargs to only include parameters accepted by the class's __init__.)r(r&)r*rallowedkvs r _filter_kwargsr/%s4s#G#\\^ .execute_with_timeouts; )) lH5$99 s 5?=?return_exceptionsTNzUnhealthy databaseF)r3rfgather isinstance Exceptionr)rrNr@rhhcresultsresults` ` r rPz!AbstractHealthCheckPolicy.executes [  1> ?2"2& ? "    F&),01ExQWXX  @ s!A3A, A3A1(A3( A3c Kt|}|jj|}|t|jt t fr7|jj}t|t }t di|}n6t|jttfr|jjj}t|t}|jj}|r|d}td|j|j|jjj |jjj"|jjj$|jj&d|}n,t)dt+dt-|j||j|<|Sw)a Get or create a health check client for the database. Creates a single client instance per database that follows topology changes automatically. For cluster databases, the client handles node discovery and slot mapping internally. r)hostportdynamic_startup_nodes address_remaprequire_full_coverageretryz?Cluster client has no nodes - cannot create health check clientzUnsupported client type: r5)idrbgetrkclient AsyncRedis SyncRedisget_connection_kwargsr/AsyncRedisClusterSyncRedisClustercopy startup_nodesrqrr nodes_manager_dynamic_startup_nodesrt_require_full_coveragerv ValueError TypeErrortype)rr@db_idry conn_kwargsfiltered_kwargsr first_nodes r rYz$AbstractHealthCheckPolicy.get_clientst8 ""5) >(//J +BC&ooCCE "0j"I#6o6HOO.?AQ-RS'ooCCEJJL "0>O"P ( = =  !.q!1J.'__'__.6oo.K.K.b.b&.oo&C&C&Q&Q.6oo.K.K.b.b&oo33 *F%Y ";D&B :  :" :$ :)S))(E((*e** 6G D  r)rc eZdZdZdedefdZy)PingHealthCheckz- Health check based on PING command. r=rcKt|tr|jdd{S|j}|D]'}|jjdd{r'yy7A7 w)NPINGFT)rkrzexecute_command get_nodesredis_connection)rr@r= all_nodesnodes r rAzPingHealthCheck.check_healthvsq i ,"226:: :"++-I! !!22BB6JJJ  !; Ks'%A-A)7A-A+ A-%A-+A-N)rCrDrErFrJrKrAr5r)r rrqs 6G D r)rceZdZdZdeeddddddddeeefde de de de e e e fd ed e e d e e d e ee efd e e de e de e de de de ffd ZdedefdZxZS)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_tolerance http_timeout auth_basic verify_tlsca_fileca_pathca_dataclient_cert_fileclient_key_fileclient_key_passwordr7r:r<ctt||ttd||||| | |  |_||_||_t|!| | |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) http_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) rerrvrrrrrrr)r7r:r<N) rrrr _http_client_rest_api_port_lag_aware_tolerancesuperr#)rrrrrrrrrrrrr7r:r< __class__s r r#zLagAwareHealthCheck.__init__soH3 $%IK3%!1 /$7   ,$7!  31!5  r)r=rcK|j tdt|jtt fr|jj d}n'|jjdj}|jd|j}||jj_ d}|jjdd{D]+}|dD]!}|d|k(r|}|d D] }||k(s |}!#-| tjd td d |d d|j} |jj| dd{y77w)a Check database health via Redis Enterprise REST API. Note: The client parameter is not used for this health check as it relies on the REST API instead of Redis protocol. The client is accepted for interface compatibility. Nz[Database health check url is not set. Please check DatabaseConfig for the current database.rqr: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_urlrrkryrzr{r|rrqrrbase_urlrxloggerwarningr) rr@r=db_hostr matching_bdbbdbendpointrurls r rAz LagAwareHealthCheck.check_healths  $ $ ,m  hoo I'> ?oo;;=fEGoo//1!499G//0$2E2E1FG,4  ) **..z:: C , J'72#&L%V,Dw'*      NNU V<= = U+,-??C?X?X>Y [ ##CU#;;;/;( -04!%!%/3*.)--1#>$>&B%8 8 !8  8 U38_- 8  8 #8 #8 %U +,8 #3-8 "#8 &c]8 !!8 ""#8 $$%8 t/6G/D/r)r)=rfrloggingabcrrenumrtypingrrrr r redis.asyncior rzredis.asyncio.clusterr r}redis.asyncio.http.http_clientr r redis.backoffr redis.clientr{ redis.clusterr~redis.http.http_clientrredis.multidb.exceptionr redis.retryrrJr$r(dictr/rDEFAULT_HEALTH_CHECK_INTERVALrrr getLoggerrCrr3rMr_rrrrrr__annotations__rrrr5r)r rsB#55-CR#+:->*&778 $ 9 =4=d=t=   !  "   8 $$ #$ N