B .l\@s>ddlZddlmZddlmZGdddeZeddZdS) N)Lock)Metricc@sLeZdZdZdddZddZddZd d Zd d Zd dZ dddZ dS)CollectorRegistryzMetric collector registry. Collectors must have a no-argument method 'collect' that returns a list of Metric objects. The returned metrics should be consistent with the Prometheus exposition formats. FcCsi|_i|_||_t|_dS)N)_collector_to_names_names_to_collectors_auto_describer_lock)self auto_describer 9lib/python3.7/site-packages/prometheus_client/registry.py__init__szCollectorRegistry.__init__c Csd|jT||}t|j|}|r4td|x|D]}||j|<q:W||j|<WdQRXdS)z Add a collector to the registry.z/Duplicated timeseries in CollectorRegistry: {0}N)r _get_namessetr intersection ValueErrorformatr)r collectornames duplicatesnamer r r registers  zCollectorRegistry.registerc Cs:|j*x|j|D] }|j|=qW|j|=WdQRXdS)z%Remove a collector from the registry.N)r rr)r rrr r r unregister"s zCollectorRegistry.unregistercCsd}y |j}Wntk r"YnX|s4|jr4|j}|s}fdd|jD}|r`t|j|j|j }||_ |q`WqRWGfdddt }|S)aReturns object that only collects some metrics. Returns an object which upon collect() will return only samples with the given names. Intended usage is: generate_latest(REGISTRY.restricted_registry(['a_timeseries'])) Experimental.Ncsg|]}|dkr|qS)rr ).0s)rr r asz9CollectorRegistry.restricted_registry..cseZdZfddZdS)zACollectorRegistry.restricted_registry..RestrictedRegistrycsS)Nr )r )metricsr r rhszICollectorRegistry.restricted_registry..RestrictedRegistry.collectN)__name__ __module__ __qualname__rr )r*r r RestrictedRegistrygsr.) rr raddrsamplesrrZ documentationr r!object) r rr&rrr#r0mr.r )r*rr restricted_registryNs    z%CollectorRegistry.restricted_registryNcCsL|dkr i}x:|D].}x(|jD]}|j|kr"|j|kr"|jSq"WqWdS)z~Returns the sample value, or None if not found. This is inefficient, and intended only for use in unittests. N)rr0rlabelsvalue)r rr4r#r(r r r get_sample_valuems z"CollectorRegistry.get_sample_value)F)N) r+r,r-__doc__rrrrrr3r6r r r r rs   rT)r )r%Z threadingrZ metrics_corerr1rZREGISTRYr r r r s  t