U h0 @s<UddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlmZmZmZmZmZmZmZmZmZmZmZmZmZddlmZmZmZm Z ddl!m"Z"e"rddl#Z#dd l#m$Z$m%Z%n dZ#dZ%dZ$dd l&m'Z'dd l(m)Z)m*Z*dd l!m+Z+m,Z,ej-d kr6ddlm.Z/n ddl/m.Z/ddl0m1Z1ddl2m3Z3ddl4m5Z5ddl6m7Z7m8Z8ddl9m:Z:m;Z;mZ>m?Z?m@Z@ddlAmBZBddlCmDZDmEZEmFZFddlGmHZHmIZImJZJmKZKmLZLdZMdZNdZOdZPdZQGdddejRZSeSjTZUeeeKeLeJfeVd<eDr$eJZWneKZWGdd d eZXGd!d"d"eZYeeXeYfZZGd#d$d$Z[Gd%d&d&e[Z\Gd'd(d(e\Z]Gd)d*d*Z^Gd+d,d,e[Z_d-Z`eead.d/d0Zbe ecededebebececebedd1 Zeeefed2egffeVd3<Gd4d5d5ed6d7Zhefehd8d9d:Zied;dd<d<ZkGd?d@d@ekZldS)AN)abstractmethod)chain)MappingProxyType) AnyCallableIterableListMappingOptionalProtocolSetTupleType TypedDictTypeVarUnion) ParseResultparse_qsunquoteurlparse) SSL_AVAILABLE) SSLContext TLSVersion)TokenInterface)!AsyncAfterConnectionReleasedEventEventDispatcher)deprecated_argsformat_error_message) r)timeout)Retry) NoBackoff)DEFAULT_RESP_VERSION)CredentialProvider"UsernamePasswordCredentialProvider)AuthenticationError$AuthenticationWrongNumberOfArgsErrorConnectionError DataError RedisError ResponseError TimeoutError) EncodableT)HIREDIS_AVAILABLEget_lib_version str_if_bytes) BaseParserEncoder_AsyncHiredisParser_AsyncRESP2Parser_AsyncRESP3Parser*$s  c@seZdZeZdS) _SentinelN)__name__ __module__ __qualname__objectsentinelrArArKrArArArBrD_srDc@seZdZddddZdS)AsyncConnectCallbackProtocolrErFcsdSrHrArIrArArBrKdr:z%AsyncConnectCallbackProtocol.__call__NrLrArArArBrMcsrMc@seZdZdZdZdddddedddedddd edddedd dd e e e fe e e e e e ee eefe e eeee e e e e e e e e e e ee eeee ee e e ed d d ZefedddZddZddZeddZeddZ ddZ!ddZ"eeddddZ#d d!Z$d\ed#d$d%Z%ed&d'Z&ee d(d)d*Z'e(e d+d,d-Z)d.d/Z*dd(d0d1Z+d]edd2d3d4Z,d^edd5d6d7Z-d8d9Z.d:d;Z/dd?d@Z3d_e e2e e1e2feddAdBdCZ4eeddDdEdFZ5dGdHZ6d`d"ddIee e ee edJdKdLZ7e8e9e2dMdNdOZ:e1e1e8e9e2dPdQdRZ;dSdTZdWdXdYZ?dZd[Z@dS)arEz0Manages communication to and from a Redis server)dbusername client_namelib_name lib_versioncredential_providerpasswordsocket_timeoutsocket_connect_timeoutredis_connect_funcretry_on_timeoutretry_on_errorhealth_check_intervalnext_health_checkZlast_active_atencoder ssl_contextprotocol_reader_writer_parser_connect_callbacks_buffer_cutoff_lock_socket_read_size__dict__rNFutf-8strictizredis-pyr)rNrTrUrVrXrYencodingencoding_errorsdecode_responses parser_classsocket_read_sizerZrPrQrRrOretryrW encoder_classrSr^event_dispatchercCs|s|r|dk rtd|dkr*t|_n||_||_| |_||_||_||_||_||_ ||_ |dkrl|}||_ ||_ |t krg}|r|t|tj|tj||_|s|r|sttd|_n t||_|j|nttd|_| |_d|_|||| |_||_d|_d|_| |_ |!| g|_"d|_#d|_$zHz t'|}Wn6t(k rjt)}Ynt*k rt%d YnXW5|dks|dkrt%d||_&XdS) Nz'username' and 'password' cannot be passed along with 'credential_provider'. Please provide only one of the following arguments: 1. 'password' and (optional) 'username' 2. 'credential_provider'riprrzprotocol must be either 2 or 3zprotocol must be an integer)+r*r_event_dispatcherrNrPrQrRrSrTrOrUrVrXSENTINELappendr-socketr!asynciorYr"r#rncopydeepcopyZupdate_supported_errorsrZr[r\rWr_r`re set_parserrbrc_re_auth_tokenr)r^int TypeErrorr$ ValueError)rJrNrTrUrVrXrYrirjrkrlrmrZrPrQrRrOrnrWrorSr^rpprArArB__init__sh       zAbstractConnection.__init__) _warningscCsPt|ddrL|jd|t|dzt|Wntk rJYnXdS)Nr`zunclosed Connection )source)getattrwarnResourceWarningrwget_running_loop_close RuntimeError)rJrrArArB__del__s  zAbstractConnection.__del__cCs |jr|jd|_|_dS)zR Internal method to silently close the connection without waiting N)r`closer_rJrArArBrs zAbstractConnection._closecCs8ddd|D}d|jjd|jjd|dS)N,css |]\}}|d|VqdS)=NrA).0kvrArArB sz.AbstractConnection.__repr__..<.()>)join repr_pieces __class__r=r<)rJZ repr_argsrArArB__repr__szAbstractConnection.__repr__cCsdSrHrArrArArBrszAbstractConnection.repr_piecescCs|jdk o|jdk SrH)r_r`rrArArB is_connectedszAbstractConnection.is_connectedcCs$t|}||jkr |j|dS)a^ Register a callback to be called when the connection is established either initially or reconnected. This allows listeners to issue commands that are ephemeral to the connection, for example pub/sub subscription or key tracking. The callback must be a _method_ and will be kept as a weak reference. N)weakref WeakMethodrbru)rJcallbackZwmrArArBregister_connect_callbacks  z,AbstractConnection.register_connect_callbackcCs0z|jt|Wntk r*YnXdS)z De-register a previously registered callback. It will no-longer receive notifications on connection events. Calling this is not required when the listener goes away, since the callbacks are kept as weak methods. N)rbremoverrr~)rJrrArArBderegister_connect_callbacksz.AbstractConnection.deregister_connect_callback)rlreturncCs||jd|_dS)z Creates a new instance of parser_class with socket size: _socket_read_size and assigns it to the parser for the connection :param parser_class: The required parser class )rmN)rera)rJrlrArArBrzszAbstractConnection.set_parsercs|jddIdHdS)z5Connects to the Redis server if not already connectedT check_healthN)connect_check_healthrrArArBconnect&szAbstractConnection.connectTrc sjr dSz(jfddfddIdHWntjk rJYn|tjtjfk rltdYnZtk r}zt |W5d}~XYn,t k r}zt ||W5d}~XYnXzDj sj |dIdHn&tj r IdHn Wn&tk r0IdHYnXddjD_jD]0}|}|}|rJt|rJ|IdHqJdS)NcsSrH)_connectrArrArB/r:z9AbstractConnection.connect_check_health..csSrH disconnecterrorrrArBr/r:zTimeout connecting to serverrcSsg|]}|r|qSrArA)rrefrArArB Msz;AbstractConnection.connect_check_health..)rrncall_with_retryrwZCancelledErrorrvr!r-OSErrorr)_error_message ExceptionrWon_connect_check_healthiscoroutinefunctionr+rrbinspect isawaitable)rJreexcrrZtaskrArrBr*s>    z'AbstractConnection.connect_check_healthcsdSrHrArrArArBrTszAbstractConnection._connectrcCsdSrHrArrArArB _host_errorXszAbstractConnection._host_error) exceptionrcCst||SrH)rr)rJrrArArBr\sz!AbstractConnection._error_messagecCs|jSrH)r^rrArArB get_protocol_szAbstractConnection.get_protocolcs|jddIdHdS)z=Initialize the connection, authenticate and select a databaseTrN)rrrArArB on_connectbszAbstractConnection.on_connect)rrc s|j||j}d}|js(|js(|jrJ|jp:t|j|j}|IdH}|r|jdkrt|jt r| t |j |j_ |j|t |dkrd|dg}|jd|jdf|ddiIdH|IdH}|d t|jkr|d t|jkrtd n|r|jd|ddiIdHz|IdH}Wn>tk rn|jd|d dd IdH|IdH}YnXt|dkrtdn`|jdkrt|jt r| t |j |j_ |j||jd|j|d IdH|IdH}|jr*|jdd|j|d IdHt|IdHdkr*td|jrN|jddd|j|d IdH|jrr|jddd|j|d IdH|jr|jd|j|d IdHdd|j|jfDD]0}z|IdHWntk rYnXq|jrt|IdHdkrtddS)N)r2rqdefaultrZHELLOAUTHrFsprotoprotozInvalid RESP versionrrrOKzInvalid Username or PasswordZCLIENTZSETNAMEzError setting client nameZSETINFOzLIB-NAMEzLIB-VERZSELECTcss|]}|r|VqdSrHrA)rsentrArArBrsz=AbstractConnection.on_connect_check_health..zInvalid Database)r)rarrSrOrTr&Zget_credentials_asyncr^ isinstancer5rzr6ZEXCEPTION_CLASSESlen send_command read_responsegetr|r)r(r1r'rPrQrRrNr,)rJrparserZ auth_argsZ cred_providerresponseZ auth_response_rArArBrfs                   z*AbstractConnection.on_connect_check_health)nowaitrc szt|j4IdHx|j|js|}t|trN|g}|jrpt | ||jIdHn|j ||j IdHWntjk r|jddIdHtddYntk rB}zd|jddIdHt|jdkrd|jd}}n|jd}|jd}td |d |d |W5d}~XYn*tk rj|jddIdHYnXdS) NFrTrzTimeout writing to socketrqUNKNOWNrzError z while writing to socket. r)rrrrstrencodebytesrUrwwait_forrr`rrr-rrrargsr) BaseException)rJrrrZerr_noerrmsgrArArBsend_packed_commandsB      z&AbstractConnection.send_packed_command)rkwargsrcs&|j|j||dddIdHdS)z+Pack and send a command to the Redis serverrTrN)r pack_commandr)rJrrrArArBr(s zAbstractConnection.send_commandc sjz|jIdHWStk rd}z4|jddIdH|}td|d|jW5d}~XYnXdS)z8Poll the socket to see if there's data that can be read.NTrError while reading from z: )racan_read_destructiverrrr)r)rJr host_errorrArArBr.s z'AbstractConnection.can_read_destructive)disconnect_on_error push_request)disable_decodingr!rrc s|dk r |n|j}|}z|dk rl|jdkrltslt|4IdH|jj||dIdH}W5QIdHRXnz|dk rt|4IdH|jj|dIdH}W5QIdHRXn:|jdkrts|jj||dIdH}n|jj|dIdH}Wntjk r6|dk r YdS|r$|j ddIdHtd|Ynt k r}z2|r`|j ddIdHt d|d |j W5d}~XYn0t k r|r|j ddIdHYnX|jrt|j} | |_t|tr|d|S) z0Read the response from a previously sent commandN)3r)rr)rTrzTimeout reading from rz : )rUrr^r/rrarrwr-rrr)rrrZrrr[rr,) rJrr!rr read_timeoutrrrZ next_timerArArBr7s^  & z AbstractConnection.read_response)rrc Gs*g}t|dtrtt|dtrFt|d|dd}n(d|dkrnt|d|dd}tt tt |t f}|j }t |jj|D]|}t |}t ||ks||kst|trt|tt|t f}||||t }qt|tt|t |t f}q|||S)z2Pack a series of arguments into the Redis protocolrrqN )rfloatAssertionErrorrtuplersplit SYM_EMPTYrSYM_STARrSYM_CRLFrcmapr\ memoryview SYM_DOLLARru)rJroutputZbuff buffer_cutoffargZ arg_lengthrArArBrusB"      zAbstractConnection.pack_command)commandsrc Csg}g}d}|j}|D]}|j|D]r}t|}||ksJ||ksJt|trf|r^|t|d}g}||ksxt|tr||q$||||7}q$q|r|t||S)z.Pack multiple commands into the Redis protocolr)rcrrrrrurr) rJrrpiecesZ buffer_lengthrcmdchunkZchunklenrArArB pack_commandss0   z AbstractConnection.pack_commandscCst|jjdkS)zCheck if the socket is emptyr)rr__bufferrrArArB_socket_is_emptysz#AbstractConnection._socket_is_emptycs |s|jddIdHqdS)NT)r)rrrrArArBprocess_invalidation_messagessz0AbstractConnection.process_invalidation_messagestokencCs ||_dSrH)r{rJrrArArBset_re_auth_tokensz$AbstractConnection.set_re_auth_tokencsD|jdk r@|d|jd|jIdH|IdHd|_dSNroid)r{rtry_get get_valuerrrArArBre_auths   zAbstractConnection.re_auth)T)T)F)T)FN)Ar<r=r>__doc__ __slots__rtrCr0r3rrr|r rboollistr;rr2r"ConnectCallbackTr%rrwarningsrrrrrrpropertyrrrrzrrrrrrrrrrrrrrrrrrrrr.rrrrrrrrrArArArBrEjs    Y    * l )  >.rEc seZdZdZddddddeeeefeee eeee ffedfdd Z d d Z e d d dZ ddZed ddZZS) Connectionz4Manages TCP communication to and from a Redis server localhostiFNr)hostportsocket_keepalivesocket_keepalive_options socket_typec s8||_t||_||_|pi|_||_tjf|dSrH)rr|rrrrsuperr)rJrrrrrrrrArBrs   zConnection.__init__cCs6d|jfd|jfd|jfg}|jr2|d|jf|S)NrrrNrP)rrrNrPrurJrrArArBrszConnection.repr_piecesrcCs|j|jdS)NrrrrrArArB_connection_argumentssz Connection._connection_argumentsc st|j4IdH tjf|IdH\}}W5QIdHRX||_||_|jd}|r| t j t j dz@|j r| t jt jd|jD]\}}| t j||qWn"ttfk r|YnXdS)zCreate a TCP socket connectionNrvrq)rrVrwZopen_connectionrr_r` transportZget_extra_info setsockoptrv IPPROTO_TCP TCP_NODELAYr SOL_SOCKET SO_KEEPALIVEritemsSOL_TCPrr}r)rJreaderwritersockrrrArArBrs" zConnection._connectcCs|jd|jS)N:rrrArArBr szConnection._host_error)r<r=r>rrrr|r r r rrrrrr __classcell__rArArrBrs" rc seZdZdZdeeeeeeejfeeeee ee eedfdd Z e dfd d Z ed d Zed dZeddZeddZeddZeddZeddZZS) SSLConnectionzManages SSL connections to and from the Redis server(s). This class extends the Connection class, adding SSL functionality, and making use of ssl.SSLContext (https://docs.python.org/3/library/ssl.html#ssl.SSLContext) NrequiredT) ssl_keyfile ssl_certfile ssl_cert_reqs ssl_ca_certs ssl_ca_datassl_check_hostnamessl_min_version ssl_ciphersc s8ts tdt||||||||d|_tjf| dS)N$Python wasn't built with SSL supportkeyfilecertfile cert_reqsca_certsca_datacheck_hostname min_versionciphers)rr+RedisSSLContextr]rr) rJr*r+r,r-r.r/r0r1rrrArBrs  zSSLConnection.__init__rcst}|j|d<|S)Nssl)rrr]rrJrrrArBr1s z#SSLConnection._connection_argumentscCs|jjSrH)r]r4rrArArBr46szSSLConnection.keyfilecCs|jjSrH)r]r5rrArArBr5:szSSLConnection.certfilecCs|jjSrH)r]r6rrArArBr6>szSSLConnection.cert_reqscCs|jjSrH)r]r7rrArArBr7BszSSLConnection.ca_certscCs|jjSrH)r]r8rrArArBr8FszSSLConnection.ca_datacCs|jjSrH)r]r9rrArArBr9JszSSLConnection.check_hostnamecCs|jjSrH)r]r:rrArArBr:NszSSLConnection.min_version)NNr)NNTNN)r<r=r>rr rrr= VerifyModer rrr rrr4r5r6r7r8r9r:r'rArArrBr(sD       r(c @sbeZdZdZd eeeeeeeejfeeeee ee eedddZ e ddd Z dS) r<) r4r5r6r7r8contextr9r:r;NFr3c Csts td||_||_|dkr(tj}n:t|trbtjtjtj d} || krZtd|| |}||_ ||_ ||_ |j tjkr|nd|_ ||_||_d|_dS)Nr2)noneoptionalr)z+Invalid SSL Certificate Requirements Flag: F)rr+r4r5r= CERT_NONErr CERT_OPTIONAL CERT_REQUIREDr6r7r8r9r:r;r@) rJr4r5r6r7r8r9r:r;Z CERT_REQSrArArBr`s0  zRedisSSLContext.__init__rcCs|jst}|j|_|j|_|jr<|jr<|j|j|jd|j sH|j rZ|j |j |j d|j dk rl|j |_ |jdk r||j||_|jS)N)r5r4)cafilecadata)r@r=create_default_contextr9r6 verify_moder5r4load_cert_chainr7r8load_verify_locationsr:minimum_versionr; set_ciphers)rJr@rArArBrs     zRedisSSLContext.get)NNNNNFNN)r<r=r>r r rrr=r?r rrrrrArArArBr<Ss( 'r<csdeZdZdZddedfddZeeeeee ffdddZ d d Z edd d Z Z S) UnixDomainSocketConnectionz4Manages UDS communication to and from a Redis serverpathc s||_tjf|dSrH)rQrr)rJrQrrrArBrsz#UnixDomainSocketConnection.__init__rcCs.d|jfd|jfg}|jr*|d|jf|S)NrQrNrP)rQrNrPrurrArArBrsz&UnixDomainSocketConnection.repr_piecesc sZt|j4IdHtj|jdIdH\}}W5QIdHRX||_||_|IdHdS)NrP)rrVrwZopen_unix_connectionrQr_r`r)rJr#r$rArArBrs (z#UnixDomainSocketConnection._connectcCs|jSrHrPrrArArBrsz&UnixDomainSocketConnection._host_error)r<r=r>rrrrr rr|rrrr'rArArrBrNs "rN)0FFALSENNOrcCs6|dks|dkrdSt|tr.|tkr.dSt|S)NrOF)rrupper FALSE_STRINGSr )valuerArArBto_bools rZ) rNrUrVrrXmax_connectionsrZr/r!.URL_QUERY_ARGUMENT_PARSERSc@sJeZdZUeed<eed<eeed<eed<eed<eed<eed<dS) ConnectKwargsrOrTconnection_classrrrNrQN)r<r=r>r__annotations__rrEr|rArArArBr]s  r]F)total)urlrc Cst|}i}t|jD]v\}}|rt|dkrt|d}t|}|rz||||<Wqtt fk rt d|dYqXq|||<q|j rt|j |d<|j rt|j |d<|j dkr|j rt|j |d<t|d<n|j d kr||jrt|j|d <|jrt|j|d <|j rfd |krfztt|j d d|d <Wntt fk rdYnX|j dkrt|d<nd}t d|d|S)NrzInvalid value for 'z' in connection URL.rOrTunixrQr^)ZredisredissrrrN/rOrczredis://, rediss://, unix://z5Redis URL must specify one of the following schemes ())rrqueryr!rrr\rr}r~rOrTschemerQrNhostnamerr|replaceAttributeErrorr()raparsedrnameZ value_listrYrZ valid_schemesrArArB parse_urlsJ         rm_CPConnectionPool)boundc@seZdZdZeeeeedddZe dfee e e dddZ d d Zd d Zed ddZedgdddd1ddZddZddZddZe dddZe ddd Zd2ed"d#d$Zdd d%d&Zd'dd(d)d*Zed+d,d-Zed.d/d0ZdS)3roa Create a connection pool. ``If max_connections`` is set, then this object raises :py:class:`~redis.ConnectionError` when the pool's limit is reached. By default, TCP connections are created unless ``connection_class`` is specified. Use :py:class:`~redis.UnixDomainSocketConnection` for unix sockets. Any additional keyword arguments are passed to the constructor of ``connection_class``. )clsrarcKst|}|||f|S)a Return a connection pool configured from the given URL. For example:: redis://[[username]:[password]]@localhost:6379/0 rediss://[[username]:[password]]@localhost:6379/0 unix://[username@]/path/to/socket.sock?db=0[&password=password] Three URL schemes are supported: - `redis://` creates a TCP socket connection. See more at: - `rediss://` creates a SSL wrapped TCP socket connection. See more at: - ``unix://``: creates a Unix Domain Socket connection. The username, password, hostname, path and all querystring values are passed through urllib.parse.unquote in order to replace any percent-encoded values with their corresponding characters. There are several ways to specify a database number. The first value found will be used: 1. A ``db`` querystring option, e.g. redis://localhost?db=0 2. If using the redis:// or rediss:// schemes, the path argument of the url, e.g. redis://localhost/0 3. A ``db`` keyword argument to this function. If none of these options are specified, the default db=0 is used. All querystring options are cast to their appropriate Python types. Boolean arguments can be specified with string values "True"/"False" or "Yes"/"No". Values that cannot be properly cast cause a ``ValueError`` to be raised. Once parsed, the querystring arguments and keyword arguments are passed to the ``ConnectionPool``'s class initializer. In the case of conflicting arguments, querystring arguments always win. )rmupdate)rqrarZ url_optionsrArArBfrom_urls+ zConnectionPool.from_urlNr^r[cKs|pd}t|tr|dkr"td||_||_||_g|_t|_|j dt |_ t |_|j dd|_|jdkr~t|_dS)Nlrz,"max_connections" must be a positive integerrorp)rr|r~r^connection_kwargsr[_available_connectionsset_in_use_connectionsrr3rorwLockrdrsr)rJr^r[rurArArBrHs  zConnectionPool.__init__cCs*d|jjd|jjd|jf|jdS)Nrrrr)rr=r<r^rurrArArBr^s(zConnectionPool.__repr__cCsg|_t|_dSrH)rvrWeakSetrxrrArArBresetdszConnectionPool.resetrcCs|jpt|j|jkS)z;Return True if a connection can be retrieved from the pool.)rvrrxr[rrArArBcan_get_connectionhsz!ConnectionPool.can_get_connection*)Use get_connection() without args instead5.3.0Z args_to_warnreasonversionc sh|j4IdHJ|}z||IdHWn&tk rR||IdHYnXW5QIdHRX|SrH)rdget_available_connectionensure_connectionrrelease)rJ command_namekeysoptionsrGrArArBget_connectionoszConnectionPool.get_connectioncCsVz|j}Wn6tk rDt|j|jkr8tdd|}YnX|j||S)zCGet a connection from the pool, without making sure it is connectedzToo many connectionsN) rvpop IndexErrorrrxr[r)make_connectionaddrIrArArBrs  z'ConnectionPool.get_available_connectioncCs.|j}|j|dd|dd|dddS)z,Return an encoder based on encoding settingsrirgrjrhrkF)rirjrk)rurorr>rArArB get_encoders    zConnectionPool.get_encodercCs|jf|jS)z=Create a new connection. Can be overridden by child classes.)r^rurrArArBrszConnectionPool.make_connectionrFc s~|IdHz|IdHr(tddWnNtttfk rx|IdH|IdH|IdHrttddYnXdS)z8Ensure that the connection object is connected and validNzConnection has datazConnection not ready)rrr)r-rrrIrArArBrsz ConnectionPool.ensure_connectioncs2|j||j||jt|IdHdS)z(Releases the connection back to the poolN)rxrrvrursZdispatch_asyncrrIrArArBrs   zConnectionPool.releaseT)inuse_connectionscsZ|rt|j|j}n|j}tjdd|DddiIdH}tdd|Dd}|rV|dS)z Disconnects connections in the pool If ``inuse_connections`` is True, disconnect connections that are current in use, potentially by other tasks. Otherwise only disconnect connections that are idle in the pool. css|]}|VqdSrHr)rrGrArArBrsz,ConnectionPool.disconnect..Zreturn_exceptionsTNcss|]}t|tr|VqdSrH)rr)rrrArArBrs )rrvrxrwZgathernext)rJr connectionsresprrArArBrs  zConnectionPool.disconnectcs|IdHdS)z-Close the pool, disconnecting all connectionsNrrrArArBacloseszConnectionPool.acloser")rnrcCs(|jD] }||_q|jD] }||_qdSrH)rvrnrx)rJrnconnrArArB set_retrys  zConnectionPool.set_retryrc sj4IdHrjD]NjfddfddIdHjfddfddIdHqjD]qlW5QIdHRXdS)NcsddSr)rrrrA)rrrArBrsz1ConnectionPool.re_auth_callback..cs |SrH_mockrrrArBrr:csSrH)rrA)rrArBrr:cs |SrHrrrrArBrr:)rdrvrnrrxrrrA)rrJrrBre_auth_callbacks     zConnectionPool.re_auth_callbackrcsdS)z Dummy functions, needs to be passed as error callback to retry object. :param error: :return: NrArrArArBrszConnectionPool._mock)N)T) r<r=r>r classmethodrrnrrsrrEr r|rrr{r r|rrrrrrrrrrrrr+rrArArArBro s8 0     csveZdZdZddeejfeeee e e ej dfdd Z e dgdd d dfd d Ze dfdd ZZS)BlockingConnectionPoola  A blocking connection pool:: >>> from redis.asyncio import Redis, BlockingConnectionPool >>> client = Redis.from_pool(BlockingConnectionPool()) It performs the same function as the default :py:class:`~redis.asyncio.ConnectionPool` implementation, in that, it maintains a pool of reusable connections that can be shared by multiple async redis clients. The difference is that, in the event that a client tries to get a connection from the pool when all of connections are in use, rather than raising a :py:class:`~redis.ConnectionError` (as the default :py:class:`~redis.asyncio.ConnectionPool` implementation does), it blocks the current `Task` for a specified number of seconds until a connection becomes available. Use ``max_connections`` to increase / decrease the pool size:: >>> pool = BlockingConnectionPool(max_connections=10) Use ``timeout`` to tell it either how many seconds to wait for a connection to become available, or to block forever: >>> # Block forever. >>> pool = BlockingConnectionPool(timeout=None) >>> # Raise a ``ConnectionError`` after five seconds if a connection is >>> # not available. >>> pool = BlockingConnectionPool(timeout=5) 2)r[r!r^ queue_classc s,tjf||d|t|_||_dS)Nrt)rrrw Condition _conditionr!)rJr[r!r^rrurrArBr s zBlockingConnectionPool.__init__r}r~rrNc szf|j4IdHHt|j4IdH$|j|jIdHt}W5QIdHRXW5QIdHRXWn.tjk r}zt d|W5d}~XYnXz| |IdH|WSt k r| |IdHYnXdS)z@Gets a connection from the pool, blocking until one is availableNzNo connection available.) rrr!rr|rrrwr-r)rrr)rJrrrrGerrrrArBrs.z%BlockingConnectionPool.get_connectionrFc s@|j4IdH"t|IdH|jW5QIdHRXdS)z)Releases the connection back to the pool.N)rrrnotifyrIrrArBr3szBlockingConnectionPool.release)N)r<r=r>rrrw LifoQueuer|r rrEQueuerrrrr'rArArrBrs$#r)mrwrxenumrrvsysr rabcr itertoolsrtypesrtypingrrrrr r r r r rrrr urllib.parserrrrutilsrr=rrZ auth.tokenreventrrrr version_infor!rZredis.asyncio.retryr"Z redis.backoffr#Zredis.connectionr$Zredis.credentialsr%r&Zredis.exceptionsr'r(r)r*r+r,r-Z redis.typingr.Z redis.utilsr/r0r1_parsersr2r3r4r5r6rrrZSYM_LFrEnumr;r@rtr_rCrDrMr rErr(r<rNrXr rZr|rr\rr?r]rmrnrorrArArArBs   <       $  q9CE 4 `