Ki}OUdZddlZddlZddlZddlmZmZddlmZddlm Z m Z ddl m Z m Z mZddlmZmZddlmZdd lmZdd lmZmZej2rdd lmZej8eZej>e d <Gd de Z!de!de"ddfdZ#ia$e%e"dfe d<ejLZ'ddZ(de"ddfdZ)GddZ*da+ddZ,y)aSRedis connection management. This module is the single point of control for Redis connections, including the fakeredis backend used for memory:// URLs. This module is designed to be the single point of cluster-awareness, so that other modules can remain simple. When Redis Cluster support is added, only this module will need to change. N)AsyncExitStackasynccontextmanager) TracebackType)AsyncGeneratorProtocol) ParseResulturlparse urlunparse)ConnectionPoolRedis)PubSub) RedisCluster) Connection SSLConnection) FakeServerloggerceZdZdZddZy)AsyncCloseablez3Protocol for objects with an async aclose() method.Nc KywNselfs Y/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/docket/_redis.pyaclosezAsyncCloseable.aclose"ssreturnN)__name__ __module__ __qualname____doc__rrrrrs='r"rresourcenamercK |jd{y7#t$rtjd|dYywxYww)znClose a resource with error handling. Designed to be used with AsyncExitStack.push_async_callback(). NzFailed to close %sTexc_info)r Exceptionrwarning)r#r$s rclose_resourcer*%s? Boo B+TDABs1AA!AAAAr_memory_serverscKt4d{tjdddd{y7)7#1d{7swYyxYww)zqClear all cached FakeServer instances. This is primarily for testing to ensure isolation between tests. N)_memory_servers_lockr+clearrr"rclear_memory_serversr/5sF $         s< A7A; A9AAA A A  AurlzFakeServer | Nonec,tj|S)zpGet the cached FakeServer for a URL, if any. This is primarily for testing to verify server isolation. )r+get)r0s rget_memory_serverr3>s   s ##r"ceZdZUdZedzed<edzed<edzed<eed<eed<de d dfd Z d"d Z d e e dzd e dzdedzd dfdZed efdZed efdZed efdZed edzfdZde d e fdZd e fdZd efdZd efdZ d#ded efdZ d#ded efdZed eeezdffdZed ee dffdZ!de de d e"fd Z#ed ee dffd!Z$y)$RedisConnectionaManages Redis connections for both standalone and cluster modes. This class encapsulates the lifecycle management of Redis connections, hiding whether the underlying connection is to a standalone Redis server or a Redis Cluster. It provides a unified interface for getting Redis clients, pub/sub connections, and publishing messages. Example: async with RedisConnection("redis://localhost:6379/0") as connection: async with connection.client() as r: await r.set("key", "value") N_connection_pool_cluster_client _node_pool_parsed_stackr0rc\||_t||_d|_d|_d|_y)zInitialize a Redis connection manager. Args: url: Redis URL (redis://, rediss://, redis+cluster://, or memory://) N)r0r r9r6r7r8)rr0s r__init__zRedisConnection.__init__^s- } $#r"cKjrJdt_jjd{jrƉj d{_jjfdjjtj dj_ jjfdjjtjdSjd{_ jjfdjjtjdS7?77Zw) z+Connect to Redis when entering the context.z RedisConnection is not reentrantNctddS)Nr7setattrrsrz,RedisConnection.__aenter__..ss7H$)Or"zcluster clientctddS)Nr8r?rsrrAz,RedisConnection.__aenter__..ys|T)Jr"z node poolctddS)Nr6r?rsrrAz,RedisConnection.__aenter__..s7I4)Pr"zconnection pool) is_connectedrr: __aenter__ is_cluster_create_cluster_clientr7callbackpush_async_callbackr*_create_node_poolr8_connection_pool_from_urlr6rs`rrEzRedisConnection.__aenter__js6$$H&HH$$& kk$$&&& ??)-)D)D)F#FD KK !O P KK + + 4 46F #446DO KK !J K KK + +  +/*H*H*J$JD ! KK !P Q KK + + 5 57H  + '$G%Ks8AF F#F 'F(CF -F.AF F F exc_typeexc_valexc_tbcnK |jj|||d{|`y7#|`wxYww)z4Close the Redis connection when exiting the context.N)r: __aexit__)rrLrMrNs rrPzRedisConnection.__aexit__s8 ++'''6B B B  C s 5 .,.5.25c>|jduxs|jduS)z'Check if the connection is established.N)r6r7rs rrDzRedisConnection.is_connecteds&$$D0TD4H4HPT4TTr"c2|jjdvS)z/Check if this connection is to a Redis Cluster.)z redis+clusterrediss+clusterr9schemers rrFzRedisConnection.is_clusters||""&IIIr"c4|jjdk(S)z>Check if this connection is to an in-memory fakeredis backend.memoryrTrs r is_memoryzRedisConnection.is_memorys||""h..r"c|jS)z5Get the cluster client, if connected in cluster mode.)r7rs rcluster_clientzRedisConnection.cluster_clients###r"r$c*|jrd|dS|S)aReturn a prefix, hash-tagged for cluster mode key slot hashing. In Redis Cluster mode, keys with the same hash tag {name} are guaranteed to be on the same slot, which is required for multi-key operations. Args: name: The base name for the prefix Returns: "{name}" for cluster mode, or just "name" for standalone mode {})rF)rr$s rprefixzRedisConnection.prefixs ??vR=  r"c|js |jS|jjj dd}t |jj |S)ahConvert a cluster URL to a standard Redis URL for redis-py. redis-py doesn't support the redis+cluster:// scheme, so we normalize it to redis:// (or rediss://) before passing to RedisCluster.from_url(). Returns: The URL with +cluster removed from the scheme if cluster mode, otherwise the original URL z+cluster)rU)rFr0r9rUreplacer _replace)r new_schemes r_normalized_urlzRedisConnection._normalized_urlsM88O\\((00R@ $,,//z/BCCr"cKtj|j}|jd{|S7w)zCreate and initialize an async RedisCluster client. Returns: An initialized RedisCluster client ready for use N)rfrom_urlrd initialize)rclients rrGz&RedisConnection._create_cluster_clients= ,44T5I5I5KL!!!  "s7AAAct|jJ|jj}|s td|d}t|jt |j |jj|jj|jjdk(r tdStdS)aCreate a connection pool to a cluster node for pub/sub operations. Redis Cluster doesn't natively support pub/sub through the cluster client, so we create a regular connection pool connected to one of the primary nodes. This pool persists for the lifetime of the RedisConnection. Returns: A ConnectionPool connected to a cluster primary node z%No primary nodes available in clusterrrSF)hostportusernamepasswordconnection_classdecode_responses) r7 get_primaries RuntimeErrorr rjintrkr9rlrmrUrr)rnodesnodes rrJz!RedisConnection._create_node_pools##///$$224FG GQxTYY\\**\\**||""&66+#  "  r"rocK|jr|j|d{Stj|j|S7%w)a4Create a Redis connection pool from the URL. Handles real Redis (redis://) and in-memory fakeredis (memory://). Args: decode_responses: If True, decode Redis responses from bytes to strings Returns: A ConnectionPool ready for use with Redis clients N)ro)rX_memory_connection_poolr rfr0)rros rrKz)RedisConnection._connection_pool_from_urlsB >>556FGG G&&txxBRSSHs!A A &A cKddlm}m}ttj |j }|t|||St4d{tj |j }|t|||cdddd{S|}|t|j <t|||cdddd{S7}7@7 #1d{7swYyxYww)z=Create a connection pool for a memory:// URL using fakeredis.r)FakeConnectionrN)rnserverro) fakeredis.aioredisrxr_patch_fakeredis_lua_runtimer+r2r0r r-)rrorxrrys rrvz'RedisConnection._memory_connection_pools B %&!$$TXX.  !!/!1  (  $((2F!%%3!%5    \F(.ODHH %!!/!1        sfA C'C C'/C C' CC''C: C'CC'C'C'C$C C$ C'cK|j|jyt|j4d{}|dddd{y77#1d{7swYyxYww)z?Get a Redis client, handling both standalone and cluster modes.Nconnection_pool)r7r r6)rrs rrhzRedisConnection.client sd    +&& &T-B-BC  q       sD7A-AA-A A-AA-A-A*A! A*&A-cK|j/|j4d{}|dddd{yt|j4d{}|j 4d{}|dddd{dddd{y7x7e#1d{7swYyxYw7\7E72#1d{7swYBxYw79#1d{7swYyxYww)zEGet a pub/sub connection, handling both standalone and cluster modes.Nr})r7_cluster_pubsubr r6pubsub)rpsrrs rrzRedisConnection.pubsub)s    +++-     T-B-BC ! !q88:!! L!! ! ! !      !!!!!! ! ! ! !s!C'BC'B  C'BC'B5C'C1B72C5B;; CB9C C'CC'C' B2&B) 'B2.C'7C9C;C CC CC'C$C C$ C'channelmessagecK|jJt|j4d{}|j||d{cdddd{St|j4d{}|j||d{cdddd{S7y7a7S#1d{7swYyxYw7J727$#1d{7swYyxYww)z'Publish a message to a pub/sub channel.Nr})r7r r8publishr6)rrrrs rrzRedisConnection.publish4s    +T__= 9 9YYw88 9 9 9T-B-BC 9 9qYYw88 9 9 9 98 9 9 9 9 98 9 9 9 9s'CB#CB)B%B) CB'C3B>4C7C CC CCC%B)'C)B;/B2 0B;7CCCC C CCctKt|j}|j} | |jd{ |jd{y7#t$rt j ddY>wxYw7-#t$rt j ddYywxYw# |jd{7n&#t$rt j ddYnwxYw |jd{7w#t$rt j ddYwwxYwxYww)aCreate a pub/sub connection using the shared node pool. Redis Cluster doesn't natively support pub/sub through the cluster client, so we use a regular Redis client connected to one of the primary nodes. The underlying connection pool is managed by the RedisConnection lifecycle. Yields: A PubSub object connected to a cluster node r}NzFailed to close cluster pubsubTr&zFailed to close cluster client)r r8rrr(rr))rrhrs rrzRedisConnection._cluster_pubsub=st7 PL Pmmo%% Pmmo%% & P?$O P& P?$O P  Pmmo%% P?$O P Pmmo%% P?$O Ps'D8B2A$A"A$ B B B !D8"A$$ BD8BD8 B B/,D8.B//D82D54CC C D5 C1.D50C11D55DD  DD5 D2/D51D22D55D8)rr5)F)%rrr r!r __annotations__rrrstrr<rEtype BaseExceptionrrPpropertyboolrDrFrXrZr^rdrGrJrKrvrrr rhr rrrrrrr"rr5r5Fs %t++!D((%%   C D 8 }%, % $   UdUUJDJJ/4//$ t 3$$33" D Dl > 6(-T $T T$(-# $# #JnU\-A4-GH!nVT\:!!9S93939P~fdl'CPPr"r5Fc h trydaddl ddl ddlm ddlm}m}ddlm ddl m }m m mm m  ddl m|t&|ft&f j(d |d t&d t*d t&d t,j.f  f d }||_y#t$$rYywxYw)NTr)_msgs)Intcommand) SimpleError)ScriptingCommandsMixin_check_for_lua_globals_lua_cjson_decode_lua_cjson_encode_lua_cjson_null_lua_redis_log) LUA_MODULE)flagsrscriptnumkeys keys_and_argsrc  |t|kDrj|dkrjj|j j }|j j|<tj dsejddj _ j j djjDcgc] }|d|d c}} jd|d } jd j _|d d d dddt j!j#j _j j$ dgj _dt(j*dt(j,ff fd }dt(j*dt(j,ff fd } |j _| j _j3 j _j3 j _j3 j _|j j.j j0j j4j j6j j8j j j j$ j j&d<j j j;|d| j;||d j=|}   j=djS| dScc}w#$r} | j>j@k(r{jBdkrjDj jFdk(r/jHjK|jMj@jBdkr0jNjK|jM| | j>d} ~ wjP$r5} jNjK|jM| d} ~ wwxYw)Nr _lua_runtimeT)encodingunpack_returned_tuples z = require('z')aA function(redis_call, redis_pcall, redis_log, cjson_encode, cjson_decode, cjson_null) redis = {} redis.call = redis_call redis.pcall = redis_pcall redis.log = redis_log redis.LOG_DEBUG = 0 redis.LOG_VERBOSE = 1 redis.LOG_NOTICE = 2 redis.LOG_WARNING = 3 redis.error_reply = function(msg) return {err=msg} end redis.status_reply = function(msg) return {ok=msg} end cjson = {} cjson.encode = cjson_encode cjson.decode = cjson_decode cjson.null = cjson_null KEYS = {} ARGV = {} z% end z function(keys, argv) KEYS = keys ARGV = argv end cyrrargss rrAzD_patch_fakeredis_lua_runtime..patched_eval..r"cyrrrs rrAzD_patch_fakeredis_lua_runtime..patched_eval..rr"cyrrrs rrAzD_patch_fakeredis_lua_runtime..patched_eval..rr"cyrrrs rrAzD_patch_fakeredis_lua_runtime..patched_eval..rr"cyrrrs rrAzD_patch_fakeredis_lua_runtime..patched_eval..rr"r.cjdtdtjdtjffd }|S)NoprrcLjd}|j|g|SNr)_lua_current_socket_lua_redis_callrrsocketeglrrys rwrapperzd_patch_fakeredis_lua_runtime..patched_eval..make_redis_call_wrapper..wrappers/#77:F1611"b"DtDDr"_serverbytestypingAnyrrrryexpected_globals lua_runtimers @@@rmake_redis_call_wrapperzS_patch_fakeredis_lua_runtime..patched_eval..make_redis_call_wrappersC %EEfjjEVZZEr"cjdtdtjdtjffd }|S)NrrrcLjd}|j|g|Sr)r_lua_redis_pcallrs rrze_patch_fakeredis_lua_runtime..patched_eval..make_redis_pcall_wrapper..wrappers/#77:F26222r2EEEr"rrs @@@rmake_redis_pcall_wrapperzT_patch_fakeredis_lua_runtime..patched_eval..make_redis_pcall_wrappersC %FFfjjFVZZFr")valkeyzcollectgarbage()F)nested)*lenTOO_MANY_KEYS_MSGNEGATIVE_KEYS_MSGsha1 hexdigestencoder script_cachehasattr LuaRuntimerjoinload_lua_moduleseval_lua_set_keys_argvsetglobalskeys_lua_expected_globalsrrCallabler_lua_redis_call_wrapper_lua_redis_pcall_wrapperpartial_lua_log_partial_lua_cjson_encode_partial_lua_cjson_decode_partial table_fromexecutevalueLUA_COMMAND_ARG_MSGversionLUA_COMMAND_ARG_MSG6 server_typeVALKEY_LUA_COMMAND_ARG_MSGformatdecodeSCRIPT_ERROR_MSGLuaError_convert_lua_result)rrrrrmodulemodules_import_strset_globals_initrrresultexrrrrrrrrr functoolshashlibmsgss` @@r patched_evalz2_patch_fakeredis_lua_runtime..patched_evals\ S' 'd445 5 Q;d445 5||F#--/668*0 !!$'t||^4(2(=(=d)>)DLL %,,33K!%AEAVAVWvF8<xr2W"  +//((())  6/:.>.>/DLL + """""  25[5H5H5J5O5O5Q1RDLL .#||AA 15vDLL , V__S&**_-M  fooc6::o.N 4K3LDLL 04L4NDLL 1,5,=,= -=-DLL )6?5F5F!;0@6DLL 26?5F5F!;0@6DLL 2  44 55 -- 66 66  ll// <<==/3 ((+ ''  " "='#: ;  " "=#: ;  O ((0F" {,<= ./''u'==QXf (xx4333<<$&%d&?&?@@\\--9%77>>t{{}M&d&>&>??||d"!$"7"7">">t{{}b"QRRbhh' '"" Od33::4;;="MN N Os+*OOS8 C%R11S80S33S8)_lua_patch_appliedrr fakeredisrfakeredis._commandsrrfakeredis._helpersr)fakeredis.commands_mixins.scripting_mixinrrrrrrr ImportErrorrFLAG_NO_SCRIPTrrrrr)rrrrrrrrrrrrrrs @@@@@@@@@@rr{r{es'0.H eS\E84+>+>?]>$]>]>]> ]>  ]>]>@]>~#/G sB%% B10B1r)-r!asynciologgingr contextlibrrtypesrrr urllib.parserr r redis.asyncior r redis.asyncio.clientr redis.asyncio.clusterrredis.asyncio.connectionrr TYPE_CHECKINGrzr getLoggerrrLoggerrrrr*r+dictLockr-r/r3r5rr{rr"rr s :+::/'.> -+**844(X( B>BBB,.c<'(-#w||~ $3$#6$NPNPx{/r"