o iQ@sddlmZmZddlmZmZddlZddlmZddlm Z ddl m Z ddl m Z Gdd d eZGd d d eZGd d d eZe eee fZGdddeeZdS))ABCabstractmethod)OptionalUnionN) RedisCluster) WeightedList)CircuitBreaker)Numberc@speZdZeedefddZejedefddZeedee fddZ e jedee fd dZ d S) AbstractDatabasereturncCdS)z]The weight of this database in compare to others. Used to determine the database failover to.Nselfr r J/home/ubuntu/.local/lib/python3.10/site-packages/redis/multidb/database.pyweight zAbstractDatabase.weightrcCr )z5Set the weight of this database in compare to others.Nr rrr r rrrcCr )z6Health check URL associated with the current database.Nr rr r rhealth_check_urlrz!AbstractDatabase.health_check_urlrcCr )z>Set the health check URL associated with the current database.Nr rrr r rrrN) __name__ __module__ __qualname__propertyrfloatrsetterrstrrr r r rr sr c@szeZdZ d dedeefddZedefddZej defd dZedeefd d Z e j deefd d Z dS) BaseDatabaseNrrcCs||_||_dSN)_weight_health_check_url)rrrr r r__init__&s zBaseDatabase.__init__r cC|jSrrrr r rr.zBaseDatabase.weightcC ||_dSrr#rr r rr2 cCr"rr rr r rr6r$zBaseDatabase.health_check_urlcCr%rr'rr r rr:r&r) rrrrrrr!rrrrr r r rr%s rc@seZdZdZeedeeje ffddZ e j edeeje ffddZ eede fddZ e j ed e fd dZ d S) SyncDatabasez5Database with an underlying synchronous redis client.r cCr )zThe underlying redis client.Nr rr r rclientBrzSyncDatabase.clientr)cCr )z Set the underlying redis client.Nr rr)r r rr)HrcCr )z)Circuit breaker for the current database.Nr rr r rcircuitNrzSyncDatabase.circuitr+cCr )z1Set the circuit breaker for the current database.Nr rr+r r rr+TrN)rrr__doc__rrrredisRedisrr)rrr+r r r rr(?sr(c seZdZ ddeejefdedede e ffdd Z e deejeffd d Z e jdeejeffd d Z e defd d Zejdefdd ZddZZS)DatabaseNr)r+rrcs&||_||_||j_t||dS)at Initialize a new Database instance. Args: client: Underlying Redis client instance for database operations circuit: Circuit breaker for handling database failures weight: Weight value used for database failover prioritization health_check_url: Health check URL associated with the current database N)_client_cbdatabasesuperr!)rr)r+rr __class__r rr!_szDatabase.__init__r cCr"rr1rr r rr)tr$zDatabase.clientcCr%rr7r*r r rr)xr&cCr"rr2rr r rr+|r$zDatabase.circuitcCr%rr8r,r r rr+r&cCsd|jd|jdS)NzDatabase(client=z , weight=))r)rrr r r__repr__szDatabase.__repr__r)rrrrr.r/rrrrrr!rr)rr+r: __classcell__r r r5rr0^s( r0)abcrrtypingrrr.rredis.data_structurerredis.multidb.circuitr redis.typingr r rr(tuple Databasesr0r r r rs