,i!PddlmZmZddlmZmZmZddlZddlm Z ddl m Z m Z m Z ddlmZmZddlmZddlmZmZdd lmZmZmZdd lmZmZdd lmZmZm Z m!Z!dd l"m#Z#m$Z$m%Z%m&Z&m'Z'dd l(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/ddl0m1Z1dZ2defdZ3eGddZ4eGddZ5y)) dataclassfield)ListTypeUnionN)Optional)ConnectionPoolRedis RedisCluster)ExponentialWithJitterBackoff NoBackoff) WeightedList)EventDispatcherEventDispatcherInterface)DEFAULT_GRACE_PERIODCircuitBreakerPBCircuitBreakerAdapter)Database Databases)DEFAULT_FAILOVER_ATTEMPTSDEFAULT_FAILOVER_DELAYFailoverStrategyWeightBasedFailoverStrategy)DEFAULT_FAILURE_RATE_THRESHOLD!DEFAULT_FAILURES_DETECTION_WINDOWDEFAULT_MIN_NUM_FAILURESCommandFailureDetectorFailureDetector)DEFAULT_HEALTH_CHECK_DELAYDEFAULT_HEALTH_CHECK_INTERVALDEFAULT_HEALTH_CHECK_POLICYDEFAULT_HEALTH_CHECK_PROBES HealthCheckHealthCheckPoliciesPingHealthCheck)RetryxreturnctSN)rY/var/www/html/langgraph-service/venv/lib/python3.12/site-packages/redis/multidb/config.pydefault_event_dispatcherr.,s  r,ceZdZUdZdZeed<eeZ eed<dZ e e ed<dZ e eed<dZe eed <eZeed <dZe e ed <d efd Zy)DatabaseConfiga Dataclass representing the configuration for a database connection. This class is used to store configuration settings for a database connection, including client options, connection sourcing details, circuit breaker settings, and cluster-specific properties. It provides a structure for defining these attributes and allows for the creation of customized configurations for various database setups. Attributes: weight (float): Weight of the database to define the active one. client_kwargs (dict): Additional parameters for the database client connection. from_url (Optional[str]): Redis URL way of connecting to the database. from_pool (Optional[ConnectionPool]): A pre-configured connection pool to use. circuit (Optional[CircuitBreaker]): Custom circuit breaker implementation. grace_period (float): Grace period after which we need to check if the circuit could be closed again. health_check_url (Optional[str]): URL for health checks. Cluster FQDN is typically used on public Redis Enterprise endpoints. Methods: default_circuit_breaker: Generates and returns a default CircuitBreaker instance adapted for use. g?weightdefault_factory client_kwargsNfrom_url from_poolcircuit grace_periodhealth_check_urlr(cXtj|j}t|S)N) reset_timeout) pybreakerrr8r)selfcircuit_breakers r-default_circuit_breakerz&DatabaseConfig.default_circuit_breakerRs##22ARARS&77r,)__name__ __module__ __qualname____doc__r1float__annotations__rdictr4r5rstrr6r r7rrr8r9r?r+r,r-r0r00ss0FE5M45"Hhsm"*.Ix'.(,GXn %,.L%.&*hsm*88r,r0ceZdZUdZeeed<eZe e ee fed<e e dddZe ed <d Zeeeed <eZeed <eZeed <eZeed<d Zeeeed<eZeed<eZeed<e Z!eed<e"Z#e$ed<d Z%ee&ed<e'Z(eed<e)Z*eed<e+Z,eed<e-e.Z/e0ed<de1fdZ2deefdZ3deefdZ4de&fdZ5y ) MultiDbConfiga Configuration class for managing multiple database connections in a resilient and fail-safe manner. Attributes: databases_config: A list of database configurations. client_class: The client class used to manage database connections. command_retry: Retry strategy for executing database commands. failure_detectors: Optional list of additional failure detectors for monitoring database failures. min_num_failures: Minimal count of failures required for failover failure_rate_threshold: Percentage of failures required for failover failures_detection_window: Time interval for tracking database failures. health_checks: Optional list of additional health checks performed on databases. health_check_interval: Time interval for executing health checks. health_check_probes: Number of attempts to evaluate the health of a database. health_check_probes_delay: Delay between health check attempts. health_check_policy: Policy for determining database health based on health checks. failover_strategy: Optional strategy for handling database failover scenarios. failover_attempts: Number of retries allowed for failover operations. failover_delay: Delay between failover attempts. auto_fallback_interval: Time interval to trigger automatic fallback. event_dispatcher: Interface for dispatching events related to database operations. Methods: databases: Retrieves a collection of database clients managed by weighted configurations. Initializes database clients based on the provided configuration and removes redundant retry objects for lower-level clients to rely on global retry logic. default_failure_detectors: Returns the default list of failure detectors used to monitor database failures. default_health_checks: Returns the default list of health checks used to monitor database health with specific retry and backoff strategies. default_failover_strategy: Provides the default failover strategy used for handling failover scenarios with defined retry and backoff configurations. databases_config client_class )basecap)backoffretries command_retryNfailure_detectorsmin_num_failuresfailure_rate_thresholdfailures_detection_window health_checkshealth_check_intervalhealth_check_probeshealth_check_probes_delayhealth_check_policyfailover_strategyfailover_attemptsfailover_delayauto_fallback_intervalr2event_dispatcherr(c t}|jD]S}|jjdt dt i|j r2|jj |j fi|j}n}|jrU|jjt dt |jj|j}n|jdi|j}|j|jn |j}|jt|||j|j|jV|S)Nretryr)rRrQ)connection_pool)clientr7r1r9r+)rrJr4updater&r r5rKr6 set_retryr7r?addrr1r9)r= databasesdatabase_configrer7s r-rizMultiDbConfig.databasessY N #44" O  ) ) 0 0%9;?@ ''3**33#,,0?0M0M!**))33!Y[9**44$3$=$=5+**K_-J-JK#**2 779$,,  MM!#*11%4%E%E   && 5" Hr,c\t|j|j|jgS)N)rUrVfailure_detection_window)rrUrVrWr=s r-default_failure_detectorsz'MultiDbConfig.default_failure_detectorss/ "!%!6!6'+'B'B)-)G)G   r,ctgSr*)r%rms r-default_health_checksz#MultiDbConfig.default_health_checkss    r,ctSr*)rrms r-default_failover_strategyz'MultiDbConfig.default_failover_strategys *,,r,)6r@rArBrCrr0rEr rKrrr r&r rSrTrrrrUintrrVrDrrWrXr#r rYr"rZrr[r!r\r$r]rrr^rr_DEFAULT_AUTO_FALLBACK_INTERVALr`rr.rarrrirnrprrr+r,r-rIrIWsG&P>**5:L$uUL012: ,!x_ 56=4c4$BEB'HuH15M8D-.5#@5@::'AuA/J,J48x 0186s62NE2$BEB1602.'9'R 4+@  tK'8 -+;-r,rI)6 dataclassesrrtypingrrrr<typing_extensionsrredisr r r redis.backoffr r redis.data_structurer redis.eventrrredis.multidb.circuitrrrredis.multidb.databaserrredis.multidb.failoverrrrrredis.multidb.failure_detectorrrrrrredis.multidb.healthcheckrr r!r"r#r$r% redis.retryr&rtr.r0rIr+r,r-rs($$&55A-A 7 !$": #8#8 #8L w-w- w-r,