o iP(@sNUddlZddlZddlmZmZddlmZddlmZm Z m Z m Z ddl m Z ddlmZmZddlmZddlmZdd lmZdd lmZd Zd Zd ZdZeeZGdddeZ GdddeZ!Gddde!Z"Gddde"Z#Gddde"Z$Gddde"Z%GdddeZ&e&j'Z(e&e)d<Gddde Z*Gd d!d!e Z+dS)"N)ABCabstractmethod)Enum)ListOptionalTupleUnion)Redis)DEFAULT_TIMEOUTAsyncHTTPClientWrapper) NoBackoff) HttpClient)UnhealthyDatabaseException)Retryg?ic@seZdZedefddZdS) HealthCheckreturncdS)z(Function to determine the health status.N)selfdatabaserrU/home/ubuntu/.local/lib/python3.10/site-packages/redis/asyncio/multidb/healthcheck.py check_healthzHealthCheck.check_healthN)__name__ __module__ __qualname__rboolrrrrrrsrc@sVeZdZdZeedefddZeedefddZ ede e de fdd Z d S) HealthCheckPolicyz) Health checks execution policy. rcCdS)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_checkscr)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__5s z"AbstractHealthCheckPolicy.__init__rcC|jSN)r/r!rrrr";z-AbstractHealthCheckPolicy.health_check_probescCr3r4)r0r!rrrr$?r5z,AbstractHealthCheckPolicy.health_check_delayr%csdSr4rr&rrrr'Csz!AbstractHealthCheckPolicy.executeN)rrrr*r+r2r)r"r$rrrrr'rrrrr,4sr,c@eZdZdZdedeffdd Zdeede fdd Z Z S) HealthyAllPolicyzM Policy that returns True if all health check probes are successful. r"r$ct||dSr4superr2r1 __class__rrr2MzHealthyAllPolicy.__init__r%rc s|D]<}t|jD]4}z||IdHsWdSWnty-}ztd||d}~ww||jdkr>t|jIdHq qdS)NFUnhealthy databaser-Tranger"r Exceptionrasynciosleepr0)rr%r health_checkattempterrrr'Ps    zHealthyAllPolicy.execute rrrr(r*r+r2rrrr' __classcell__rrr;rr7Hr7cr6) HealthyMajorityPolicyzW Policy that returns True if a majority of health check probes are successful. r"r$cr8r4r9r1r;rrr2cr=zHealthyMajorityPolicy.__init__r%rc s|D]f}|jddkr|jd}n|jdd}t|jD]J}z||IdHs6|d8}|dkr6WdSWn tyW}z|d8}|dkrMtd||WYd}~nd}~ww||jdkrht|jIdHqqdS)Nrr-Fr>T)r"r@rrArrBrCr0)rr%rrDallowed_unsuccessful_probesrErFrrrr'fs4   zHealthyMajorityPolicy.executerGrrr;rrJ^rIrJcr6) HealthyAnyPolicyzT Policy that returns True if at least one health check probe is successful. r"r$cr8r4r9r1r;rrr2r=zHealthyAnyPolicy.__init__r%rc sd}|D]T}d}t|jD]<}z||IdHrd}Wn-d}Wnty9}z td||}WYd}~nd}~ww||jdkrJt|jIdHq|sS|sS|S|sY|rY|q|S)NFTr>r-r?)rr%r is_healthyrD exceptionrErFrrrr's2zHealthyAnyPolicy.executerGrrr;rrMrIrMc@seZdZeZeZeZdS)HealthCheckPoliciesN) rrrr7 HEALTHY_ALLrJHEALTHY_MAJORITYrM HEALTHY_ANYrrrrrPsrPDEFAULT_HEALTH_CHECK_POLICYc@seZdZdZdefddZdS)PingHealthCheckz- Health check based on PING command. rcsPt|jtr|jdIdHS|j}|D]}|jdIdHs%dSqdS)NPINGFT) isinstanceclientr execute_command get_nodesredis_connection)rr all_nodesnoderrrrs  zPingHealthCheck.check_healthN)rrrr(rrrrrrrUsrUc@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 Cs<tt||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) rarbretryrcrdrerfrgrhriN)r r rr _http_client_rest_api_port_lag_aware_tolerance) rr_r`rarbrcrdrerfrgrhrirrrr2s    zLagAwareHealthCheck.__init__rc s|jdur tdt|jtr|jd}n|jjdj}|jd|j}||j j_ d}|j dIdHD]"}|dD]}|d|krJ|}n|dD] }||krX|}nqNq>q8|durht d td d |d d |j}|j j |ddIdHdS)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.rWrXr get_connection_kwargs startup_nodesrormrlbase_urlgetloggerwarningrn) rrdb_hostry matching_bdbbdbendpointrsurlrrrrs>        z LagAwareHealthCheck.check_health)rrrr(DEFAULT_LAG_AWARE_TOLERANCEr r*r+rrstrrrbytesr2rrrrrr^sJ     1r^),rBloggingabcrrenumrtypingrrrr redis.asyncior redis.asyncio.http.http_clientr r redis.backoffr redis.http.http_clientr redis.multidb.exceptionr redis.retryrDEFAULT_HEALTH_CHECK_PROBESDEFAULT_HEALTH_CHECK_INTERVALDEFAULT_HEALTH_CHECK_DELAYr getLoggerrr{rrr,r7rJrMrPrQrT__annotations__rUr^rrrrs4        !%