&`i4,ddlZddlZddlmZddlmZmZddlm Z ddl m Z m Z m Z mZmZejeZGddeZGdd eZGd d ejZd Zd ZdZejdejejzZdZdS)N)sync)aggregation_data base_exporter)start_http_server)REGISTRYCounterMetricFamilyGaugeMetricFamilyHistogramMetricFamilyUnknownMetricFamilyczeZdZdZdddefdZedZedZedZ edZ d S) OptionsaOptions contains options for configuring the exporter. The address can be empty as the prometheus client will assume it's localhost :type namespace: str :param namespace: The prometheus namespace to be used. Defaults to ''. :type port: int :param port: The Prometheus port to be used. Defaults to 8000. :type address: str :param address: The Prometheus address to be used. Defaults to ''. :type registry: registry :param registry: The Prometheus address to be used. Defaults to ''. :type registry: :class:`~prometheus_client.core.CollectorRegistry` :param registry: A Prometheus collector registry instance. i@cX||_||_t||_||_dSN) _namespace _registryint_port_address)self namespaceportaddressregistrys t/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/_private/prometheus_exporter.py__init__zOptions.__init__'s(#!YY  c|jSz&Prometheus Collector Registry instancerrs rrzOptions.registry- ~rc|jS)z Prefix to be used with view name)rr!s rrzOptions.namespace2s rc|jS)zPort number to listen)rr!s rrz Options.port7s zrc|jS)z'Endpoint address (default is localhost))rr!s rrzOptions.address< }rN) __name__ __module__ __qualname____doc__rrpropertyrrrrrrr r s  "$$X    XXXXrr ceZdZdZedfdZedZedZedZ edZ dZ d Z d Z d ZdS) Collectorz4Collector represents the Prometheus Collector objectNcP|i}||_|j|_||_i|_dSr)_optionsrr_view_name_to_data_map_registered_views)roptionsview_name_to_data_maps rrzCollector.__init__Es5 ($& !  )&;#!#rc|jSz,Options to be used to configure the exporterr0r!s rr3zCollector.optionsMr&rc|jSrr r!s rrzCollector.registryRr"rc|jS)zOMap with all view data objects that will be sent to Prometheus )r1r!s rr4zCollector.view_name_to_data_mapWs **rc|jS)zMap with all registered views)r2r!s rregistered_viewszCollector.registered_views^s %%rct|jj|}||jvrG||jt t t|j|j j d}||j|<dSdS)zrregister_view will create the needed structure in order to be able to sent all data to Prometheus )name documentationlabelsunitsN) get_view_namer3rr; descriptionlistmapsanitizecolumnsmeasureunit)rviewv_namedescs r register_viewzCollector.register_viewcswt|5t<< . . .!%!1s8T\::;;* D -1D !& ) ) ) / .rc||jt|jj|j}||j|<dS)z)Add view data object to be sent to serverN)rLrIrAr3rr4)r view_datarJs r add_view_datazCollector.add_view_datarsA 9>***t|5y~FF-6"6***rcB|d}|d}|d}|d}t|t|ks J||fd|D}t|tjrM||} | st ||||} | ||<| ||jdSt|tjr|j t|j ksJg} d } t|j D];\} } | |j | z } t| | g}| |<| d |jg||} | st||| } | ||<| || |j dSt|tjrL||} | st%||| } | ||<| ||jdSt|tjrL||} | st+||| } | ||<| ||jdSt/d t1|)ato_metric translate the data that OpenCensus create to Prometheus format, using Prometheus Metric object :type desc: dict :param desc: The map that describes view definition :type tag_values: tuple of :class: `~opencensus.tags.tag_value.TagValue` :param object of opencensus.tags.tag_value.TagValue: TagValue object used as label values :type agg_data: object of :class: `~opencensus.stats.aggregation_data.AggregationData` :param object of opencensus.stats.aggregation_data.AggregationData: Aggregated data that needs to be converted as Prometheus samples :rtype: :class:`~prometheus_client.core.CounterMetricFamily` or :class:`~prometheus_client.core.HistogramMetricFamily` or :class:`~prometheus_client.core.UnknownMetricFamily` or :class:`~prometheus_client.core.GaugeMetricFamily` r=r>r?r@cg|]}|r|nd S)rr,).0tvs r z'Collector.to_metric..s!<<<2B&bbB<<rHr?)r?valueNrz+Inf)r=r>r?)r?buckets sum_valuezunsupported aggregation type )len isinstanceaggregation_data_moduleCountAggregationDatagetr add_metric count_dataDistributionAggregationDataboundssorted enumeratecounts_per_bucketstrappendr sumSumAggregationDatar sum_dataLastValueAggregationDatar rU ValueErrortype)rrK tag_valuesagg_data metrics_map metric_namemetric_description label_keys metric_unitsmetricrV cum_countiiboundbuckets r to_metriczCollector.to_metricys$6l !/2(^ G} :#j//111J 3K111=<<<< h 7 L M ME O __[11F 2,$"4%%  ,2 K(   Zx7J  K K K F "9"U V V8 O?fX_&=&=====GI&x77 ' ' EX7;; e**i0v&&&& NNFH$78 9 9 9 __[11F 2.$"4% ,2 K(   !",     F "9"L M M O __[11F 2,$"4% ,2 K(   Zx7H  I I I F "9"R S S O __[11F 2*$"4% ,2 K(   Zx~  F F F FMT(^^MMNN Nrc#2Ki}|jD]K\}}||jvr|j|}|jD]'}|j|}|||||(L|D]}|VdS)zCollect fetches the statistics from OpenCensus and delivers them as Prometheus Metrics. Collect is invoked every time a prometheus.Gatherer is run for example when the HTTP endpoint is invoked by Prometheus. N)r1copyitemsr;tag_value_aggregation_data_maprxvalues)rrnrJrNrKrlrmrss rcollectzCollector.collects !%!>&wyy$$$$XX++X+ &&X& 1 1 1777aOaOaOFrr.ceZdZdZejefdZedZ edZ edZ edZ dZ dZd Zd Zd S) PrometheusStatsExportera^Exporter exports stats to Prometheus, users need to register the exporter as an HTTP Handler to be able to export. :type options: :class:`~opencensus.ext.prometheus.stats_exporter.Options` :param options: An options object with the parameters to instantiate the prometheus exporter. :type gatherer: :class:`~prometheus_client.core.CollectorRegistry` :param gatherer: A Prometheus collector registry instance. :type transport: :class:`opencensus.common.transports.sync.SyncTransport` or :class:`opencensus.common.transports.async_.AsyncTransport` :param transport: An instance of a Transpor to send data with. :type collector: :class:`~opencensus.ext.prometheus.stats_exporter.Collector` :param collector: An instance of the Prometheus Collector object. c||_||_||_|||_|t j|jdSr)r0 _gatherer _collector _transport serve_httprregister)rr3gatherer transport collectors rrz PrometheusStatsExporter.__init__sQ  !##)D// $/*****rc|jS)zOThe transport way to be sent data to server (default is sync). )rr!s rrz!PrometheusStatsExporter.transport rc|jS)zSCollector class instance to be used to communicate with Prometheus )rr!s rrz!PrometheusStatsExporter.collectorrrc|jSr)rr!s rrz PrometheusStatsExporter.gathererr"rc|jSr6r7r!s rr3zPrometheusStatsExporter.options!r&rcB||j|dSdS)zvexport send the data to the transport class in order to be sent to Prometheus in a sync or async way. N)rexport)rrNs rrzPrometheusStatsExporter.export&s.  N ! !) , , , , , ! rc tdS)NzNot supported by Prometheus)NotImplementedError)rrIs ron_register_viewz(PrometheusStatsExporter.on_register_view-s"#@AAArc`|D]*}|ji|_|j|+dS)a=Emit exports to the Prometheus if view data has one or more rows. Each OpenCensus AggregationData will be converted to corresponding Prometheus Metric: SumData will be converted to Untyped Metric, CountData will be a Counter Metric DistributionData will be a Histogram Metric. N)r|rrO)rrNv_datas remitzPrometheusStatsExporter.emit0sG  1 1F4<8:5 N ( ( 0 0 0 0  1 1rcxt|jj}|rd|ini}tdd|jji|dS)z*serve_http serves the Prometheus endpoint.addrrNr,)rdr3rrr)rrkwargss rrz"PrometheusStatsExporter.serve_http>sLdl*++&-5&'""2;;t|0;F;;;;;rN)r'r(r)r*r SyncTransportr.rr+rrrr3rrrrr,rrrrs&,0+=++++X X XX---BBB 1 1 1<<<<>>f%%I&IH Orc"t|S)znew_collector should be used to create instance of Collector class in order to prevent the usage of constructor directly r3)r.rs rrrTs W % % %%rcJd}|dkr|dz}t||jzS)zcreate the name for the viewr_)rEr=)rrIr=s rrArA\s. DB3 D49$ % %%rz[^\w]c8td|S)zsanitize the given metric name or label according to Prometheus rule. Replace all characters other than [A-Za-z0-9_] with '_'. r)_NON_LETTERS_NOR_DIGITS_REsub)keys rrErEgs & ) )#s 3 33r) loggingreopencensus.common.transportsropencensus.statsrrZrprometheus_clientrprometheus_client.corerrr r r getLoggerr'loggerobjectr r. StatsExporterrrrrAcompileUNICODE IGNORECASErrEr,rrrs  ------WWWWWWWW//////  8 $ $(((((f(((Vllllllll^Q<Q<Q<Q<Q