Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add datadog span processor. #6092

Open
wants to merge 1 commit into
base: bnjjj/more_telemetry_tests
Choose a base branch
from

Conversation

BrynCooke
Copy link
Contributor

@BrynCooke BrynCooke commented Oct 1, 2024

This PR makes the datadog agent sampling feature use Record rather than RecordAndSample for spans that would otherwise have been dropped.
This allows us to stay closer to the original telemetry logic for setting the sampled trace flag.

This complexity arrises because BatchSPanProcessors usually drop spans that are not sampled.
To get around this we introduce a wrapper that is active for otlp and datadog exporters if agent sampling is enabled that updates the spandata to say that it was sampled.


Checklist

Complete the checklist (and note appropriate exceptions) before the PR is marked ready-for-review.

  • Changes are compatible1
  • Documentation2 completed
  • Performance impact assessed and acceptable
  • Tests added and passing3
    • Unit Tests
    • Integration Tests
    • Manual Tests

Exceptions

Note any exceptions here

Notes

Footnotes

  1. It may be appropriate to bring upcoming changes to the attention of other (impacted) groups. Please endeavour to do this before seeking PR approval. The mechanism for doing this will vary considerably, so use your judgement as to how and when to do this.

  2. Configuration is an important part of many changes. Where applicable please try to document configuration examples.

  3. Tick whichever testing boxes are applicable. If you are adding Manual Tests, please document the manual testing (extensively) in the Exceptions.

@router-perf
Copy link

router-perf bot commented Oct 1, 2024

CI performance tests

  • connectors-const - Connectors stress test that runs with a constant number of users
  • const - Basic stress test that runs with a constant number of users
  • demand-control-instrumented - A copy of the step test, but with demand control monitoring and metrics enabled
  • demand-control-uninstrumented - A copy of the step test, but with demand control monitoring enabled
  • enhanced-signature - Enhanced signature enabled
  • events - Stress test for events with a lot of users and deduplication ENABLED
  • events_big_cap_high_rate - Stress test for events with a lot of users, deduplication enabled and high rate event with a big queue capacity
  • events_big_cap_high_rate_callback - Stress test for events with a lot of users, deduplication enabled and high rate event with a big queue capacity using callback mode
  • events_callback - Stress test for events with a lot of users and deduplication ENABLED in callback mode
  • events_without_dedup - Stress test for events with a lot of users and deduplication DISABLED
  • events_without_dedup_callback - Stress test for events with a lot of users and deduplication DISABLED using callback mode
  • extended-reference-mode - Extended reference mode enabled
  • large-request - Stress test with a 1 MB request payload
  • no-tracing - Basic stress test, no tracing
  • reload - Reload test over a long period of time at a constant rate of users
  • step-jemalloc-tuning - Clone of the basic stress test for jemalloc tuning
  • step-local-metrics - Field stats that are generated from the router rather than FTV1
  • step-with-prometheus - A copy of the step test with the Prometheus metrics exporter enabled
  • step - Basic stress test that steps up the number of users over time
  • xlarge-request - Stress test with 10 MB request payload
  • xxlarge-request - Stress test with 100 MB request payload

Move the datadog agent processor out to separate files.
@@ -239,7 +239,7 @@ async fn test_otlp_request_with_zipkin_trace_context_propagator_with_datadog(
.await?;
// ----------------------
// Testing for a trace containing a tracestate m and psr with psr set to 1 for DD and an unsampled trace, so it should be sent to the otlp exporter with sampling priority set to 1
// But it should send the trace to subgraph as the tracestate contains a psr set to 1
// It should not send the trace to the subgraph as we didn't use the datadog propagator and therefore the trace will remain unsampled.
Copy link
Contributor Author

@BrynCooke BrynCooke Oct 1, 2024

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@bnjjj This is a key test change. As we're not using datadog propagation I don't think the subgraph should pick up the trace.

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Are we sure we should not take care of this tracestate ? Do we have any specs or docs to help us to choose ?

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@BrynCooke BrynCooke requested a review from bnjjj October 1, 2024 14:05
@BrynCooke BrynCooke marked this pull request as ready for review October 1, 2024 14:06
@BrynCooke BrynCooke requested review from a team as code owners October 1, 2024 14:06
// The propagator gets first dibs on setting the trace state, so if it sets it, we don't override it unless we are not parent based.
match result.decision {
SamplingDecision::Drop | SamplingDecision::RecordOnly => {
result.decision = SamplingDecision::RecordOnly;
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@bnjjj This is the only change to the logic

}

fn on_end(&self, mut span: SpanData) {
let new_trace_flags = span.span_context.trace_flags().with_sampled(true);
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Should we check for either sampling priority to be set or measured to be set to set it to true ? I know for now we don't have that case but should we already try to handle this ?

@@ -239,7 +239,7 @@ async fn test_otlp_request_with_zipkin_trace_context_propagator_with_datadog(
.await?;
// ----------------------
// Testing for a trace containing a tracestate m and psr with psr set to 1 for DD and an unsampled trace, so it should be sent to the otlp exporter with sampling priority set to 1
// But it should send the trace to subgraph as the tracestate contains a psr set to 1
// It should not send the trace to the subgraph as we didn't use the datadog propagator and therefore the trace will remain unsampled.
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Are we sure we should not take care of this tracestate ? Do we have any specs or docs to help us to choose ?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants