o i)'@sFUddlZddlmZmZddlmZddlmZddlm Z m Z m Z m Z ddl mZddlmZddlmZmZdd lmZdd lmZd Zd Zd ZdZeeZGdddeZGdddeZ Gddde Z!Gddde!Z"Gddde!Z#Gddde!Z$GdddeZ%e%j&Z'e%e(d<GdddeZ)Gd d!d!eZ*dS)"N)ABCabstractmethod)Enum)sleep)ListOptionalTupleUnion)Redis) NoBackoff)DEFAULT_TIMEOUT HttpClient)UnhealthyDatabaseException)Retryg?ic@seZdZedefddZdS) HealthCheckreturncCdS)z(Function to determine the health status.N)selfdatabaserrM/home/ubuntu/.local/lib/python3.10/site-packages/redis/multidb/healthcheck.py check_healthzHealthCheck.check_healthN)__name__ __module__ __qualname__rboolrrrrrrsrc@sVeZdZdZeedefddZeedefddZ ede e de fdd Z d S) HealthCheckPolicyz) Health checks execution policy. rcCr)z*Number of probes to execute health checks.Nrrrrrhealth_check_probes!z%HealthCheckPolicy.health_check_probescCr)z"Delay between health check probes.Nrr rrrhealth_check_delay'r"z$HealthCheckPolicy.health_check_delay health_checkscCr)z8Execute health checks and return database health status.Nrrr$rrrrexecute-rzHealthCheckPolicy.executeN)rrr__doc__propertyrintr!floatr#rrrr&rrrrrsrc@s\eZdZdedefddZedefddZedefdd Ze d e e de fd d Z d S)AbstractHealthCheckPolicyr!r#cCs |dkrtd||_||_dS)Nz*health_check_probes must be greater than 0) ValueError_health_check_probes_health_check_delayrr!r#rrr__init__4s z"AbstractHealthCheckPolicy.__init__rcC|jSN)r.r rrrr!:z-AbstractHealthCheckPolicy.health_check_probescCr2r3)r/r rrrr#>r4z,AbstractHealthCheckPolicy.health_check_delayr$cCsdSr3rr%rrrr&Bsz!AbstractHealthCheckPolicy.executeN)rrrr)r*r1r(r!r#rrrrr&rrrrr+3sr+c@eZdZdZdedeffdd Zdeede fdd Z Z S) HealthyAllPolicyzM Policy that returns True if all health check probes are successful. r!r#ct||dSr3superr1r0 __class__rrr1LzHealthyAllPolicy.__init__r$rc Cst|D]5}t|jD]-}z ||sWdSWnty)}ztd||d}~ww||jdkr6t|jq qdS)NFUnhealthy databaser,Tranger!r Exceptionrrr/)rr$r health_checkattempterrrr&Os     zHealthyAllPolicy.execute rrrr'r)r*r1rrrr& __classcell__rrr:rr6Gr6cr5) HealthyMajorityPolicyzW Policy that returns True if a majority of health check probes are successful. r!r#cr7r3r8r0r:rrr1br<zHealthyMajorityPolicy.__init__r$rc Cs|D]_}|jddkr|jd}n|jdd}t|jD]C}z||s2|d8}|dkr2WdSWn tyS}z|d8}|dkrItd||WYd}~nd}~ww||jdkr`t|jqqdS)Nrr,Fr=T)r!r?rr@rrr/)rr$rrAallowed_unsuccessful_probesrBrCrrrr&es2     zHealthyMajorityPolicy.executerDrrr:rrG]rFrGcr5) HealthyAnyPolicyzT Policy that returns True if at least one health check probe is successful. r!r#cr7r3r8r0r:rrr1r<zHealthyAnyPolicy.__init__r$rc Csd}|D]M}d}t|jD]5}z||rd}Wn)d}Wnty5}z td||}WYd}~nd}~ww||jdkrBt|jq |sK|sK|S|sQ|rQ|q|S)NFTr=r,r>)rr$r is_healthyrA exceptionrBrCrrrr&s0  zHealthyAnyPolicy.executerDrrr:rrJ~rFrJc@seZdZeZeZeZdS)HealthCheckPoliciesN) rrrr6 HEALTHY_ALLrGHEALTHY_MAJORITYrJ HEALTHY_ANYrrrrrMsrMDEFAULT_HEALTH_CHECK_POLICYc@seZdZdZdefddZdS)PingHealthCheckz- Health check based on PING command. rcCsBt|jtr |jdS|j}|D] }|jdsdSqdS)NPINGFT) isinstanceclientr execute_command get_nodesredis_connection)rr all_nodesnoderrrrs    zPingHealthCheck.check_healthN)rrrr'rrrrrrrRsrRc@seZdZdZdeeddddddddf dedededee e e fd e d ee d ee d ee e e fd ee dee dee fddZde fddZdS)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 Cs8t||ttdd||||| | | d |_||_||_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) r^r_retryr`rarbrcrdrerfN)r rr _http_client_rest_api_port_lag_aware_tolerance) rr\r]r^r_r`rarbrcrdrerfrrrr1s   zLagAwareHealthCheck.__init__rc Cs|jdur tdt|jtr|jd}n|jjdj}|jd|j}||j _ d}|j dD]"}|dD]}|d|krE|}n|dD] }||krS|}nqIq9q3|durct d td d |d d |j}|j j |dddS)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_urlr-rTrUr get_connection_kwargs startup_nodesrlrjribase_urlgetloggerwarningrk) rrdb_hostrv matching_bdbbdbendpointrpurlrrrrs<       z LagAwareHealthCheck.check_health)rrrr'DEFAULT_LAG_AWARE_TOLERANCEr r)r*rrstrrr bytesr1rrrrrr[sJ     /r[)+loggingabcrrenumrtimertypingrrrr redisr redis.backoffr redis.http.http_clientr r redis.multidb.exceptionr redis.retryrDEFAULT_HEALTH_CHECK_PROBESDEFAULT_HEALTH_CHECK_INTERVALDEFAULT_HEALTH_CHECK_DELAYr getLoggerrrxrrr+r6rGrJrMrNrQ__annotations__rRr[rrrrs2        !%