o i&4@s^dZddlmZddlmZmZmZmZmZddl Z er.ddl m Z ddl m Z ddlmZdZd Zd Zd Zd Zd ZdZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'd Z(d!Z)d"Z*d#Z+d$Z,d%Z-d&Z.Gd'd(d(eZ/Gd)d*d*eZ0Gd+d,d,eZ1Gd-d.d.eZ2Gd/d0d0eZ3Gd1d2d2Z4d3d4d5e5fd6d7Z6d8ed9fd:d;Z7dS)r?r+rrrbuild_connection_attributess z,AttributeBuilder.build_connection_attributes error_type is_internalcCs~i}|dur5|jj|t<t|dr|jdur|j|t<nd|t<t|dr1|jdur1|jj|t<nd|t<|dur=||t <|S)z Build error attributes. Args: is_internal: Whether the error is internal (e.g., timeout, network error) error_type: The exception that occurred Returns: Dictionary of error attributes N status_codeerrorrFother) __class__r ERROR_TYPEhasattrrHDB_RESPONSE_STATUS_CODErFrAREDIS_CLIENT_ERROR_CATEGORYREDIS_CLIENT_ERROR_INTERNAL)rFrGr+rrrbuild_error_attributess   z'AttributeBuilder.build_error_attributes directionchannelshardedcCs6t}|j|t<|dur||t<|dur||t<|S)a. Build attributes for a PubSub message. Args: direction: Message direction ('publish' or 'receive') channel: Pub/Sub channel name sharded: True if sharded Pub/Sub channel Returns: Dictionary of PubSub message attributes N)rr,rA%REDIS_CLIENT_PUBSUB_MESSAGE_DIRECTIONREDIS_CLIENT_PUBSUB_CHANNELREDIS_CLIENT_PUBSUB_SHARDED)rRrSrTr+rrrbuild_pubsub_message_attributess z0AttributeBuilder.build_pubsub_message_attributes stream_nameconsumer_group consumer_namecCs,t}|dur ||t<|dur||t<|S)a  Build attributes for a streaming operation. Args: stream_name: Name of the stream consumer_group: Name of the consumer group consumer_name: Name of the consumer Returns: Dictionary of streaming attributes N)rr,REDIS_CLIENT_STREAM_NAMEREDIS_CLIENT_CONSUMER_GROUP)rYrZr[r+rrrbuild_streaming_attributess z+AttributeBuilder.build_streaming_attributesresultreasoncCs@t}|dur ||t<|dur|j|t<|dur|j|t<|S)a\ Build attributes for a Client Side Caching (CSC) operation. Args: pool_name: Connection pool name (used only for csc_items metric) result: CSC result ('hit' or 'miss') reason: Reason for CSC eviction ('full' or 'invalidation') Returns: Dictionary of CSC attributes N)rr,r@rAREDIS_CLIENT_CSC_RESULTREDIS_CLIENT_CSC_REASON)r<r_r`r+rrrbuild_csc_attributes0s  z%AttributeBuilder.build_csc_attributes fail_fromr rfail_tocCs.t}t||t<t||t<|j|t<|S)a  Build attributes for a geo failover. Args: fail_from: Database failed from fail_to: Database failed to reason: Reason for the failover Returns: Dictionary of geo failover attributes )rr, get_db_nameDB_CLIENT_GEOFAILOVER_FAIL_FROMDB_CLIENT_GEOFAILOVER_FAIL_TOrADB_CLIENT_GEOFAILOVER_REASON)rdrfr`r+rrrbuild_geo_failover_attributesNs    z.AttributeBuilder.build_geo_failover_attributesrcCs|d|d|S)a Build a unique connection pool name. Args: server_address: Redis server address server_port: Redis server port db_namespace: Redis database index Returns: Unique pool name in format "address:port/db" :/r)rr r!rrrbuild_pool_namegsz AttributeBuilder.build_pool_name)NNN)NNNNNNN)NNNN)NN)r)rrr__doc__ staticmethodrr)intrrr,boolr;r rE ExceptionrQrrXr^rrrcrrrkrnrrrrr[s     .  "  %        rpoolr r"cCsN|jdd}|jdd}t|dd}|r |d|d|S|d|S) a Get a short string representation of a connection pool for observability. This provides a concise pool identifier suitable for use as a metric attribute, in the format: host:port_uniqueID (matching go-redis format) Args: pool: Connection pool instance Returns: Short pool name in format "host:port_uniqueID" Example: >>> pool = ConnectionPool(host='localhost', port=6379, db=0) >>> get_pool_name(pool) 'localhost:6379_a1b2c3d4' hostunknownportiZ_pool_idrl_)Zconnection_kwargsgetgetattr)rtrurwZpool_idrrr get_pool_name{s  r|databaserecCs6|jd}|jd}|j}|d|d|S)z Get a short string representation of a database for observability. Args: database: Database instance Returns: Short database name in format "{host}:{port}/{weight}" rurwrlrm)clientZget_connection_kwargsweight)r}rurwrrrrrgs rg)8roenumrtypingrrrrrr#Zredis.asyncio.multidb.databaserZredis.connectionr Zredis.multidb.databaser r$r*r5rNr8rLr6r7r'r(r@rBrDrhrirjr%rCZ$REDIS_CLIENT_CONNECTION_CLOSE_REASONZ$REDIS_CLIENT_CONNECTION_NOTIFICATIONr9r:rUrVrWrPrOr\r]rarbr rrrrrr)r|rgrrrrsb    "