o Á¿iõã@sˆddlmZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l mZe eƒZGd d „d eƒZd S) é)ÚAny)ÚOptional)ÚSpan)ÚMAX_UINT_64BITS)ÚSAMPLING_HASH_MODULO)ÚSAMPLING_KNUTH_FACTOR)Ú GlobMatcher)Ú get_logger)Ú cachedmethodc@sôeZdZdZ     ddedeedeedeedeeeefd ed dfd d „Z e d efd d„ƒZ e j ded dfdd„ƒZ e ƒdeeeeeefd efdd„ƒZded efdd„Zded efdd„Zdd„Zdd„Zded efdd„ZdS) Ú SamplingRulezl Definition of a sampling rule used by :class:`DatadogSampler` for applying a sample rate on a span NÚdefaultÚ sample_rateÚserviceÚnameÚresourceÚtagsÚ provenanceÚreturncCsztdtdt|ƒƒƒ|_|rdd„| ¡Dƒni|_|dur!t|ƒnd|_|dur,t|ƒnd|_|r5t|ƒnd|_ ||_ dS)an Configure a new :class:`SamplingRule` .. code:: python DatadogSampler([ # Sample 100% of any trace SamplingRule(sample_rate=1.0), # Sample no healthcheck traces SamplingRule(sample_rate=0, name='flask.request'), # Sample all services ending in `-db` based on a regular expression SamplingRule(sample_rate=0.5, service=re.compile('-db$')), # Sample based on service name using custom function SamplingRule(sample_rate=0.75, service=lambda service: 'my-app' in service), ]) :param sample_rate: The sample rate to apply to any matching spans :type sample_rate: :obj:`float` clamped between 0.0 and 1.0 inclusive :param service: Rule to match the `span.service` on, default no rule defined :type service: :obj:`object` to directly compare, :obj:`function` to evaluate, or :class:`re.Pattern` to match :param name: Rule to match the `span.name` on, default no rule defined :type name: :obj:`object` to directly compare, :obj:`function` to evaluate, or :class:`re.Pattern` to match :param tags: A dictionary whose keys exactly match the names of tags expected to appear on spans, and whose values are glob-matches with the expected span tag values. Glob matching supports "*" meaning any number of characters, and "?" meaning any one character. If all tags specified in a SamplingRule are matches with a given span, that span is considered to have matching tags with the rule. gð?gcSsi|] \}}|tt|ƒƒ“qS©)rÚstr)Ú.0ÚkÚvrrúP/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/_trace/sampling_rule.pyÚ ?sz)SamplingRule.__init__..N) ÚminÚmaxÚfloatr Úitemsrrrrrr)Úselfr rrrrrrrrÚ__init__s ' zSamplingRule.__init__cCs|jS©N)Ú _sample_rate©rrrrr EszSamplingRule.sample_ratecCs||_|t|_dSr!)r"rÚ_sampling_id_threshold)rr rrrr IsÚkeycCsP|\}}}||jf||jf||jffD]\}}|dur%| t|ƒ¡s%dSqdS)NFT)rrrÚmatchr)rr%rrrÚpropÚpatternrrrÚ name_matchNs $€zSamplingRule.name_matchÚspancCs | |¡o| |j|j|jf¡S)zì Return if this span matches this rule :param span: The span to match against :type span: :class:`ddtrace._trace.span.Span` :returns: Whether this span matches or not :rtype: :obj:`bool` )Ú tags_matchr)rrr©rr*rrrÚmatchesWs zSamplingRule.matchescCsª|jsdS|jp i}|jpi}|s|sdS|j ¡D]8\}}| || |¡¡}|dur.dSt|tƒrH| ¡rs