o sz#_get_init_params..)r signature__init__ frozenset parametersitems)rsigrrr_get_init_paramss r&kwargscst|fdd|DS)zJFilter kwargs to only include parameters accepted by the class's __init__.csi|] \}}|vr||qSrr)rkvallowedrr (sz"_filter_kwargs..)r&r$)r'rrr*r_filter_kwargs%sr-g?ic@sheZdZdZeedefddZeedefddZ eedefddZ ed e de fd d Z d S) HealthCheckz! Health check interface. rcCdS)z*Number of probes to execute health checks.Nrrrrrhealth_check_probes9zHealthCheck.health_check_probescCr1)z"Delay between health check probes.Nrr2rrrhealth_check_delay?r4zHealthCheck.health_check_delaycCr1)zCTimeout for the full health check operation (including all probes).Nrr2rrrhealth_check_timeoutEr4z HealthCheck.health_check_timeout hc_clientcdS)ap Function to determine the health status. Args: database: The database being checked hc_client: A Redis client (AsyncRedis or AsyncRedisCluster) to use for health checks. This client follows topology changes automatically. Returns: True if the database is healthy, False otherwise. Nrrdatabaser7rrr check_healthKs zHealthCheck.check_healthN)__name__ __module__ __qualname____doc__propertyrintr3floatr5r6AsyncRedisClientTboolr;rrrrr04sr0c@s`eZdZdZedeedefddZededefddZ ede fd d Z edd d Z d S)HealthCheckPolicyz) Health checks execution policy. health_checksrcr8)z8Execute health checks and return database health status.Nr)rrFr:rrrexecute`zHealthCheckPolicy.execute health_checkcr8z? Executes health check against given database. NrrrIr:rrr_executeezHealthCheckPolicy._executecr8)z= Get a health check client for the database. Nr)rr:rrr get_clientlrMzHealthCheckPolicy.get_clientNcr8)Close all health check clients.Nrr2rrrclosesrHzHealthCheckPolicy.closerN) r<r=r>r?rrr0rDrGrLrCrNrPrrrrrE[srEc@s\eZdZdZddZdeedefddZde fdd Z dd d Z e d edefddZ d S)AbstractHealthCheckPolicyz' Abstract health check policy. cCs i|_dSN)_clientsr2rrrr!~s z"AbstractHealthCheckPolicy.__init__rFrcshdtffdd tjfdd|DddiIdH}|D]}t|tr,td ||s1d SqdS) a Execute all health checks concurrently with individual timeouts. Each health check runs with its own timeout, and all run in parallel. All exception handling is centralized here - _execute() methods just propagate exceptions naturally. rIcs tj||jdIdHS)N)timeout)asynciowait_forrLr6)rI)r:rrrexecute_with_timeouts   z?AbstractHealthCheckPolicy.execute..execute_with_timeoutcsg|]}|qSrr)rhc)rXrr sz5AbstractHealthCheckPolicy.execute..return_exceptionsTNzUnhealthy databaseF)r0rVgather isinstance Exceptionr)rrFr:resultsresultr)r:rXrrrGs    z!AbstractHealthCheckPolicy.executec st|}|j|}|durvt|jttfr)|j}t|t}tdi|}nHt|jt t frg|j }t|t }|jj }|rc|d}t d|j |j|jjj|jjj|jjj|jjd|}ntdtdt|j||j|<|S)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. Nr)hostportdynamic_startup_nodes address_remaprequire_full_coverageretryz?Cluster client has no nodes - cannot create health check clientzUnsupported client type: r)idrTgetr]client AsyncRedis SyncRedisget_connection_kwargsr-AsyncRedisClusterSyncRedisClustercopy startup_nodesrarb nodes_manager_dynamic_startup_nodesrd_require_full_coveragerf ValueError TypeErrortype)rr:db_idri conn_kwargsfiltered_kwargsrp first_noderrrrNs<      z$AbstractHealthCheckPolicy.get_clientNcs>dd|jD}|rtj|ddiIdH|jdS)rOcSsg|] }t|qSr)rV create_taskaclose)rrirrrrZsz3AbstractHealthCheckPolicy.close..r[TN)rTvaluesrVr\clear)r close_tasksrrrrPszAbstractHealthCheckPolicy.closerIcr8rJrrKrrrrLrMz"AbstractHealthCheckPolicy._executerQ)r<r=r>r?r!rr0rDrGrCrNrPrrLrrrrrRys! - rRc@"eZdZdZdedefddZdS)HealthyAllPolicyzM Policy that returns True if all health check probes are successful. rIrcsd||IdH}|j}t|D]}|||IdH}|s dS||dkr/t|jIdHqdS) Executes health check against given database. Uses a single client that handles topology changes automatically. NFT)rNr3ranger;rVsleepr5)rrIr:riprobesattemptr`rrrrLs  zHealthyAllPolicy._executeNr<r=r>r?r0rDrLrrrrrrc@r)HealthyMajorityPolicya3 Policy that returns True if a majority of health check probes are successful. Majority means more than half must pass: - 3 probes: need 2+ to pass (1 failure allowed) - 4 probes: need 3+ to pass (1 failure allowed, tie = unhealthy) - 5 probes: need 3+ to pass (2 failures allowed) rIrc s|j}|dd}||IdH}d}t|D]I}z|||IdH}|s2|d8}|dkr2WdSWntyQ} z| }|d8}|dkrG|WYd} ~ nd} ~ ww||dkrat|jIdHqdS)rrNrFTr3rNrr;r^rVrr5) rrIr:rallowed_unsuccessful_probesrilast_exceptionrr`errrrLs2    zHealthyMajorityPolicy._executeNrrrrrrs rc@r)HealthyAnyPolicyzT Policy that returns True if at least one health check probe is successful. rIrc s|j}d}||IdH}t|D]7}z|||IdH}|r$WdSWnty9}z|}WYd}~nd}~ww||dkrIt|jIdHq|rN|dS)rNTrFr) rrIr:rrrirr`rrrrrL-s(  zHealthyAnyPolicy._executeNrrrrrr(rrc@seZdZeZeZeZdS)HealthCheckPoliciesN) r<r=r>r HEALTHY_ALLrHEALTHY_MAJORITYr HEALTHY_ANYrrrrrJsrDEFAULT_HEALTH_CHECK_POLICYc@sveZdZeeefdededefddZe defddZ e defd d Z e defd d Z e d edefddZdS)AbstractHealthCheckr3r5r6cCs&|dkrtd||_||_||_dS)Nrz*health_check_probes must be greater than 0)rt_health_check_probes_health_check_delay_health_check_timeout)rr3r5r6rrrr!Ts  zAbstractHealthCheck.__init__rcC|jSrS)rr2rrrr3`z'AbstractHealthCheck.health_check_probescCrrS)rr2rrrr5drz&AbstractHealthCheck.health_check_delaycCrrS)rr2rrrr6hrz(AbstractHealthCheck.health_check_timeoutr7csdSrSrr9rrrr;lsz AbstractHealthCheck.check_healthN)r<r=r>DEFAULT_HEALTH_CHECK_PROBESDEFAULT_HEALTH_CHECK_DELAYDEFAULT_HEALTH_CHECK_TIMEOUTrArBr!r@r3r5r6rrCrDr;rrrrrSs&  rc@r)PingHealthCheckz- Health check based on PING command. r7rcsJt|tr|dIdHS|}|D]}|jdIdHs"dSqdS)NPINGFT)r]rjexecute_command get_nodesredis_connection)rr:r7 all_nodesnoderrrr;vs zPingHealthCheck.check_healthN)r<r=r>r?rCrDr;rrrrrqrrcseZdZdZdeeddddddddeeefde 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 ffdd ZdedefddZZS)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_passwordr3r5r6c sNtt||ttdd||||| | | d |_||_||_tj| | |ddS)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) rUrrfrrrrrrr)r3r5r6N) r rrr _http_client_rest_api_port_lag_aware_tolerancesuperr!)rrrrrrrrrrrrr3r5r6 __class__rrr!s*$  zLagAwareHealthCheck.__init__r7rc s|jdur tdt|jttfr|jd}n|jdj}|jd|j }||j j_ d}|j dIdHD]"}|dD]}|d|krM|}n|d D] }||kr[|}nqQqAq;|durkt d td d |d d|j} |j j | ddIdHdS)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.rar: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_urlrtr]rirjrkrlrrarrbase_urlrhloggerwarningr) rr:r7db_hostr matching_bdbbdbendpointrurlrrrr;s>       z LagAwareHealthCheck.check_health)r<r=r>r?DEFAULT_LAG_AWARE_TOLERANCEr rrrrArBrrstrrDr bytesr!rCr; __classcell__rrrrrs\    :r)=rVrloggingabcrrenumrtypingrrrrr redis.asyncior rjredis.asyncio.clusterr rmredis.asyncio.http.http_clientr r redis.backoffr redis.clientrk redis.clusterrnredis.http.http_clientrredis.multidb.exceptionr redis.retryrrCr"r&dictr-rDEFAULT_HEALTH_CHECK_INTERVALrrr getLoggerr<rr0rErRrrrrrr__annotations__rrrrrrrsF            'j,"