Skip to content

feat(trace): Trigger API Key refresh on 403#45674

Merged
gh-worker-dd-mergequeue-cf854d[bot] merged 8 commits intomainfrom
ichinaski/api-key-refresh-retry-payloads
Feb 23, 2026
Merged

feat(trace): Trigger API Key refresh on 403#45674
gh-worker-dd-mergequeue-cf854d[bot] merged 8 commits intomainfrom
ichinaski/api-key-refresh-retry-payloads

Conversation

@ichinaski
Copy link
Contributor

@ichinaski ichinaski commented Jan 29, 2026

What does this PR do?

The agent now supports API Key rotation. Instead of dropping payloads when a 403 is received from the backend, we will instead trigger an API Key (secrets) refresh, and mark the payload as retriable.

This PR also refactors the API Key management within pkg/trace/writer/sender.go, making reads and writes safe for concurrent use.

Motivation

See above

Describe how you validated your changes

Added unit tests

Additional Notes

The agent now supports API Key rotation. Instead of dropping payloads
when a 403 is received from the backend, we will instead trigger an API
Key (secrets) refresh, and mark the payload as retriable.
@ichinaski ichinaski requested a review from a team as a code owner January 29, 2026 14:37
@github-actions github-actions bot added team/agent-apm trace-agent medium review PR review might take time labels Jan 29, 2026
@ichinaski ichinaski added qa/done QA done before merge and regressions are covered by tests and removed qa/skip-qa labels Jan 29, 2026
@agent-platform-auto-pr
Copy link
Contributor

agent-platform-auto-pr bot commented Jan 29, 2026

Static quality checks

✅ Please find below the results from static quality gates
Comparison made with ancestor bc99348
📊 Static Quality Gates Dashboard
🔗 SQG Job

Successful checks

Info

Quality gate Change Size (prev → curr → max)
agent_deb_amd64 +4.35 KiB (0.00% increase) 755.591 → 755.595 → 759.670
agent_deb_amd64_fips +4.35 KiB (0.00% increase) 714.542 → 714.547 → 720.180
agent_heroku_amd64 +4.0 KiB (0.00% increase) 323.359 → 323.363 → 329.530
agent_msi +5.0 KiB (0.00% increase) 621.912 → 621.917 → 1073.220
agent_rpm_amd64 +4.35 KiB (0.00% increase) 755.575 → 755.579 → 759.640
agent_rpm_amd64_fips +4.35 KiB (0.00% increase) 714.526 → 714.530 → 720.160
agent_rpm_arm64 +4.57 KiB (0.00% increase) 733.870 → 733.874 → 741.840
agent_rpm_arm64_fips +4.57 KiB (0.00% increase) 695.724 → 695.729 → 703.390
agent_suse_amd64 +4.35 KiB (0.00% increase) 755.575 → 755.579 → 759.640
agent_suse_amd64_fips +4.35 KiB (0.00% increase) 714.526 → 714.530 → 720.160
agent_suse_arm64 +4.57 KiB (0.00% increase) 733.870 → 733.874 → 741.840
agent_suse_arm64_fips +4.57 KiB (0.00% increase) 695.724 → 695.729 → 703.390
docker_agent_amd64 +4.36 KiB (0.00% increase) 816.650 → 816.655 → 821.990
docker_agent_arm64 +11.62 KiB (0.00% increase) 819.646 → 819.657 → 828.520
docker_agent_jmx_amd64 +4.35 KiB (0.00% increase) 1007.562 → 1007.566 → 1012.870
docker_agent_jmx_arm64 +11.62 KiB (0.00% increase) 999.340 → 999.351 → 1008.120
15 successful checks with minimal change (< 2 KiB)
Quality gate Current Size
docker_cluster_agent_amd64 190.326 MiB
docker_cluster_agent_arm64 205.734 MiB
docker_cws_instrumentation_amd64 7.135 MiB
docker_cws_instrumentation_arm64 6.689 MiB
docker_dogstatsd_amd64 38.469 MiB
docker_dogstatsd_arm64 36.812 MiB
dogstatsd_deb_amd64 29.688 MiB
dogstatsd_deb_arm64 27.853 MiB
dogstatsd_rpm_amd64 29.688 MiB
dogstatsd_suse_amd64 29.688 MiB
iot_agent_deb_amd64 42.889 MiB
iot_agent_deb_arm64 39.998 MiB
iot_agent_deb_armhf 40.572 MiB
iot_agent_rpm_amd64 42.889 MiB
iot_agent_suse_amd64 42.889 MiB
On-wire sizes (compressed)
Quality gate Change Size (prev → curr → max)
agent_deb_amd64 +28.08 KiB (0.01% increase) 185.029 → 185.056 → 186.090
agent_deb_amd64_fips +5.32 KiB (0.00% increase) 176.071 → 176.076 → 180.330
agent_heroku_amd64 neutral 87.220 MiB → 88.440
agent_msi +32.0 KiB (0.02% increase) 143.316 → 143.348 → 143.470
agent_rpm_amd64 +2.12 KiB (0.00% increase) 186.983 → 186.985 → 189.170
agent_rpm_amd64_fips -19.54 KiB (0.01% reduction) 178.100 → 178.081 → 181.060
agent_rpm_arm64 +5.95 KiB (0.00% increase) 169.157 → 169.163 → 170.020
agent_rpm_arm64_fips -7.39 KiB (0.00% reduction) 162.098 → 162.091 → 164.130
agent_suse_amd64 +2.12 KiB (0.00% increase) 186.983 → 186.985 → 189.170
agent_suse_amd64_fips -19.54 KiB (0.01% reduction) 178.100 → 178.081 → 181.060
agent_suse_arm64 +5.95 KiB (0.00% increase) 169.157 → 169.163 → 170.020
agent_suse_arm64_fips -7.39 KiB (0.00% reduction) 162.098 → 162.091 → 164.130
docker_agent_amd64 +10.25 KiB (0.00% increase) 277.384 → 277.394 → 279.410
docker_agent_arm64 -2.61 KiB (0.00% reduction) 264.728 → 264.725 → 267.960
docker_agent_jmx_amd64 +4.96 KiB (0.00% increase) 346.033 → 346.038 → 348.040
docker_agent_jmx_arm64 neutral 329.355 MiB → 332.560
docker_cluster_agent_amd64 neutral 66.492 MiB → 67.220
docker_cluster_agent_arm64 neutral 62.544 MiB → 63.640
docker_cws_instrumentation_amd64 neutral 2.995 MiB → 3.330
docker_cws_instrumentation_arm64 neutral 2.726 MiB → 3.090
docker_dogstatsd_amd64 neutral 14.892 MiB → 15.820
docker_dogstatsd_arm64 neutral 14.230 MiB → 14.830
dogstatsd_deb_amd64 neutral 7.849 MiB → 8.790
dogstatsd_deb_arm64 neutral 6.735 MiB → 7.710
dogstatsd_rpm_amd64 neutral 7.860 MiB → 8.800
dogstatsd_suse_amd64 neutral 7.860 MiB → 8.800
iot_agent_deb_amd64 -3.24 KiB (0.03% reduction) 11.258 → 11.255 → 12.040
iot_agent_deb_arm64 neutral 9.621 MiB → 10.450
iot_agent_deb_armhf +3.07 KiB (0.03% increase) 9.819 → 9.822 → 10.620
iot_agent_rpm_amd64 neutral 11.271 MiB → 12.060
iot_agent_suse_amd64 neutral 11.271 MiB → 12.060

@@ -419,7 +431,8 @@ func (s *sender) do(req *http.Request) error {

// isRetriable reports whether the give HTTP status code should be retried.
func isRetriable(code int) bool {
if code == http.StatusRequestTimeout || code == http.StatusTooManyRequests {
// TODO: Double check what response codes are expected from the backend when API Key is invalid
if code == http.StatusRequestTimeout || code == http.StatusTooManyRequests || code == http.StatusForbidden {
Copy link
Contributor

Choose a reason for hiding this comment

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

One concern I have here is what do we do when the Forbidden status code is legit / the API key has been rotated and the customer is not using RC / api key refresh? I suppose it's probably not a big deal since if the API key is bad then we're going to drop things anyways, but it does mean we'll retry requests more than we probably should? I'm wondering if there's a way we can gate these retries so that they only retry when we know we can try to get a new api key (And then only retry after we have a new api key?).

Copy link
Contributor Author

Choose a reason for hiding this comment

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

This is a good question. I initially approached the issue from that angle, but quickly saw that keeping track of invalid keys and blocking the sender had it's own tradeoffs: more complex api key changes propagation, refactor goroutines to avoid race conditions when reading/writing the key... and the sender will still be unable to send other payloads if the api key is invalid. This concern you raise is also the reason I'm not 100% fond of the proposed solution 🤔

@cit-pr-commenter-54b7da
Copy link

cit-pr-commenter-54b7da bot commented Jan 29, 2026

Regression Detector

Regression Detector Results

Metrics dashboard
Target profiles
Run ID: a230046b-4e35-4e87-b965-e8de71f1636c

Baseline: bc99348
Comparison: c9a1cef
Diff

Optimization Goals: ✅ No significant changes detected

Experiments ignored for regressions

Regressions in experiments with settings containing erratic: true are ignored.

perf experiment goal Δ mean % Δ mean % CI trials links
docker_containers_cpu % cpu utilization -2.22 [-5.30, +0.86] 1 Logs

Fine details of change detection per experiment

perf experiment goal Δ mean % Δ mean % CI trials links
otlp_ingest_logs memory utilization +0.60 [+0.50, +0.70] 1 Logs
tcp_syslog_to_blackhole ingress throughput +0.53 [+0.43, +0.62] 1 Logs
quality_gate_idle memory utilization +0.51 [+0.46, +0.55] 1 Logs bounds checks dashboard
quality_gate_idle_all_features memory utilization +0.42 [+0.38, +0.46] 1 Logs bounds checks dashboard
docker_containers_memory memory utilization +0.21 [+0.13, +0.29] 1 Logs
file_tree memory utilization +0.14 [+0.09, +0.19] 1 Logs
ddot_logs memory utilization +0.12 [+0.06, +0.17] 1 Logs
tcp_dd_logs_filter_exclude ingress throughput +0.01 [-0.09, +0.10] 1 Logs
file_to_blackhole_100ms_latency egress throughput -0.00 [-0.05, +0.05] 1 Logs
uds_dogstatsd_to_api_v3 ingress throughput -0.01 [-0.13, +0.11] 1 Logs
uds_dogstatsd_to_api ingress throughput -0.01 [-0.14, +0.11] 1 Logs
file_to_blackhole_0ms_latency egress throughput -0.05 [-0.52, +0.43] 1 Logs
file_to_blackhole_500ms_latency egress throughput -0.05 [-0.43, +0.33] 1 Logs
file_to_blackhole_1000ms_latency egress throughput -0.07 [-0.49, +0.36] 1 Logs
otlp_ingest_metrics memory utilization -0.09 [-0.24, +0.06] 1 Logs
ddot_metrics_sum_delta memory utilization -0.10 [-0.31, +0.11] 1 Logs
uds_dogstatsd_20mb_12k_contexts_20_senders memory utilization -0.24 [-0.29, -0.19] 1 Logs
ddot_metrics_sum_cumulativetodelta_exporter memory utilization -0.31 [-0.55, -0.08] 1 Logs
quality_gate_metrics_logs memory utilization -0.38 [-0.59, -0.17] 1 Logs bounds checks dashboard
ddot_metrics memory utilization -0.63 [-0.85, -0.42] 1 Logs
ddot_metrics_sum_cumulative memory utilization -0.64 [-0.81, -0.48] 1 Logs
quality_gate_logs % cpu utilization -1.51 [-2.99, -0.02] 1 Logs bounds checks dashboard
docker_containers_cpu % cpu utilization -2.22 [-5.30, +0.86] 1 Logs

Bounds Checks: ✅ Passed

perf experiment bounds_check_name replicates_passed links
docker_containers_cpu simple_check_run 10/10
docker_containers_memory memory_usage 10/10
docker_containers_memory simple_check_run 10/10
file_to_blackhole_0ms_latency lost_bytes 10/10
file_to_blackhole_0ms_latency memory_usage 10/10
file_to_blackhole_1000ms_latency lost_bytes 10/10
file_to_blackhole_1000ms_latency memory_usage 10/10
file_to_blackhole_100ms_latency lost_bytes 10/10
file_to_blackhole_100ms_latency memory_usage 10/10
file_to_blackhole_500ms_latency lost_bytes 10/10
file_to_blackhole_500ms_latency memory_usage 10/10
quality_gate_idle intake_connections 10/10 bounds checks dashboard
quality_gate_idle memory_usage 10/10 bounds checks dashboard
quality_gate_idle_all_features intake_connections 10/10 bounds checks dashboard
quality_gate_idle_all_features memory_usage 10/10 bounds checks dashboard
quality_gate_logs intake_connections 10/10 bounds checks dashboard
quality_gate_logs lost_bytes 10/10 bounds checks dashboard
quality_gate_logs memory_usage 10/10 bounds checks dashboard
quality_gate_metrics_logs cpu_usage 10/10 bounds checks dashboard
quality_gate_metrics_logs intake_connections 10/10 bounds checks dashboard
quality_gate_metrics_logs lost_bytes 10/10 bounds checks dashboard
quality_gate_metrics_logs memory_usage 10/10 bounds checks dashboard

Explanation

Confidence level: 90.00%
Effect size tolerance: |Δ mean %| ≥ 5.00%

Performance changes are noted in the perf column of each table:

  • ✅ = significantly better comparison variant performance
  • ❌ = significantly worse comparison variant performance
  • ➖ = no significant change in performance

A regression test is an A/B test of target performance in a repeatable rig, where "performance" is measured as "comparison variant minus baseline variant" for an optimization goal (e.g., ingress throughput). Due to intrinsic variability in measuring that goal, we can only estimate its mean value for each experiment; we report uncertainty in that value as a 90.00% confidence interval denoted "Δ mean % CI".

For each experiment, we decide whether a change in performance is a "regression" -- a change worth investigating further -- if all of the following criteria are true:

  1. Its estimated |Δ mean %| ≥ 5.00%, indicating the change is big enough to merit a closer look.

  2. Its 90.00% confidence interval "Δ mean % CI" does not contain zero, indicating that if our statistical model is accurate, there is at least a 90.00% chance there is a difference in performance between baseline and comparison variants.

  3. Its configuration does not mark it "erratic".

CI Pass/Fail Decision

Passed. All Quality Gates passed.

  • quality_gate_idle_all_features, bounds check memory_usage: 10/10 replicas passed. Gate passed.
  • quality_gate_idle_all_features, bounds check intake_connections: 10/10 replicas passed. Gate passed.
  • quality_gate_logs, bounds check lost_bytes: 10/10 replicas passed. Gate passed.
  • quality_gate_logs, bounds check memory_usage: 10/10 replicas passed. Gate passed.
  • quality_gate_logs, bounds check intake_connections: 10/10 replicas passed. Gate passed.
  • quality_gate_idle, bounds check intake_connections: 10/10 replicas passed. Gate passed.
  • quality_gate_idle, bounds check memory_usage: 10/10 replicas passed. Gate passed.
  • quality_gate_metrics_logs, bounds check lost_bytes: 10/10 replicas passed. Gate passed.
  • quality_gate_metrics_logs, bounds check cpu_usage: 10/10 replicas passed. Gate passed.
  • quality_gate_metrics_logs, bounds check intake_connections: 10/10 replicas passed. Gate passed.
  • quality_gate_metrics_logs, bounds check memory_usage: 10/10 replicas passed. Gate passed.

ichinaski and others added 3 commits February 4, 2026 12:47
Make API Key access safe for concurrent use. API Key can be read from
worker goroutines, and updated from a different one.
@ichinaski ichinaski requested a review from a team as a code owner February 4, 2026 11:58
Instead of relying on the secrets component throttling,
we will issue a single refresh request when a 403 is received, and
throttle these calls within the Trace Agent writer.
@dd-octo-sts dd-octo-sts bot added the internal Identify a non-fork PR label Feb 18, 2026
@ichinaski
Copy link
Contributor Author

/merge

@gh-worker-devflow-routing-ef8351
Copy link

gh-worker-devflow-routing-ef8351 bot commented Feb 23, 2026

View all feedbacks in Devflow UI.

2026-02-23 15:41:12 UTC ℹ️ Start processing command /merge


2026-02-23 15:41:17 UTC ℹ️ MergeQueue: pull request added to the queue

The expected merge time in main is approximately 1h (p90).


2026-02-23 16:08:32 UTC ℹ️ MergeQueue: Readding this merge request to the queue because another merge request processed with yours failed. No action is needed from your side.


2026-02-23 18:48:30 UTC ℹ️ MergeQueue: This merge request was merged

@gh-worker-dd-mergequeue-cf854d gh-worker-dd-mergequeue-cf854d bot merged commit e4a4e2d into main Feb 23, 2026
292 checks passed
@gh-worker-dd-mergequeue-cf854d gh-worker-dd-mergequeue-cf854d bot deleted the ichinaski/api-key-refresh-retry-payloads branch February 23, 2026 18:48
@github-actions github-actions bot added this to the 7.78.0 milestone Feb 23, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

internal Identify a non-fork PR medium review PR review might take time qa/done QA done before merge and regressions are covered by tests team/agent-apm trace-agent

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants