i&ddlmZmZddlmZddlmZmZmZm Z ddl Z ddl m Z m Z mZddlmZmZmZmZmZmZmZmZddlmZmZddlmZdd lmZmZdd l m!Z!dd l"m#Z#m$Z$m%Z%dd l&m'Z'm(Z(dd l)m*Z*m+Z+m,Z,m-Z-ddl.m/Z/m0Z0m1Z1m2Z2m3Z3ddl4m5Z5dZ6GddeZ7defdZ8eGddZ9eGddZ:dS)) dataclassfield)Enum)ListOptionalTypeUnionN)ConnectionPoolRedis RedisCluster)DEFAULT_HEALTH_CHECK_DELAYDEFAULT_HEALTH_CHECK_INTERVALDEFAULT_HEALTH_CHECK_POLICYDEFAULT_HEALTH_CHECK_PROBESDEFAULT_HEALTH_CHECK_TIMEOUT HealthCheckHealthCheckPoliciesPingHealthCheck)ExponentialWithJitterBackoff NoBackoff) WeightedList)EventDispatcherEventDispatcherInterface)MaintNotificationsConfig)DEFAULT_GRACE_PERIODCircuitBreakerPBCircuitBreakerAdapter)Database Databases)DEFAULT_FAILOVER_ATTEMPTSDEFAULT_FAILOVER_DELAYFailoverStrategyWeightBasedFailoverStrategy)DEFAULT_FAILURE_RATE_THRESHOLD!DEFAULT_FAILURES_DETECTION_WINDOWDEFAULT_MIN_NUM_FAILURESCommandFailureDetectorFailureDetector)RetryxceZdZdZdZdZdS)InitialHealthCheck all_availablemajority_available one_availableN)__name__ __module__ __qualname__ ALL_AVAILABLEMAJORITY_AVAILABLE ONE_AVAILABLEf/Users/user/workspace/sujinbaek/cqa-test-app/venv/lib/python3.11/site-packages/redis/multidb/config.pyr,r,.s#M-#MMMr7r,returnctSN)rr6r7r8default_event_dispatcherr<4s   r7ceZdZUdZdZeed<eeZ eed<dZ e e ed<dZ e eed<dZe eed <eZeed <dZe e ed <d efd ZdS)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_urlr9cTtj|j}t|S)N) reset_timeout) pybreakerrrFr)selfcircuit_breakers r8default_circuit_breakerz&DatabaseConfig.default_circuit_breakerZs&#2ARSSS&777r7)r0r1r2__doc__r?float__annotations__rdictrBrCrstrrDr rErrrFrGrMr6r7r8r>r>8s0FE%555M4555"Hhsm"""*.Ix'...(,GXn %,,,.L%...&*hsm***8888888r7r>ceZdZUdZeeed<eZe e ee fed<e e dddZe ed <d Zeeeed <eZeed <eZeed <eZeed<d Zeeeed<eZeed<eZeed<e Z!eed<e"Z#eed<e$Z%e&ed<d Z'ee(ed<e)Z*eed<e+Z,eed<e-Z.eed<e/e0Z1e2ed<e3j4Z5e3ed<de6fdZ7deefdZ8deefdZ9de(fd Z:d 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. health_check_timeout: Timeout for the full health check operation (including all probes). 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. 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_timeouthealth_check_policyfailover_strategyfailover_attemptsfailover_delayauto_fallback_intervalr@event_dispatcherinitial_health_check_policyr9c t}|jD]5}tdt|jd<d|jvrt d|jd<|jr|jj|jfi|j}no|jrV|j tdt|j|j}n|jd i|j}|j | n|j }| t|||j|j|j7|S) Nr)r]r\retrymaint_notifications_configF)enabled)connection_pool)clientrEr?rGr6)rrUr)rrBrrCrVrD set_retryrErMaddrr?rG)rK databasesdatabase_configrtrEs r8rwzMultiDbConfig.databasess NN #4) ) O6;9;;666O )' 2 ,?3PPP,U;;; -.JK' L3*3#,0?0M!* L)33!Y[[999*44$3$=5+*KK_-JKK#*2 77999$,  MM!#*1%4%E   &    r7cFt|j|j|jgS)N)r`rafailure_detection_window)r'r`rarbrKs r8default_failure_detectorsz'MultiDbConfig.default_failure_detectorss0 "!%!6'+'B)-)G     r7cFt|j|j|jgS)N)rerfrg)rrerfrgr{s r8default_health_checksz#MultiDbConfig.default_health_checkss0 $($<#'#:%)%>     r7ctSr;)r#r{s r8default_failover_strategyz'MultiDbConfig.default_failover_strategys*,,,r7);r0r1r2rNrr>rPr rVrr r r)rr^r_rr(r&r`intr$rarOr%rbrcrrrdrrer rfrrgrrhrrir"r rjr!rkDEFAULT_AUTO_FALLBACK_INTERVALrlrr<rmrr,r3rnrrwr|r~rr6r7r8rTrT_sT))V>****5:L$uUL012::: 5,,!<<x_ 56===4c444$BEBBB'HuHHH15M8D-.555#@5@@@:::: ::::">%>>>/J,JJJ48x 018886s6662NE222$BEBBB160222.7I6V!3VVV.9....` 4+@     tK'8    -+;------r7rT); dataclassesrrenumrtypingrrrr rJredisr r r !redis.asyncio.multidb.healthcheckr rrrrrrr redis.backoffrrredis.data_structurer redis.eventrrredis.maint_notificationsrredis.multidb.circuitrrrredis.multidb.databaserrredis.multidb.failoverr r!r"r#redis.multidb.failure_detectorr$r%r&r'r( redis.retryr)rr,r<r>rTr6r7r8rs((((((((............5555555555                    BAAAAAAA------AAAAAAAA>>>>>> 76666666 !$$$$$$$$$ ": #8#8#8#8#8#8#8 #8L G-G-G-G-G-G-G- G-G-G-r7