fix(metrics): Support multiple addMetric() call with the same metric name (#390)

* fix(metrics): Support multiple addMetric() call with the same metric name

* docs(metrics): Add section about adding the same metric name multiple time

* docs(metrics): Rearrange excerpt example to be the same as others

* docs(metrics) Change to more appropriate term

Co-authored-by: Heitor Lessa <[email protected]>

* fix(metrics): Fix typos + grammar in a comment

Co-authored-by: Sara Gerion <[email protected]>

* fix(metrics): change an error message to suggest a potential solution

Co-authored-by: Heitor Lessa <[email protected]>
Co-authored-by: Sara Gerion <[email protected]>

Author: 3 people

docs/core/metrics.md

@@ -132,6 +132,54 @@ You can create metrics using `addMetric`, and you can create dimensions for all
 !!! warning "Do not create metrics or dimensions outside the handler"
     Metrics or dimensions added in the global scope will only be added during cold start. Disregard if that's the intended behaviour.
 
+### Adding multi-value metrics
+You can call `addMetric()` with the same name multiple times. The values will be grouped together in an array.
+
+=== "addMetric() with the same name"
+
+    ```typescript hl_lines="8 10"
+    import { Metrics, MetricUnits } from '@aws-lambda-powertools/metrics';
+    import { Context } from 'aws-lambda'; 
+
+
+    const metrics = new Metrics({namespace:"serverlessAirline", service:"orders"});
+
+    export const handler = async (event: any, context: Context) => {
+        metrics.addMetric('performedActionA', MetricUnits.Count, 2);
+        // do something else...
+        metrics.addMetric('performedActionA', MetricUnits.Count, 1);
+    }
+    ```
+=== "Example CloudWatch Logs excerpt"
+
+    ```json hl_lines="2-5 18-19"
+    {
+        "performedActionA": [
+            2,
+            1
+        ],
+        "_aws": {
+            "Timestamp": 1592234975665,
+            "CloudWatchMetrics": [
+                {
+                "Namespace": "serverlessAirline",
+                "Dimensions": [
+                    [
+                    "service"
+                    ]
+                ],
+                "Metrics": [
+                    {
+                    "Name": "performedActionA",
+                    "Unit": "Count"
+                    }
+                ]
+                }
+            ]
+        },
+        "service": "orders"
+    }
+    ```
 ### Adding default dimensions
 
 You can use add default dimensions to your metrics by passing them as parameters in 4 ways:  
@@ -264,23 +312,23 @@ See below an example of how to automatically flush metrics with the Middy-compat
     {
         "bookingConfirmation": 1.0,
         "_aws": {
-        "Timestamp": 1592234975665,
-        "CloudWatchMetrics": [
-            {
-            "Namespace": "exampleApplication",
-            "Dimensions": [
-                [
-                "service"
-                ]
-            ],
-            "Metrics": [
+            "Timestamp": 1592234975665,
+            "CloudWatchMetrics": [
                 {
-                "Name": "bookingConfirmation",
-                "Unit": "Count"
+                "Namespace": "exampleApplication",
+                "Dimensions": [
+                    [
+                    "service"
+                    ]
+                ],
+                "Metrics": [
+                    {
+                    "Name": "bookingConfirmation",
+                    "Unit": "Count"
+                    }
+                ]
                 }
             ]
-            }
-        ]
         },
         "service": "exampleService"
     }
@@ -316,23 +364,23 @@ export class MyFunction {
     {
         "bookingConfirmation": 1.0,
         "_aws": {
-        "Timestamp": 1592234975665,
-        "CloudWatchMetrics": [
-            {
-            "Namespace": "exampleApplication",
-            "Dimensions": [
-                [
-                "service"
-                ]
-            ],
-            "Metrics": [
+            "Timestamp": 1592234975665,
+            "CloudWatchMetrics": [
                 {
-                "Name": "bookingConfirmation",
-                "Unit": "Count"
+                "Namespace": "exampleApplication",
+                "Dimensions": [
+                    [
+                    "service"
+                    ]
+                ],
+                "Metrics": [
+                    {
+                    "Name": "bookingConfirmation",
+                    "Unit": "Count"
+                    }
+                ]
                 }
             ]
-            }
-        ]
         },
         "service": "exampleService"
     }
@@ -456,23 +504,23 @@ You can add high-cardinality data as part of your Metrics log with `addMetadata`
     {
         "successfulBooking": 1.0,
         "_aws": {
-        "Timestamp": 1592234975665,
-        "CloudWatchMetrics": [
-            {
-            "Namespace": "exampleApplication",
-            "Dimensions": [
-                [
-                "service"
-                ]
-            ],
-            "Metrics": [
+            "Timestamp": 1592234975665,
+            "CloudWatchMetrics": [
                 {
-                "Name": "successfulBooking",
-                "Unit": "Count"
+                "Namespace": "exampleApplication",
+                "Dimensions": [
+                    [
+                    "service"
+                    ]
+                ],
+                "Metrics": [
+                    {
+                    "Name": "successfulBooking",
+                    "Unit": "Count"
+                    }
+                ]
                 }
             ]
-            }
-        ]
         },
         "service": "booking",
         "bookingId": "7051cd10-6283-11ec-90d6-0242ac120003"

packages/metrics/package-lock.json

@@ -10,7 +10,7 @@
     "commit": "commit",
     "test": "jest --group=unit --detectOpenHandles --coverage --verbose",
     "test:e2e": "jest --group=e2e",
-    "watch": "jest --watch",
+    "watch": "jest --group=unit --watch ",
     "build": "tsc",
     "lint": "eslint --ext .ts --fix --no-error-on-unmatched-pattern src tests",
     "format": "eslint --fix --ext .ts --fix --no-error-on-unmatched-pattern src tests",

packages/metrics/package.json

@@ -312,7 +312,7 @@ class Metrics implements MetricsInterface {
     if (!this.namespace) console.warn('Namespace should be defined, default used');
 
     const metricValues = Object.values(this.storedMetrics).reduce(
-      (result: { [key: string]: number }, { name, value }: { name: string; value: number }) => {
+      (result: { [key: string]: number | number[] }, { name, value }: { name: string; value: number | number[] }) => {
         result[name] = value;
 
         return result;
@@ -391,6 +391,20 @@ class Metrics implements MetricsInterface {
     return <EnvironmentVariablesService> this.envVarsService;
   }
 
+  private isNewMetric(name: string, unit: MetricUnit): boolean {
+    if (this.storedMetrics[name]){
+      // Inconsistent units indicates a bug or typos and we want to flag this to users early
+      if (this.storedMetrics[name].unit !== unit) {
+        const currentUnit = this.storedMetrics[name].unit;
+        throw new Error(`Metric "${name}" has already been added with unit "${currentUnit}", but we received unit "${unit}". Did you mean to use metric unit "${currentUnit}"?`);
+      }
+      
+      return false;
+    } else {
+      return true;
+    }
+  }
+
   private setCustomConfigService(customConfigService?: ConfigServiceInterface): void {
     this.customConfigService = customConfigService ? customConfigService : undefined;
   }
@@ -431,12 +445,22 @@ class Metrics implements MetricsInterface {
     if (Object.keys(this.storedMetrics).length >= MAX_METRICS_SIZE) {
       this.publishStoredMetrics();
     }
-    this.storedMetrics[name] = {
-      unit,
-      value,
-      name,
-    };
+
+    if (this.isNewMetric(name, unit)) {
+      this.storedMetrics[name] = {
+        unit,
+        value,
+        name,
+      };
+    } else {
+      const storedMetric = this.storedMetrics[name];
+      if (!Array.isArray(storedMetric.value)) {
+        storedMetric.value = [storedMetric.value];
+      }
+      storedMetric.value.push(value);
+    }
   }
+
 }
 
 export { Metrics, MetricUnits };

packages/metrics/src/Metrics.ts

@@ -59,7 +59,7 @@ type ExtraOptions = {
 type StoredMetric = {
   name: string
   unit: MetricUnit
-  value: number
+  value: number | number[]
 };
 
 type StoredMetrics = {

packages/metrics/src/types/Metrics.ts

@@ -435,6 +435,55 @@ describe('Class: Metrics', () => {
       expect(serializedMetrics._aws.CloudWatchMetrics[0].Namespace).toBe(DEFAULT_NAMESPACE);
       expect(console.warn).toHaveBeenNthCalledWith(1, 'Namespace should be defined, default used');
     });
+
+    test('Should contain a metric value if added once', ()=> {
+      const metrics = new Metrics();
+      
+      metrics.addMetric('test_name', MetricUnits.Count, 1);
+      const serializedMetrics = metrics.serializeMetrics();
+
+      expect(serializedMetrics._aws.CloudWatchMetrics[0].Metrics.length).toBe(1);
+      
+      expect(serializedMetrics['test_name']).toBe(1);
+    });
+
+    test('Should convert multiple metrics with the same name and unit into an array', ()=> {
+      const metrics = new Metrics();
+      
+      metrics.addMetric('test_name', MetricUnits.Count, 2);
+      metrics.addMetric('test_name', MetricUnits.Count, 1);
+      const serializedMetrics = metrics.serializeMetrics();
+
+      expect(serializedMetrics._aws.CloudWatchMetrics[0].Metrics.length).toBe(1);
+      expect(serializedMetrics['test_name']).toStrictEqual([ 2, 1 ]);
+    });
+
+    test('Should throw an error if the same metric name is added again with a different unit', ()=> {
+      const metrics = new Metrics();
+      
+      metrics.addMetric('test_name', MetricUnits.Count, 2);
+      try {
+        metrics.addMetric('test_name', MetricUnits.Seconds, 10);
+      } catch (e) {
+        expect((<Error>e).message).toBe('Metric "test_name" has already been added with unit "Count", but we received unit "Seconds". Did you mean to use metric unit "Count"?');
+      }
+    });
+
+    test('Should contain multiple metric values if added with multiple names', ()=> {
+      const metrics = new Metrics();
+      
+      metrics.addMetric('test_name', MetricUnits.Count, 1);
+      metrics.addMetric('test_name2', MetricUnits.Count, 2);
+      const serializedMetrics = metrics.serializeMetrics();
+
+      expect(serializedMetrics._aws.CloudWatchMetrics[0].Metrics).toStrictEqual([
+        { Name: 'test_name', Unit: 'Count' },
+        { Name: 'test_name2', Unit: 'Count' },
+      ]);
+      
+      expect(serializedMetrics['test_name']).toBe(1);
+      expect(serializedMetrics['test_name2']).toBe(2);
+    });
   });
 
   describe('Feature: Clearing Metrics ', () => {

packages/metrics/tests/unit/Metrics.test.ts

@@ -43,14 +43,47 @@ describe('Middy middleware', () => {
       succeed: () => console.log('Succeeded!'),
     };
 
+    test('when a metrics instance receive multiple metrics with the same name, it prints multiple values in an array format', async () => {
+      // Prepare
+      const metrics = new Metrics({ namespace:'serverlessAirline', service:'orders' });
+
+      const lambdaHandler = (): void => {
+        metrics.addMetric('successfulBooking', MetricUnits.Count, 2);
+        metrics.addMetric('successfulBooking', MetricUnits.Count, 1);
+      };
+
+      const handler = middy(lambdaHandler).use(logMetrics(metrics));
+      
+      // Act
+      await handler(event, context, () => console.log('Lambda invoked!'));
+      
+      // Assess
+      expect(console.log).toHaveBeenNthCalledWith(1, JSON.stringify({
+        '_aws': {
+          'Timestamp': 1466424490000,
+          'CloudWatchMetrics': [{
+            'Namespace': 'serverlessAirline',
+            'Dimensions': [
+              ['service']
+            ],
+            'Metrics': [{ 'Name': 'successfulBooking', 'Unit': 'Count' }],
+          }],
+        },
+        'service': 'orders',
+        'successfulBooking': [
+          2,
+          1,
+        ],
+      }));
+    });
+
     test('when a metrics instance is passed WITH custom options, it prints the metrics in the stdout', async () => {
 
       // Prepare
       const metrics = new Metrics({ namespace:'serverlessAirline', service:'orders' });
 
       const lambdaHandler = (): void => {
         metrics.addMetric('successfulBooking', MetricUnits.Count, 1);
-        metrics.addMetric('successfulBooking', MetricUnits.Count, 1);
       };
       const metricsOptions: ExtraOptions = {
         raiseOnEmptyMetrics: true,
@@ -106,7 +139,6 @@ describe('Middy middleware', () => {
 
       const lambdaHandler = (): void => {
         metrics.addMetric('successfulBooking', MetricUnits.Count, 1);
-        metrics.addMetric('successfulBooking', MetricUnits.Count, 1);
       };
 
       const handler = middy(lambdaHandler).use(logMetrics(metrics));
@@ -139,7 +171,6 @@ describe('Middy middleware', () => {
 
       const lambdaHandler = (): void => {
         metrics.addMetric('successfulBooking', MetricUnits.Count, 1);
-        metrics.addMetric('successfulBooking', MetricUnits.Count, 1);
       };
       const metricsOptions: ExtraOptions = {
         raiseOnEmptyMetrics: true