/ iXJdZddlZddlZddlmZddlmZmZmZm Z erddl m Z ddl m Z ddlmZddlmZdd lmZmZmZmZmZmZmZmZdd lmZmZdd lmZej e!Z" dd l#m$Z$d Z%n#e&$r dZ%dZ'dZ(dZ$dZ)YnwxYwGddeZ*GddZ+dS)z OpenTelemetry metrics collector for redis-py. This module defines and manages all metric instruments according to OTel semantic conventions for database clients. N)Enum) TYPE_CHECKINGCallableOptionalUnion)ConnectionPool) AsyncDatabase)ConnectionPoolInterface) SyncDatabase)$REDIS_CLIENT_CONNECTION_CLOSE_REASON$REDIS_CLIENT_CONNECTION_NOTIFICATIONAttributeBuilder CSCReason CSCResultGeoFailoverReasonPubSubDirection get_pool_name) MetricGroup OTelConfig)deprecated_args)MeterTFceZdZdZdZdZdZdS) CloseReasona Enum representing the reason why a Redis client connection was closed. Values: APPLICATION_CLOSE: The connection was closed intentionally by the application (for example, during normal shutdown or explicit cleanup). ERROR: The connection was closed due to an unexpected error (for example, network failure or protocol error). HEALTHCHECK_FAILED: The connection was closed because a health check or liveness check for the connection failed. application_closeerrorhealthcheck_failedN)__name__ __module__ __qualname____doc__APPLICATION_CLOSEERRORHEALTHCHECK_FAILEDC:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\redis/observability/metrics.pyrr/s-  , E-r%rceZdZdZdZdZdedefdZdId Z dId Z dId Z dId Z dId Z dIdZdIdZ dJdeedeedeedeedeedeedeefdZdededededef dZdeddeddefdZdeddfd Zdeddfd!Zd"eddfd#Zd$ed%d&eddfd'Zd"ed&eddfd(Z e!d)gd*d+, dKd-ed&edeedeed.eed)eedeedeedeedeed/eeddfd0Z" dLd1ee#deeddfd2Z$d3eded4eddfd5Z%d"eddfd6Z& dLd7e'd8eed9eeddfd:Z(e!d;gdeed?eed;eeddf d@Z) dNdAee*ddfdBZ+ dNdCedee,ddfdDZ-dEeddfdFZ.e/defdGZ0defdHZ1dS)ORedisMetricsCollectorap Collects and records OpenTelemetry metrics for Redis operations. This class manages all metric instruments and provides methods to record various Redis operations including connection pool events, command execution, and cluster-specific operations. Args: meter: OpenTelemetry Meter instance config: OTel configuration object zredis-pyz1.0.0meterconfigctstd||_||_t |_d|_tj|jj vr| tj |jj vr| tj |jj vr|tj|jj vr|tj|jj vr|tj|jj vr|tj|jj vr|t.ddS)NzROpenTelemetry API is not installed. Install it with: pip install opentelemetry-apiz!RedisMetricsCollector initialized)OTEL_AVAILABLE ImportErrorr)r*r attr_builderconnection_countr RESILIENCY metric_groups_init_resiliency_metricsCOMMAND_init_command_metricsCONNECTION_BASIC_init_connection_basic_metricsCONNECTION_ADVANCED!_init_connection_advanced_metricsPUBSUB_init_pubsub_metrics STREAMING_init_streaming_metricsCSC_init_csc_metricsloggerinfo)selfr)r*s r&__init__zRedisMetricsCollector.__init__Qse A    ,.. $  !T[%> > >  ) ) + + +  $+"; ; ;  & & ( ( (  '4;+D D D  / / 1 1 1  *dk.G G G  2 2 4 4 4  !: : :  % % ' ' '  DK$= = =  ( ( * * * ?dk7 7 7  " " $ $ $ 788888r%returnNc|jddd|_|jddd|_|jdd d |_d S) zInitialize resiliency metrics.zredis.client.errorsz{error}z`A counter of all errors (both returned to the user and handled internally in the client library)nameunit descriptionz&redis.client.maintenance.notificationsz{notification}z,Tracks server-side maintenance notificationsz"redis.client.geofailover.failoversz {geofailover}z6Total count of failovers happened using MultiDbClient.N)r)create_counter client_errorsmaintenance_notifications geo_failoversrAs r&r2z.RedisMetricsCollector._init_resiliency_metricsvs!Z66&z7   *.)B)B9!F*C* * & "Z665 P7  r%c|jddd|jj|_|jddd|_|jd d d |_d S) z$Initialize basic connection metrics.z db.client.connection.create_timeszTime to create a new connectionrFrGrH#explicit_bucket_boundaries_advisoryz'redis.client.connection.relaxed_timeoutz {relaxation}z@Counts up for relaxed timeout, counts down for unrelaxed timeoutrEzredis.client.connection.handoffz {handoff}zIConnections that have been handed off (e.g., after a MOVING notification)N) r)create_histogramr*buckets_connection_create_timeconnection_create_timecreate_up_down_counterconnection_relaxed_timeoutrIconnection_handoffrMs r&r6z4RedisMetricsCollector._init_connection_basic_metricss&*j&A&A3904 0Z 'B' ' #+/**K*K:Z+L+ + ' #'*";";2c#<# # r%c|jddd|_|jddd|jj|_|jd d d |_d S) z'Initialize advanced connection metrics.zdb.client.connection.timeoutsz {timeout}zaThe number of connection timeouts that have occurred trying to obtain a connection from the pool.rEzdb.client.connection.wait_timerOz/Time to obtain an open connection from the poolrPzredis.client.connection.closed {connection}z"Total number of closed connectionsN)r)rIconnection_timeoutsrRr*buckets_connection_wait_timeconnection_wait_timeconnection_closedrMs r&r8z7RedisMetricsCollector._init_connection_advanced_metricss#':#<#<0{$=$ $  %)J$?$?1I04 0X %@% % !"&!:!:1<";" " r%c`|jddd|jj|_dS)z0Initialize command execution metric instruments.zdb.client.operation.durationrOzCommand execution durationrPN)r)rRr*buckets_operation_durationoperation_durationrMs r&r4z+RedisMetricsCollector._init_command_metricss8"&*"="=/404 0V #># # r%cJ|jddd|_dS)z%Initialize PubSub metric instruments.zredis.client.pubsub.messagesz {message}z&Tracks published and received messagesrEN)r)rIpubsub_messagesrMs r&r:z*RedisMetricsCollector._init_pubsub_metricss0#z88/@ 9  r%c`|jddd|jj|_dS)z(Initialize Streaming metric instruments.zredis.client.stream.lagrOzkEnd-to-end lag per message, showing how stale are the messages when the application starts processing them.rPN)r)rRr*"buckets_stream_processing_duration stream_lagrMs r&r<z-RedisMetricsCollector._init_streaming_metricss8*55*F04 0^ 6  r%c|jddd|_|jddd|_|jdd d |_d S) z8Initialize Client Side Caching (CSC) metric instruments.zredis.client.csc.requestsz {request}z)The total number of requests to the cacherEzredis.client.csc.evictionsz {eviction}z#The total number of cache evictionszredis.client.csc.network_savedByz,The total number of bytes saved by using CSCN)r)rI csc_requests csc_evictionscsc_network_savedrMs r&r>z'RedisMetricsCollector._init_csc_metricss J55,C6   "Z66-=7   "&!:!:1F";" " r%server_address server_portnetwork_peer_addressnetwork_peer_port error_typeretry_attempts is_internalcXt|dsdS|j||}||j|||||j|||jd|dS)a Record error count Args: server_address: Server address server_port: Server port network_peer_address: Network peer address network_peer_port: Network peer port error_type: Error type retry_attempts: Retry attempts is_internal: Whether the error is internal (e.g., timeout, network error) rJNrkrl)rmrnrp)rorq attributes)hasattrr.build_base_attributesupdatebuild_operation_attributesbuild_error_attributesrJadd) rArkrlrmrnrorprqattrss r&record_error_countz(RedisMetricsCollector.record_error_counts,t_--  F!77)#8      8 8%9"3- 9         4 4%' 5      qU33333r%maint_notificationc t|dsdS|j||}||j||||t <|jd|dS)a7 Record maintenance notification count Args: server_address: Server address server_port: Server port network_peer_address: Network peer address network_peer_port: Network peer port maint_notification: Maintenance notification rKNrs)rmrnrtru)rwr.rxryrzr rKr|)rArkrlrmrnrr}s r&record_maint_notification_countz5RedisMetricsCollector.record_maint_notification_counts$t899  F!77)#8      8 8%9"3 9     7I23 &**1*?????r% fail_from)r r fail_toreasonct|dsdS|j|||}|jd|S)z Record geo failover Args: fail_from: Database failed from fail_to: Database failed to reason: Reason for the failover rLN)rrrrtru)rwr.build_geo_failover_attributesrLr|)rArrrr}s r&record_geo_failoverz)RedisMetricsCollector.record_geo_failover6s`t_--  F!??@   !%%aE%:::r%callbackctj|jjvr |jsdS|jddd|g|_dS)z Initialize observable gauge for connection count metric. Args: callback: Callback function to retrieve connection count Nzdb.client.connection.countrYz!Number of connections in the poolrFrGrH callbacks)rr5r*r1r/r)create_observable_gaugerArs r&init_connection_countz+RedisMetricsCollector.init_connection_countPs^  ( 0I I I) J F $ B B-;j !C! ! r%ctj|jjvr |jsdS|jddd|g|_dS)z Initialize observable gauge for CSC items metric. Args: callback: Callback function to retrieve CSC items count Nzredis.client.csc.itemsz{item}z5The total number of cached responses currently storedr)rr=r*r1 csc_itemsr)rrs r&init_csc_itemsz$RedisMetricsCollector.init_csc_itemsgsR ?$+"; ; ;DN ; F;;)Oj <  r% pool_namect|dsdS|j|}|jd|dS)zo Record a connection timeout event. Args: pool_name: Connection pool name rZNrrtru)rwr.build_connection_attributesrZr|rArr}s r&record_connection_timeoutz/RedisMetricsCollector.record_connection_timeout{sWt233  F!== =RR  $$Q5$99999r%connection_pool)r rduration_secondsct|dsdS|jt|}|j||dS)z Record time taken to create a new connection. Args: connection_pool: Connection pool implementation duration_seconds: Creation time in seconds rTNrru)rwr.rrrTrecord)rArrr}s r&record_connection_create_timez3RedisMetricsCollector.record_connection_create_timesit566  F!==#O44>   #**+;*NNNNNr%ct|dsdS|j|}|j||dS)z Record time taken to obtain a connection from the pool. Args: pool_name: Connection pool name duration_seconds: Wait time in seconds r\Nrru)rwr.rr\r)rArrr}s r&record_connection_wait_timez1RedisMetricsCollector.record_connection_wait_timesXt344  F!== =RR !(()9e(LLLLLr% batch_sizezXThe batch_size argument is no longer used and will be removed in the next major version.z7.2.1) args_to_warnrversion command_name db_namespace is_blockingc t|dsdS|j|sdS|j|||} | |j||| | | | |j||j || dS)a Record command execution duration. Args: command_name: Redis command name (e.g., 'GET', 'SET', 'MULTI') duration_seconds: Execution time in seconds server_address: Redis server address server_port: Redis server port db_namespace: Redis database index batch_size: Number of commands in batch (for pipelines/transactions) error_type: Error type if operation failed network_peer_address: Resolved peer address network_peer_port: Peer port number retry_attempts: Number of retry attempts made is_blocking: Whether the operation is a blocking command r`N)rkrlr)rrmrnrprroru) rwr*should_track_commandr.rxryrzr{r`r) rArrrkrlrrrormrnrprr}s r&record_operation_durationz/RedisMetricsCollector.record_operation_durationsFt122  F{// ==  F!77)#%8      8 8)%9"3-' 9         4 4% 5     &&'7E&JJJJJr% close_reasonct|dsdS|j}|r|j|t<||j||jd|dS)z Record a connection closed event. Args: close_reason: Reason for closing (e.g. 'error', 'application_close') error_type: Error type if closed due to error r]Nrrtru) rwr.rvaluer ryr{r]r|)rArror}s r&record_connection_closedz.RedisMetricsCollector.record_connection_closedst011  F!==??  M:F:LE6 7   4 4% 5     ""1"77777r%connection_namerelaxedct|dsdS|j|}||t<|j|rdnd|dS)a  Record a connection timeout relaxation event. Args: connection_name: Connection name maint_notification: Maintenance notification type relaxed: True to count up (relaxed), False to count down (unrelaxed) rVN)rrtru)rwr.rr rVr|)rArrrr}s r&!record_connection_relaxed_timeoutz7RedisMetricsCollector.record_connection_relaxed_timeout sut9::  F!==+>  7I23 '++,@AAbU+SSSSSr%ct|dsdS|j|}|jd|dS)z Record a connection handoff event (e.g., after MOVING notification). Args: pool_name: Connection pool name rWNrrtru)rwr.rrWr|rs r&record_connection_handoffz/RedisMetricsCollector.record_connection_handoff$sWt122  F!== =RR ##A%#88888r% directionchannelshardedct|dsdS|j|||}|jd|dS)z Record a PubSub message (published or received). Args: direction: Message direction ('publish' or 'receive') channel: Pub/Sub channel name sharded: True if sharded Pub/Sub channel rbN)rrrrtru)rwr.build_pubsub_message_attributesrbr|)rArrrr}s r&record_pubsub_messagez+RedisMetricsCollector.record_pubsub_message6sgt.//  F!AAB     u 55555r% consumer_namez[The consumer_name argument is no longer used and will be removed in the next major version. lag_seconds stream_nameconsumer_groupct|dsdS|j||}|j||dS)z Record the lag of a streaming message. Args: lag_seconds: Lag in seconds stream_name: Stream name consumer_group: Consumer group name consumer_name: Consumer name reN)rrru)rwr.build_streaming_attributesrer)rArrrrr}s r&record_streaming_lagz*RedisMetricsCollector.record_streaming_lagPsa*t\**  F!<<#)=   {u=====r%resultct|dsdS|j|}|jd|dS)z} Record a Client Side Caching (CSC) request. Args: result: CSC result ('hit' or 'miss') rhN)rrtru)rwr.build_csc_attributesrhr|)rArr}s r&record_csc_requestz(RedisMetricsCollector.record_csc_requestpsVt^,,  F!66f6EE aE22222r%countct|dsdS|j|}|j||dS)z Record a Client Side Caching (CSC) eviction. Args: count: Number of evictions reason: Reason for eviction riN)rru)rwr.rrir|)rArrr}s r&record_csc_evictionz)RedisMetricsCollector.record_csc_evictionsVt_--  F!66f6EE u77777r% bytes_savedct|dsdS|j}|j||dS)z Record the number of bytes saved by using Client Side Caching (CSC). Args: bytes_saved: Number of bytes saved rjNru)rwr.rrjr|)rArr}s r&record_csc_network_savedz.RedisMetricsCollector.record_csc_network_savedsRt011  F!6688 "";5"AAAAAr%c(tjS)z Get monotonic time for duration measurements. Returns: Current monotonic time in seconds )time monotonicr$r%r&monotonic_timez$RedisMetricsCollector.monotonic_times~r%c(d|jd|jdS)NzRedisMetricsCollector(meter=z , config=))r)r*rMs r&__repr__zRedisMetricsCollector.__repr__sQdjQQ4;QQQQr%)rCN)NNNNNNN) NNNNNNNNN)NN)NNN)N)2rrrr METER_NAME METER_VERSIONrrrBr2r6r8r4r:r<r>rstrint Exceptionboolr~rrrrrrrrfloatrrrrrrrrrrrrrrrr staticmethodrrr$r%r&r(r(AsX  JM#9e#9Z#9#9#9#9J    (    *    *                0)-%).2+/*.(,&*,4,4  ,4c],4'sm ,4 $C= ,4 Y' ,4! ,4d^,4,4,4,4\"@"@"@" "@  "@  "@"@"@"@H;89;67;" ;;;;4       .       ( :3 :4 : : : :OJKO O  OOOO(MM M  MMMM(_"^i)-%)&*$(*..2+/(,&*;K;K;K ;K! ;K c] ;K sm ;KSM;KY';K'sm;K$C=;K! ;Kd^;K ;K;K;K  ;K~/3*.88{+8Y'8  88886TT T T  TTTT.99 9999*"&"& 66"6#6$ 6  66664_%&l&*(,'+ >>>c]>! >  } >  >>>  >:'+33#3 3333&'+888#8  8888$BB BBBB$ E   \ R#RRRRRRr%r(),r loggingrenumrtypingrrrrredis.asyncio.connectionrredis.asyncio.multidb.databaser redis.connectionr redis.multidb.databaser redis.observability.attributesr r rrrrrrredis.observability.configrr redis.utilsr getLoggerrr?opentelemetry.metricsrr,r-Counter Histogram UpDownCounterrr(r$r%r&rs ;;;;;;;;;;;;4777777<<<<<<888888333333                    ?>>>>>>>''''''  8 $ $ ++++++NNNGI EMMM .....$...$n Rn Rn Rn Rn Rn Rn Rn Rn Rn Rs*A33BB