mh42dZddlZddlZddlmZddlmZddlmZddlmZddlmZddl m Z dd l m Z d d l mZd d lmZd d lmZd dlmZd dlmZd dlmZd dlmZd dlmZd dlmZd dlmZgdZ ddl!m"Z"n #e#$re$Z"YnwxYwerddl%m&Z&ee'Z(eZ)dZ*Gdde+Z,Gddej-Z.Gdde.Z/Gd d!e.Z0Gd"d#e.Z1Gd$d%e1Z2Gd&d'e/Z3Gd(d)e3Z4dS)*zSamplers manage the client-side trace sampling Any `sampled = False` trace won't be written, and can be ignored by the instrumentation. N) TYPE_CHECKING)Dict)List)Optional)Tuple)config)SAMPLING_LIMIT_DECISION)ENV_KEY)_PRIORITY_CATEGORY)DEFAULT_SAMPLING_RATE_LIMIT)MAX_UINT_64BITS) get_logger) RateLimiter)%_get_highest_precedence_rule_matching)_set_sampling_tags) SamplingRule)_config)customerdynamicdefault)JSONDecodeError)SpanlqWV[{ceZdZdS) SamplingErrorN__name__ __module__ __qualname__:/usr/local/lib/python3.11/site-packages/ddtrace/sampler.pyrr4Dr!rc2eZdZdZejdZdS) BaseSamplerr cdSNr selfspans r"samplezBaseSampler.sample;s  r!N)rrr __slots__abcabstractmethodr+r r!r"r%r%8s6I     r!r%) metaclassc2eZdZdZejdZdS)BasePrioritySamplerr cdSr'r )r) sample_ratess r"#update_rate_by_service_sample_ratesz7BasePrioritySampler.update_rate_by_service_sample_ratesDs r!N)rrrr,r-r.r4r r!r"r1r1As6I     r!r1ceZdZdZdZdS) AllSamplerzSampler sampling all the tracescdS)NTr r(s r"r+zAllSampler.sampleLstr!N)rrr__doc__r+r r!r"r6r6Is)))r!r6c:eZdZdZd deddfdZdeddfdZdZdS) RateSamplerzSampler based on a rate Keep (100 * `sample_rate`)% of the traces. It samples randomly, its main purpose is to reduce the instrumentation footprint. ? sample_ratereturnNctdtd|}||tdd|zdS)z0sample_rate is clamped between 0 and 1 inclusiver rz.initialized RateSampler, sample %s%% of tracesdN)minmaxset_sample_ratelogdebugr)r<s r"__init__zRateSampler.__init__WsO!SK0011  [))) BC+DUVVVVVr!cVt||_|jtz|_dSr')floatr<_MAX_UINT_64BITSsampling_id_thresholdrEs r"rBzRateSampler.set_sample_rate]s) --%)%58H%H"""r!cF|jtztz|jk}|Sr')_trace_id_64bits KNUTH_FACTORrIrJ)r)r*sampleds r"r+zRateSampler.sampleas#)L8 d+ + +%- - !2 3 3 /%* *%. .r!c|t}||j|}|j|p|j}||}||fSr')get_tagr rZrXrTgetrUr+)r)r*rYkeyr`rNs r"r^z,RateByServiceSampler._make_sampling_decisionsbll7##ii c**+//44M8M..&&r!cpi}|D]\}}t|||<||_dSr')itemsrQrT)r)rate_by_servicesamplersrir<s r"r4z8RateByServiceSampler.update_rate_by_service_sample_ratessH / 5 5 7 7 ; ; C-k::HSMM$,!!!r!)NNrO)rrrr8r, _default_key staticmethodrZrFrBr+r_r^r4r r!r"rSrSjs LI"L 444\4''''  \\\\ ///   -----r!rScneZdZdZdZdZeZ d fd ZdZ e Z e d Z fd Z fd ZxZS) DatadogSamplera By default, this sampler relies on dynamic sample rates provided by the trace agent to determine which traces are kept or dropped. You can also configure a static sample rate via ``default_sample_rate`` to use for sampling. When a ``default_sample_rate`` is configured, that is the only sample rate used, and the agent provided rates are ignored. You may also supply a list of ``SamplingRule`` instances to set sample rates for specific services. Example rules:: DatadogSampler(rules=[ SamplingRule(sample_rate=1.0, service="my-svc"), SamplingRule(sample_rate=0.0, service="less-important"), ]) Rules are evaluated in the order they are provided, and the first rule that matches is used. If no rule matches, then the agent sample rates are used. This sampler can be configured with a rate limit. This will ensure the max number of sampled traces per second does not exceed the supplied limit. The default is 100 traces kept per second. )limiterrulesdefault_sample_rate_rate_limit_always_onNeAFctt|||_|}|1t jddkrt tj}|ttj }||_ |.tj }|r| |}ng}||_ njg|_ |D]`}t|tr|j |2t"jr"t'd|a|(|j t|t+|||_t.d|dS)a Constructor for DatadogSampler sampler :param rules: List of :class:`SamplingRule` rules to apply to the root span of every trace, default no rules :param default_sample_rate: The default sample rate to apply if no rules matched (default: ``None`` / Use :class:`RateByServiceSampler` only) :param rate_limit: Global rate limit (traces per second) to apply to all traces regardless of the rules applied to them, (default: ``100``) N_trace_sample_raterzCRule {!r} must be a sub-class of type ddtrace.sampler.SamplingRulesr<zinitialized %r)superrqrFrtddconfig _get_sourcerHryint_trace_rate_limitru_trace_sampling_rules_parse_rules_from_strrsrcrappendr_raise TypeErrorformatrrrrCrD) r)rsrt rate_limitrate_limit_windowrate_limit_always_oneffective_sample_rateenv_sampling_rulesrule __class__s r"rFzDatadogSampler.__init__s& nd##,,...#6 3  &#$899YFF(-h.I(J(J%  X788J%9" =!)!? ! 223EFFDJJDJ x xdL11xJ%%d++++]x#$i$p$pqu$v$vwwwx ! , J  l7LMMM N N N#:/@AA  "D)))))r!cd|jD}d|jj||j|jS)Nc$i|] \}}||jSr rz).0rir`s r" z*DatadogSampler.__str__..s!```lc7g)```r!z.{}(agent_rates={!r}, limiter={!r}, rules={!r}))rTrkrrrrrrs)r)ratess r"__str__zDatadogSampler.__str__sN``d>W>]>]>_>_```?FF N #UDL$*   r!c g} tj|}n>#t$r1tjr"t d|YnwxYw|D]x}d|vrAtjr4tdtj|Ht|d}| dtj }| dtj }| dtj }| dtj }| dd } t|||||| } nY#t $rL} tjr5t d tj|| Yd} ~ [d} ~ wwxYw| | zt|d }|S)Nz*Unable to parse DD_TRACE_SAMPLING_RULES={}r<z-No sample_rate provided for sampling rule: {}rXnameresourcetags provenancer)r<rXrrrrz#Error creating sampling rule {}: {}c@t|jSr')PROVENANCE_ORDERindexr)rs r"z6DatadogSampler._parse_rules_from_str..?sAQAWAWX\XgAhAhr!)ri)jsonloadsrrr ValueErrorrKeyErrordumpsrHrhrNO_RULErsorted) rssampling_rules json_rulesrr<rXrrrr sampling_rulees r"rz$DatadogSampler._parse_rules_from_strs  ]E**JJ ] ] ]} ] !M!T!TUZ![![\\\ ] ] ] 1 1DD((=m"#R#Y#YZ^ZdeiZjZj#k#klll] 344Khhy,*>??G88FL$899Dxx L,@AAH88FL$899D, ::J  , +#%) !!!    =h$%J%Q%QRVR\]aRbRbde%f%fggg   ! !- 0 0 0 0 4h4hiiis(8AA E## F9-AF44F9c ,|j|t||j}|j}|j}|r||}|j}nGtt| |\}}t|tr|j}|s|j r@|r>|j }|t |j jt%|||||||Sr')context _update_tagsrrsrUr<r+r{rqr^rcr:rurr is_allowed set_metricr effective_rater#_choose_priority_category_with_rule)r)r* matched_ruler`r<rNrs r"r+zDatadogSampler.sampleBs !!$'''rsZ  555555222222;;;;;;CCCCCC''''''......DDDDDD111111''''''))))))655!,,,,,,,!!! OOO!)((((((j  "      I        CK         +   +,        M-M-M-M-M-.M-M-M-`vNvNvNvNvN)vNvNvNvNvNs6A==BB