o i@s&ddlmZddlZGdddZdS))LockNc@s.eZdZddZddZd ddZdd ZdS) MaxSampleMetricContextscCst|_i|_||_dSN)rlockvaluesmax_sample_metric_type)selfrr _/home/ubuntu/.local/lib/python3.10/site-packages/datadog/dogstatsd/max_sample_metric_context.py__init__s z MaxSampleMetricContexts.__init__cCs\g} |j|j}i|_Wdn1swY|D] \}}||q |Sr)rritemsappendflush)rmetricstemp_metricr r r r szMaxSampleMetricContexts.flushNc Cs||}|j#||jvr|j||||d|j|<|j|} | jWdn1s.wY|r;| |n| | jdS)z6Sample a metric and store it if it meets the criteria.) cardinalityN) should_samplerrracquiremaybe_keep_sample_work_unsafe skip_samplerelease) rnamevaluetagsrate context_keymax_samples_per_contextrkeeping_samplerr r r samples     zMaxSampleMetricContexts.samplecCs|dkrdSt|kS)zADetermine if a sample should be kept based on the specified rate.T)random)rrr r r r&s z%MaxSampleMetricContexts.should_sampler)__name__ __module__ __qualname__r rr rr r r r rs  r) threadingrr"rr r r r s