Skip to content

perf(preprocessor): 3 Go tokenizer pipeline optimizations#51181

Draft
DDuongNguyen wants to merge 1 commit into
mainfrom
yoon/tokenizer-in-rust
Draft

perf(preprocessor): 3 Go tokenizer pipeline optimizations#51181
DDuongNguyen wants to merge 1 commit into
mainfrom
yoon/tokenizer-in-rust

Conversation

@DDuongNguyen
Copy link
Copy Markdown
Contributor

Summary

  • JSON fast-reject: looksLikeJSON() checks first byte before json.Valid() — 10x faster for non-JSON messages
  • IsMatch length pre-filter: couldMatch() rejects by token count before full comparison — 4% faster sampler scan
  • Index zero-copy: Tokenize() returns indices as buffer view, eliminating 1 alloc per call — 24% faster tokenizer

Test plan

  • All existing preprocessor tests pass
  • Microbenchmarks show improvements
  • SMP regression test with CI-built image

🤖 Generated with Claude Code

1. JSON fast-reject: looksLikeJSON() checks first non-whitespace byte
   before calling json.Valid(). Non-JSON messages skip parsing entirely.
   10x faster for non-JSON (990ns → 99ns), 7x on mixed corpus.

2. IsMatch length pre-filter: couldMatch() rejects pattern candidates
   by token count ratio before full token-by-token comparison.
   ~4% faster on sampler scan path.

3. Index zero-copy: Tokenize() returns indices as a view into internal
   buffer instead of allocating a fresh slice. Callers only read indices
   (in limitTokensToBytes), never store them.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
@github-actions
Copy link
Copy Markdown
Contributor

@codex review

@dd-octo-sts dd-octo-sts Bot added internal Identify a non-fork PR team/agent-log-pipelines labels May 21, 2026
@github-actions github-actions Bot added the medium review PR review might take time label May 21, 2026
@datadog-datadog-prod-us1
Copy link
Copy Markdown
Contributor

datadog-datadog-prod-us1 Bot commented May 21, 2026

Pipelines

Fix all issues with BitsAI

⚠️ Warnings

🚦 21 Pipeline jobs failed

DataDog/datadog-agent | lint_cross_macos-arm64   View in Datadog   GitLab

🔧 Fix in code (Fix with Cursor). Undefined function 'loadBenchCorpus' at pkg/logs/internal/decoder/preprocessor/json_aggregator_benchmark_test.go:106:32

DataDog/datadog-agent | lint_cross_macos-x64   View in Datadog   GitLab

🔧 Fix in code (Fix with Cursor). Linter error: undefined: loadBenchCorpus in json_aggregator_benchmark_test.go:106

DataDog/datadog-agent | lint_cross_windows-x64   View in Datadog   GitLab

🔧 Fix in code (Fix with Cursor). Compilation error in json_aggregator_benchmark_test.go:106:32: undefined: loadBenchCorpus

View all 21 failed jobs.

ℹ️ Info

🎯 Code Coverage (details)
Patch Coverage: 100.00%
Overall Coverage: 50.29% (-0.07%)

Useful? React with 👍 / 👎

This comment will be updated automatically if new data arrives.
🔗 Commit SHA: 133e44c | Docs | Datadog PR Page | Give us feedback!

Copy link
Copy Markdown

@chatgpt-codex-connector chatgpt-codex-connector Bot left a comment

Choose a reason for hiding this comment

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

💡 Codex Review

Here are some automated review suggestions for this pull request.

Reviewed commit: 133e44ce96

ℹ️ About Codex in GitHub

Codex has been enabled to automatically review pull requests in this repo. Reviews are triggered when you

  • Open a pull request for review
  • Mark a draft as ready
  • Comment "@codex review".

If Codex has suggestions, it will comment; otherwise it will react with 👍.

When you sign up for Codex through ChatGPT, Codex can also answer questions or update the PR, like "@codex address that feedback".

Comment on lines +204 to +206
if !couldMatch(len(e.tokens), tokLen, s.config.MatchThreshold) {
continue
}
Copy link
Copy Markdown

Choose a reason for hiding this comment

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

P1 Badge Preserve IsMatch semantics when prefiltering by length

couldMatch is used here as a hard gate before IsMatch, but it is not equivalent to IsMatch’s matching rules: IsMatch compares only min(lenA,lenB) tokens and can return true even when lengths differ a lot (see TestIsMatch case where a long sequence matches "!" at threshold 1.0). With this prefilter, those previously valid matches are now skipped whenever shorter/longer < MatchThreshold (e.g., default 0.9), so logs with optional/variable suffixes are treated as new patterns, reducing sampling effectiveness and causing unnecessary pattern-table churn.

Useful? React with 👍 / 👎.

@dd-octo-sts
Copy link
Copy Markdown
Contributor

dd-octo-sts Bot commented May 21, 2026

Files inventory check summary

File checks results against ancestor e14f24c6:

Results for datadog-agent_7.81.0~devel.git.58.133e44c.pipeline.114602139-1_amd64.deb:

No change detected

@dd-octo-sts
Copy link
Copy Markdown
Contributor

dd-octo-sts Bot commented May 21, 2026

Static quality checks

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

Successful checks

Info

Quality gate Change Size (prev → curr → max)
docker_agent_amd64 +2.54 KiB (0.00% increase) 805.194 → 805.196 → 806.340
docker_agent_jmx_amd64 +2.53 KiB (0.00% increase) 996.114 → 996.116 → 997.060
docker_host_profiler_amd64 +4.56 KiB (0.00% increase) 302.253 → 302.257 → 315.880
docker_host_profiler_arm64 +4.57 KiB (0.00% increase) 313.750 → 313.755 → 327.470
28 successful checks with minimal change (< 2 KiB)
Quality gate Current Size
agent_deb_amd64 745.083 MiB
agent_deb_amd64_fips 702.878 MiB
agent_heroku_amd64 310.120 MiB
agent_rpm_amd64 745.067 MiB
agent_rpm_amd64_fips 702.861 MiB
agent_rpm_arm64 722.637 MiB
agent_rpm_arm64_fips 683.616 MiB
agent_suse_amd64 745.067 MiB
agent_suse_amd64_fips 702.861 MiB
agent_suse_arm64 722.637 MiB
agent_suse_arm64_fips 683.616 MiB
docker_agent_arm64 807.615 MiB
docker_agent_jmx_arm64 987.314 MiB
docker_cluster_agent_amd64 206.897 MiB
docker_cluster_agent_arm64 220.889 MiB
docker_cws_instrumentation_amd64 7.154 MiB
docker_cws_instrumentation_arm64 6.689 MiB
docker_dogstatsd_amd64 39.499 MiB
docker_dogstatsd_arm64 37.690 MiB
dogstatsd_deb_amd64 30.154 MiB
dogstatsd_deb_arm64 28.280 MiB
dogstatsd_rpm_amd64 30.154 MiB
dogstatsd_suse_amd64 30.154 MiB
iot_agent_deb_amd64 44.448 MiB
iot_agent_deb_arm64 41.413 MiB
iot_agent_deb_armhf 42.126 MiB
iot_agent_rpm_amd64 44.449 MiB
iot_agent_suse_amd64 44.449 MiB

@cit-pr-commenter-54b7da
Copy link
Copy Markdown

Regression Detector

Regression Detector Results

Metrics dashboard
Target profiles
Run ID: 1f6cf86b-42a8-4811-bf82-7433e5e68461

Baseline: e14f24c
Comparison: 133e44c
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 +0.38 [-2.55, +3.30] 1 Logs

Fine details of change detection per experiment

perf experiment goal Δ mean % Δ mean % CI trials links
quality_gate_metrics_logs memory utilization +1.22 [+0.96, +1.47] 1 Logs bounds checks dashboard
ddot_logs memory utilization +0.53 [+0.46, +0.60] 1 Logs
tcp_syslog_to_blackhole ingress throughput +0.41 [+0.22, +0.59] 1 Logs
docker_containers_cpu % cpu utilization +0.38 [-2.55, +3.30] 1 Logs
docker_containers_memory memory utilization +0.26 [+0.16, +0.36] 1 Logs
file_to_blackhole_100ms_latency egress throughput +0.03 [-0.10, +0.16] 1 Logs
uds_dogstatsd_to_api_v3 ingress throughput +0.03 [-0.17, +0.22] 1 Logs
uds_dogstatsd_20mb_12k_contexts_20_senders memory utilization +0.01 [-0.04, +0.06] 1 Logs
uds_dogstatsd_to_api ingress throughput +0.00 [-0.20, +0.21] 1 Logs
tcp_dd_logs_filter_exclude ingress throughput -0.00 [-0.10, +0.09] 1 Logs
file_to_blackhole_1000ms_latency egress throughput -0.01 [-0.44, +0.41] 1 Logs
file_to_blackhole_500ms_latency egress throughput -0.05 [-0.45, +0.35] 1 Logs
file_to_blackhole_0ms_latency egress throughput -0.06 [-0.56, +0.44] 1 Logs
quality_gate_idle_all_features memory utilization -0.11 [-0.14, -0.07] 1 Logs bounds checks dashboard
quality_gate_logs % cpu utilization -0.19 [-1.20, +0.81] 1 Logs bounds checks dashboard
quality_gate_idle memory utilization -0.20 [-0.25, -0.15] 1 Logs bounds checks dashboard
otlp_ingest_metrics memory utilization -0.20 [-0.35, -0.05] 1 Logs
otlp_ingest_logs memory utilization -0.33 [-0.42, -0.24] 1 Logs
ddot_metrics_sum_cumulativetodelta_exporter memory utilization -0.34 [-0.58, -0.10] 1 Logs
ddot_metrics_sum_cumulative memory utilization -0.51 [-0.67, -0.35] 1 Logs
ddot_metrics memory utilization -0.52 [-0.72, -0.32] 1 Logs
file_tree memory utilization -0.53 [-0.57, -0.48] 1 Logs
ddot_metrics_sum_delta memory utilization -0.57 [-0.75, -0.39] 1 Logs

Bounds Checks: ✅ Passed

perf experiment bounds_check_name replicates_passed observed_value links
docker_containers_cpu simple_check_run 10/10 531 ≥ 26
docker_containers_memory memory_usage 10/10 251.01MiB ≤ 370MiB
docker_containers_memory simple_check_run 10/10 692 ≥ 26
file_to_blackhole_0ms_latency memory_usage 10/10 0.16GiB ≤ 1.20GiB
file_to_blackhole_0ms_latency missed_bytes 10/10 0B = 0B
file_to_blackhole_1000ms_latency memory_usage 10/10 0.20GiB ≤ 1.20GiB
file_to_blackhole_1000ms_latency missed_bytes 10/10 0B = 0B
file_to_blackhole_100ms_latency memory_usage 10/10 0.17GiB ≤ 1.20GiB
file_to_blackhole_100ms_latency missed_bytes 10/10 0B = 0B
file_to_blackhole_500ms_latency memory_usage 10/10 0.18GiB ≤ 1.20GiB
file_to_blackhole_500ms_latency missed_bytes 10/10 0B = 0B
quality_gate_idle intake_connections 10/10 3 ≤ 4 bounds checks dashboard
quality_gate_idle memory_usage 10/10 144.66MiB ≤ 147MiB bounds checks dashboard
quality_gate_idle_all_features intake_connections 10/10 3 ≤ 4 bounds checks dashboard
quality_gate_idle_all_features memory_usage 10/10 471.64MiB ≤ 495MiB bounds checks dashboard
quality_gate_logs intake_connections 10/10 3 ≤ 6 bounds checks dashboard
quality_gate_logs memory_usage 10/10 177.16MiB ≤ 195MiB bounds checks dashboard
quality_gate_logs missed_bytes 10/10 0B = 0B bounds checks dashboard
quality_gate_metrics_logs cpu_usage 10/10 357.40 ≤ 2000 bounds checks dashboard
quality_gate_metrics_logs intake_connections 10/10 3 ≤ 6 bounds checks dashboard
quality_gate_metrics_logs memory_usage 10/10 399.44MiB ≤ 430MiB bounds checks dashboard
quality_gate_metrics_logs missed_bytes 10/10 0B = 0B 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_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_logs, bounds check missed_bytes: 10/10 replicas passed. Gate passed.
  • quality_gate_metrics_logs, bounds check memory_usage: 10/10 replicas passed. Gate passed.
  • quality_gate_metrics_logs, bounds check missed_bytes: 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 cpu_usage: 10/10 replicas passed. Gate 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_idle, bounds check memory_usage: 10/10 replicas passed. Gate passed.
  • quality_gate_idle, bounds check intake_connections: 10/10 replicas passed. Gate passed.

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 team/agent-log-pipelines

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant