o `j @s:ddlZddlZddlmZeeZGdddeZdS)N)AbstractMetricsReporterc@sReZdZdZdddZddZddZd d Zd d Zd dZ ddZ ddZ dS) DictReporterz}A basic dictionary based metrics reporter. Store all metrics in a two level dictionary of category > name > metric. cCs"t|_|r |nd|_i|_dS)Nr) threadingLock_lock_prefix_store)selfprefixr Y/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/kafka/metrics/dict_reporter.py__init__s  zDictReporter.__init__cCstddt|jDS)z Return a nested dictionary snapshot of all metrics and their values at this time. Example: { 'category': { 'metric1_name': 42.0, 'metric2_name': 'foo' } } css2|]\}}|tddt|DfVqdS)css |] \}}||fVqdSN)value).0namemetricr r r sz2DictReporter.snapshot...N)dictlistitems)rcategorymetricsr r r rs  z(DictReporter.snapshot..)rrr rr r r r snapshots zDictReporter.snapshotcCs|D]}||qdSr) metric_change)r rrr r r init#s zDictReporter.initcCs\|j!||}||jvri|j|<||j||jj<WddS1s'wYdSr)r get_categoryr metric_namer)r rrr r r r's    "zDictReporter.metric_changecCsh|j'||}|j|i}||jjd}|s!|j|d|WdS1s-wYdSr)rrr getpoprr)r rrrremovedr r r metric_removal.s $zDictReporter.metric_removalcCs@dddt|jjD}ddd|j|jj|fDS)a Return a string category for the metric. The category is made up of this reporter's prefix and the metric's group and tags. Examples: prefix = 'foo', group = 'bar', tags = {'a': 1, 'b': 2} returns: 'foo.bar.a=1,b=2' prefix = 'foo', group = 'bar', tags = None returns: 'foo.bar' prefix = None, group = 'bar', tags = None returns: 'bar' ,css |] \}}d||fVqdS)z%s=%sNr )rkvr r r rHsz,DictReporter.get_category...css|]}|r|VqdSrr )rxr r r rJs)joinsortedrtagsrrgroup)r rr+r r r r7s  zDictReporter.get_categorycCdSrr )r configsr r r configureMzDictReporter.configurecCr-rr rr r r closePr0zDictReporter.closeN)r) __name__ __module__ __qualname____doc__rrrrr#rr/r1r r r r r s   r)loggingrkafka.metrics.metrics_reporterr getLoggerr2loggerrr r r r s