feat(metrics): always set namespace and service, update tests for ser…

…vice handling
aws-powertools · Mar 7, 2025 · 8b25456 · 8b25456
1 parent 1e95a20
commit 8b25456
Show file tree

Hide file tree

Showing 7 changed files with 40 additions and 12 deletions.
diff --git a/libraries/src/AWS.Lambda.Powertools.Metrics/Metrics.cs b/libraries/src/AWS.Lambda.Powertools.Metrics/Metrics.cs
@@ -180,8 +180,9 @@ internal Metrics(IPowertoolsConfigurations powertoolsConfigurations, string name
         Instance = this;
         _powertoolsConfigurations.SetExecutionEnvironment(this);
 
-        if (!string.IsNullOrEmpty(nameSpace)) SetNamespace(nameSpace);
-        if (!string.IsNullOrEmpty(service)) SetService(service);
+        // set namespace and service always
+        SetNamespace(nameSpace);
+        SetService(service);
     }
 
     /// <inheritdoc />
@@ -364,7 +365,19 @@ void IMetrics.PushSingleMetric(string name, double value, MetricUnit unit, strin
 
         var context = new MetricsContext();
         context.SetNamespace(nameSpace ?? GetNamespace());
-        context.SetService(service ?? _context.GetService());
+
+        var parsedService = !string.IsNullOrWhiteSpace(service)
+            ? service
+            : _powertoolsConfigurations.Service == "service_undefined"
+                ? null
+                : _powertoolsConfigurations.Service;
+
+        // var serviceTodAdd = parsedService ?? _context.GetService();
+        if (!string.IsNullOrWhiteSpace(parsedService))
+        {
+            context.SetService(parsedService);
+            context.AddDimension("Service", parsedService);
+        }
 
         if (dimensions != null)
         {

diff --git a/...es/tests/AWS.Lambda.Powertools.Metrics.AspNetCore.Tests/MetricsEndpointExtensionsTests.cs b/...es/tests/AWS.Lambda.Powertools.Metrics.AspNetCore.Tests/MetricsEndpointExtensionsTests.cs
@@ -20,8 +20,7 @@ public async Task When_WithMetrics_Should_Add_ColdStart()
         var options = new MetricsOptions
         {
             CaptureColdStart = true,
-            Namespace = "TestNamespace",
-            Service = "TestService"
+            Namespace = "TestNamespace"
         };
 
         var conf = Substitute.For<IPowertoolsConfigurations>();
@@ -61,7 +60,6 @@ public async Task When_WithMetrics_Should_Add_ColdStart_Dimensions()
         {
             CaptureColdStart = true,
             Namespace = "TestNamespace",
-            Service = "TestService"
         };
 
         var conf = Substitute.For<IPowertoolsConfigurations>();
@@ -111,7 +109,6 @@ public async Task When_WithMetrics_Should_Add_ColdStart_Default_Dimensions()
         {
             CaptureColdStart = true,
             Namespace = "TestNamespace",
-            Service = "TestService",
             DefaultDimensions = new Dictionary<string, string>
             {
                 { "Environment", "Prod" }

diff --git a/.../tests/AWS.Lambda.Powertools.Metrics.AspNetCore.Tests/MetricsMiddlewareExtensionsTests.cs b/.../tests/AWS.Lambda.Powertools.Metrics.AspNetCore.Tests/MetricsMiddlewareExtensionsTests.cs
@@ -21,7 +21,6 @@ public async Task When_UseMetrics_Should_Add_ColdStart()
         {
             CaptureColdStart = true,
             Namespace = "TestNamespace",
-            Service = "TestService"
         };
 
         var conf = Substitute.For<IPowertoolsConfigurations>();
@@ -59,7 +58,6 @@ public async Task When_UseMetrics_Should_Add_ColdStart_With_LambdaContext()
         {
             CaptureColdStart = true,
             Namespace = "TestNamespace",
-            Service = "TestService"
         };
 
         var conf = Substitute.For<IPowertoolsConfigurations>();

diff --git a/libraries/tests/AWS.Lambda.Powertools.Metrics.Tests/EMFValidationTests.cs b/libraries/tests/AWS.Lambda.Powertools.Metrics.Tests/EMFValidationTests.cs
@@ -216,6 +216,12 @@ public void When_PushSingleMetric_With_Env_Namespace()
 
             // Assert
             Assert.Contains("\"CloudWatchMetrics\":[{\"Namespace\":\"EnvNamespace\",\"Metrics\":[{\"Name\":\"SingleMetric\",\"Unit\":\"Count\",\"StorageResolution\":1}],\"Dimensions\":[[\"Default\"]]}]},\"Default\":\"SingleMetric\",\"SingleMetric\":1}", metricsOutput);
+
+            // assert with different service name
+            Assert.Contains("\"CloudWatchMetrics\":[{\"Namespace\":\"EnvNamespace\",\"Metrics\":[{\"Name\":\"SingleMetric2\",\"Unit\":\"Count\",\"StorageResolution\":1}],\"Dimensions\":[[\"Service\",\"Default\"]]}]},\"Service\":\"service1\",\"Default\":\"SingleMetric\",\"SingleMetric2\":1}", metricsOutput);
+
+            // assert with different service name
+            Assert.Contains("\"CloudWatchMetrics\":[{\"Namespace\":\"EnvNamespace\",\"Metrics\":[{\"Name\":\"SingleMetric3\",\"Unit\":\"Count\",\"StorageResolution\":1}],\"Dimensions\":[[\"Service\",\"Default\"]]}]},\"Service\":\"service2\",\"Default\":\"SingleMetric\",\"SingleMetric3\":1}", metricsOutput);
         }
 
         [Trait("Category", "MetricsImplementation")]

diff --git a/libraries/tests/AWS.Lambda.Powertools.Metrics.Tests/Handlers/FunctionHandler.cs b/libraries/tests/AWS.Lambda.Powertools.Metrics.Tests/Handlers/FunctionHandler.cs
@@ -86,6 +86,18 @@ public void PushSingleMetricWithEnvNamespace()
             dimensions: new Dictionary<string, string> {
                 { "Default", "SingleMetric" }
             });
+
+        Metrics.PushSingleMetric("SingleMetric2", 1, MetricUnit.Count, resolution: MetricResolution.High,
+            service: "service1",
+            dimensions: new Dictionary<string, string> {
+                { "Default", "SingleMetric" }
+            });
+
+        Metrics.PushSingleMetric("SingleMetric3", 1, MetricUnit.Count, resolution: MetricResolution.High,
+            service: "service2",
+            dimensions: new Dictionary<string, string> {
+                { "Default", "SingleMetric" }
+            });
     }
 
     [Metrics(Namespace = "dotnet-powertools-test", Service = "testService")]

diff --git a/libraries/tests/AWS.Lambda.Powertools.Metrics.Tests/MetricsTests.cs b/libraries/tests/AWS.Lambda.Powertools.Metrics.Tests/MetricsTests.cs
@@ -234,7 +234,6 @@ public void When_ColdStart_Should_Use_DefaultDimensions_From_Options()
         {
             CaptureColdStart = true,
             Namespace = "dotnet-powertools-test",
-            Service = "testService",
             DefaultDimensions = new Dictionary<string, string>
             {
                 { "Environment", "Test" },
@@ -268,7 +267,6 @@ public void When_ColdStart_And_DefaultDimensions_Is_Null_Should_Only_Add_Service
         {
             CaptureColdStart = true,
             Namespace = "dotnet-powertools-test",
-            Service = "testService",
             DefaultDimensions = null
         };
 

diff --git a/libraries/tests/e2e/functions/core/metrics/Function/test/Function.Tests/FunctionTests.cs b/libraries/tests/e2e/functions/core/metrics/Function/test/Function.Tests/FunctionTests.cs
@@ -270,11 +270,15 @@ private void AssertSingleMetric(string output)
         Assert.Equal("Count", unitElement.GetString());
 
         Assert.True(cloudWatchMetricsElement[0].TryGetProperty("Dimensions", out JsonElement dimensionsElement));
-        Assert.Equal("FunctionName", dimensionsElement[0][0].GetString());
+        Assert.Equal("Service", dimensionsElement[0][0].GetString());
+        Assert.Equal("FunctionName", dimensionsElement[0][1].GetString());
 
         Assert.True(root.TryGetProperty("FunctionName", out JsonElement functionNameElement));
         Assert.Equal(_functionName, functionNameElement.GetString());
 
+        Assert.True(root.TryGetProperty("Service", out JsonElement serviceElement));
+        Assert.Equal("Test", serviceElement.GetString());
+
         Assert.True(root.TryGetProperty("SingleMetric", out JsonElement singleMetricElement));
         Assert.Equal(1, singleMetricElement.GetInt32());
     }