fix(metrics): ensure metrics are sent

See #4

Author: coderbyheart

lambda/fetchDeviceShadow.ts

@@ -1,5 +1,10 @@
-import { Metrics, MetricUnits } from '@aws-lambda-powertools/metrics'
+import {
+	logMetrics,
+	Metrics,
+	MetricUnits,
+} from '@aws-lambda-powertools/metrics'
 import { DynamoDBClient } from '@aws-sdk/client-dynamodb'
+import middy from '@middy/core'
 import { fromEnv } from '@nordicsemiconductor/from-env'
 import { chunk } from 'lodash-es'
 import pLimit from 'p-limit'
@@ -58,7 +63,7 @@ const deviceShadowPromise = (async () => {
 })()
 const deviceShadowPublisher = createDeviceShadowPublisher(eventBusName)
 
-export const handler = async (): Promise<void> => {
+const h = async (): Promise<void> => {
 	const deviceShadow = await deviceShadowPromise
 	const lockAcquired = await lock.acquiredLock(lockName, lockTTL)
 	if (lockAcquired === false) {
@@ -117,3 +122,5 @@ export const handler = async (): Promise<void> => {
 		await lock.releaseLock(lockName)
 	}
 }
+
+export const handler = middy(h).use(logMetrics(metrics))

lambda/onConnect.ts

@@ -1,4 +1,8 @@
-import { MetricUnits, Metrics } from '@aws-lambda-powertools/metrics'
+import {
+	MetricUnits,
+	Metrics,
+	logMetrics,
+} from '@aws-lambda-powertools/metrics'
 import {
 	DynamoDBClient,
 	PutItemCommand,
@@ -7,6 +11,7 @@ import {
 import { EventBridge } from '@aws-sdk/client-eventbridge'
 import { unmarshall } from '@aws-sdk/util-dynamodb'
 import { Context, DeviceIdentity } from '@hello.nrfcloud.com/proto/hello'
+import middy from '@middy/core'
 import { fromEnv } from '@nordicsemiconductor/from-env'
 import type { Static } from '@sinclair/typebox'
 import type {
@@ -34,7 +39,7 @@ const metrics = new Metrics({
 	serviceName: 'websocket',
 })
 
-export const handler = async (
+const h = async (
 	event: APIGatewayProxyWebsocketEventV2 & {
 		queryStringParameters?: Record<string, any>
 	},
@@ -126,3 +131,5 @@ export const handler = async (
 		body: `Connected. Hello ${event.requestContext.connectionId}@${device.deviceId}!`,
 	}
 }
+
+export const handler = middy(h).use(logMetrics(metrics))

package-lock.json

@@ -117,6 +117,7 @@
   "dependencies": {
     "@aws-lambda-powertools/metrics": "1.9.0",
     "@hello.nrfcloud.com/proto": "1.0.0",
+    "@middy/core": "4.5.2",
     "@nordicsemiconductor/from-env": "2.0.0",
     "@nordicsemiconductor/timestream-helpers": "5.0.0",
     "@sinclair/typebox": "0.28.15",