Merge pull request #2276 from DataDog/yl/metrics/add-env-tag

Add env tag in benchmark metrics
DataDog · Sep 23, 2024 · 26c28e2 · 26c28e2
2 parents 5855613 + c546b72
commit 26c28e2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/tools/benchmark/src/main/java/com/datadog/benchmark/internal/MetricRequestBodyBuilder.kt b/tools/benchmark/src/main/java/com/datadog/benchmark/internal/MetricRequestBodyBuilder.kt
@@ -63,7 +63,8 @@ internal class MetricRequestBodyBuilder(private val benchmarkContext: BenchmarkC
             benchmarkContext.scenario?.let {
                 "$KEY_SCENARIO:$it"
             },
-            "$KEY_TAG_APPLICATION_ID:${benchmarkContext.applicationId}"
+            "$KEY_TAG_APPLICATION_ID:${benchmarkContext.applicationId}",
+            "$KEY_ENV:${benchmarkContext.env}"
         )
     }
 
@@ -114,6 +115,7 @@ internal class MetricRequestBodyBuilder(private val benchmarkContext: BenchmarkC
         private const val KEY_TAG_DEVICE_MODEL = "device_model"
         private const val KEY_TAG_OS_VERSION = "os_version"
         private const val KEY_TAG_RUN = "run"
+        private const val KEY_ENV = "env"
         private const val KEY_TAG_APPLICATION_ID = "application_id"
     }
 }
diff --git a/tools/benchmark/src/test/java/internal/MetricRequestBodyBuilderTest.kt b/tools/benchmark/src/test/java/internal/MetricRequestBodyBuilderTest.kt
@@ -68,7 +68,8 @@ class MetricRequestBodyBuilderTest {
                         "os_version:${benchmarkContext.osVersion}",
                         "run:${benchmarkContext.run}",
                         "scenario:${benchmarkContext.scenario}",
-                        "application_id:${benchmarkContext.applicationId}"
+                        "application_id:${benchmarkContext.applicationId}",
+                        "env:${benchmarkContext.env}"
                     )
                 )
                 hasMetricType(resolveMetricType(metricData[metricIndex].type).value)