tail_sampling

type AndCfg struct {
	SubPolicyCfg []AndSubPolicyCfg `river:"and_sub_policy,block"`
}

type AndSubPolicyCfg struct {
	SharedPolicyCfg SharedPolicyCfg `river:",squash"`
}

type Arguments struct {
	PolicyCfgs              []PolicyCfg   `river:"policy,block"`
	DecisionWait            time.Duration `river:"decision_wait,attr,optional"`
	NumTraces               uint64        `river:"num_traces,attr,optional"`
	ExpectedNewTracesPerSec uint64        `river:"expected_new_traces_per_sec,attr,optional"`
	// Output configures where to send processed data. Required.
	Output *otelcol.ConsumerArguments `river:"output,block"`
}

type CompositeCfg struct {
	MaxTotalSpansPerSecond int64                   `river:"max_total_spans_per_second,attr"`
	PolicyOrder            []string                `river:"policy_order,attr"`
	SubPolicyCfg           []CompositeSubPolicyCfg `river:"composite_sub_policy,block,optional"`
	RateAllocation         []RateAllocationCfg     `river:"rate_allocation,block,optional"`
}

type CompositeSubPolicyCfg struct {
	SharedPolicyCfg SharedPolicyCfg `river:",squash"`

	// Configs for and policy evaluator.
	AndCfg AndCfg `river:"and,block,optional"`
}

type LatencyCfg struct {
	// ThresholdMs in milliseconds.
	ThresholdMs int64 `river:"threshold_ms,attr"`
}

type NumericAttributeCfg struct {
	// Tag that the filter is going to be matching against.
	Key string `river:"key,attr"`
	// MinValue is the minimum value of the attribute to be considered a match.
	MinValue int64 `river:"min_value,attr"`
	// MaxValue is the maximum value of the attribute to be considered a match.
	MaxValue int64 `river:"max_value,attr"`
}

type PolicyCfg struct {
	SharedPolicyCfg SharedPolicyCfg `river:",squash"`

	// Configs for defining composite policy
	CompositeCfg CompositeCfg `river:"composite,block,optional"`

	// Configs for defining and policy
	AndCfg AndCfg `river:"and,block,optional"`
}

type ProbabilisticCfg struct {
	// HashSalt allows one to configure the hashing salts. This is important in scenarios where multiple layers of collectors
	// have different sampling rates: if they use the same salt all passing one layer may pass the other even if they have
	// different sampling rates, configuring different salts avoids that.
	HashSalt string `river:"hash_salt,attr,optional"`
	// SamplingPercentage is the percentage rate at which traces are going to be sampled. Defaults to zero, i.e.: no sample.
	// Values greater or equal 100 are treated as "sample all traces".
	SamplingPercentage float64 `river:"sampling_percentage,attr"`
}

type RateAllocationCfg struct {
	Policy  string `river:"policy,attr"`
	Percent int64  `river:"percent,attr"`
}

type RateLimitingCfg struct {
	// SpansPerSecond sets the limit on the maximum nuber of spans that can be processed each second.
	SpansPerSecond int64 `river:"spans_per_second,attr"`
}

type SharedPolicyCfg struct {
	Name                string              `river:"name,attr"`
	Type                string              `river:"type,attr"`
	LatencyCfg          LatencyCfg          `river:"latency,block,optional"`
	NumericAttributeCfg NumericAttributeCfg `river:"numeric_attribute,block,optional"`
	ProbabilisticCfg    ProbabilisticCfg    `river:"probabilistic,block,optional"`
	StatusCodeCfg       StatusCodeCfg       `river:"status_code,block,optional"`
	StringAttributeCfg  StringAttributeCfg  `river:"string_attribute,block,optional"`
	RateLimitingCfg     RateLimitingCfg     `river:"rate_limiting,block,optional"`
	SpanCountCfg        SpanCountCfg        `river:"span_count,block,optional"`
	TraceStateCfg       TraceStateCfg       `river:"trace_state,block,optional"`
}

type SpanCountCfg struct {
	// Minimum number of spans in a Trace
	MinSpans int32 `river:"min_spans,attr"`
}

type StatusCodeCfg struct {
	StatusCodes []string `river:"status_codes,attr"`
}

type StringAttributeCfg struct {
	// Tag that the filter is going to be matching against.
	Key string `river:"key,attr"`
	// Values indicate the set of values or regular expressions to use when matching against attribute values.
	// StringAttribute Policy will apply exact value match on Values unless EnabledRegexMatching is true.
	Values []string `river:"values,attr"`
	// EnabledRegexMatching determines whether match attribute values by regexp string.
	EnabledRegexMatching bool `river:"enabled_regex_matching,attr,optional"`
	// CacheMaxSize is the maximum number of attribute entries of LRU Cache that stores the matched result
	// from the regular expressions defined in Values.
	// CacheMaxSize will not be used if EnabledRegexMatching is set to false.
	CacheMaxSize int `river:"cache_max_size,attr,optional"`
	// InvertMatch indicates that values or regular expressions must not match against attribute values.
	// If InvertMatch is true and Values is equal to 'acme', all other values will be sampled except 'acme'.
	// Also, if the specified Key does not match on any resource or span attributes, data will be sampled.
	InvertMatch bool `river:"invert_match,attr,optional"`
}

type TraceStateCfg struct {
	// Tag that the filter is going to be matching against.
	Key string `river:"key,attr"`
	// Values indicate the set of values to use when matching against trace_state values.
	Values []string `river:"values,attr"`
}