o i#@sZddlmZmZddlmZddlmZmZmZm Z ddl Z ddl m Z m Z mZddlmZmZddlmZmZmZmZddlmZmZdd lmZmZmZmZmZm Z m!Z!dd l"m#Z#dd l$m%Z%m&Z&dd l'm(Z(dd l)m*Z*m+Z+ddl,m-Z-m.Z.m/Z/ddl0m1Z1m2Z2m3Z3m4Z4dZ5GdddeZ6de+fddZ7eGdddZ8eGdddZ9dS)) dataclassfield)Enum)ListOptionalTypeUnionN)ConnectionPoolRedis RedisCluster)Database Databases)DEFAULT_FAILOVER_ATTEMPTSDEFAULT_FAILOVER_DELAYAsyncFailoverStrategyWeightBasedFailoverStrategy)AsyncFailureDetectorFailureDetectorAsyncWrapper)DEFAULT_HEALTH_CHECK_DELAYDEFAULT_HEALTH_CHECK_INTERVALDEFAULT_HEALTH_CHECK_POLICYDEFAULT_HEALTH_CHECK_PROBES HealthCheckHealthCheckPoliciesPingHealthCheck)Retry)ExponentialWithJitterBackoff NoBackoff) WeightedList)EventDispatcherEventDispatcherInterface)DEFAULT_GRACE_PERIODCircuitBreakerPBCircuitBreakerAdapter)DEFAULT_FAILURE_RATE_THRESHOLD!DEFAULT_FAILURES_DETECTION_WINDOWDEFAULT_MIN_NUM_FAILURESCommandFailureDetectorxc@seZdZdZdZdZdS)InitialHealthCheck all_availablemajority_available one_availableN)__name__ __module__ __qualname__ ALL_AVAILABLEMAJORITY_AVAILABLE ONE_AVAILABLEr3r3P/home/ubuntu/.local/lib/python3.10/site-packages/redis/asyncio/multidb/config.pyr)/sr)returncCtSN)rr3r3r3r4default_event_dispatcher5r8c@seZdZUdZdZeed<eedZ eed<dZ e e ed<dZ e eed<dZe eed <eZeed <dZe e ed <d efd dZdS)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_urlr5cCstj|jd}t|S)N) reset_timeout) pybreakerr"rBr#)selfcircuit_breakerr3r3r4default_circuit_breaker[sz&DatabaseConfig.default_circuit_breaker)r-r.r/__doc__r;float__annotations__rdictr>r?rstrr@r rAr"r!rBrCrHr3r3r3r4r:9s   r:c@sbeZdZUdZeeed<eZe e ee fed<e e dddddZe 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.dZ/e0ed<e1j2Z3e1ed<de4fddZ5deefddZ6deefd d!Z7de&fd"d#Z8d S)$ 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_delay: Delay between health check attempts. 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. initial_health_check_policy: Defines the policy used to determine whether the databases setup is healthy during the initial health check. 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_delayhealth_check_policyfailover_strategyfailover_attemptsfailover_delayauto_fallback_intervalr<event_dispatcherinitial_health_check_policyr5c Cst}|jD]Z}|jdtdtdi|jr%|jj|jfi|j}n |jr<|j tdtd|jj|jd}n |jdi|j}|j durN| n|j }| t |||j|jd|jq|S)Nretryr)rWrV)connection_pool)clientrAr;rCr3)rrOr>updaterrr?rPr@ set_retryrArHaddr r;rC)rF databasesdatabase_configrjrAr3r3r4rnsB    zMultiDbConfig.databasescCstt|j|j|jdgS)N)rZr[failure_detection_window)rr'rZr[r\rFr3r3r4default_failure_detectorssz'MultiDbConfig.default_failure_detectorscCstgSr7)rrqr3r3r4default_health_checkssz#MultiDbConfig.default_health_checkscCr6r7)rrqr3r3r4default_failover_strategyr9z'MultiDbConfig.default_failover_strategy)9r-r.r/rIrr:rKr rPrrr rrrXrYrrr&rZintr$r[rJr%r\r]rrr^rr_rr`rrarrbrrrcrrdDEFAULT_AUTO_FALLBACK_INTERVALrerr8rfr r)r0rgr rnrrrsrtr3r3r3r4rN`s8  )           ) rN): dataclassesrrenumrtypingrrrrrE redis.asyncior r r redis.asyncio.multidb.databaser r redis.asyncio.multidb.failoverrrrr&redis.asyncio.multidb.failure_detectorrr!redis.asyncio.multidb.healthcheckrrrrrrrredis.asyncio.retryr redis.backoffrrredis.data_structurer redis.eventrr redis.multidb.circuitr!r"r#redis.multidb.failure_detectorr$r%r&r'rvr)r8r:rNr3r3r3r4s, $  &