register lamba context

Author: stevebux

lambdas/upsert-letter/src/handler/upsert-handler.ts

@@ -1,4 +1,4 @@
-import { SQSBatchItemFailure, SQSEvent, SQSHandler } from "aws-lambda";
+import { SQSBatchItemFailure, SQSEvent, SQSHandler, Context } from "aws-lambda";
 import { $LetterRequestPreparedEvent } from "@nhsdigital/nhs-notify-event-schemas-letter-rendering-v1";
 import {
   InsertLetter,
@@ -25,6 +25,10 @@ import {
   UpsertOperation,
 } from "./schemas";
 
+const idempotencyConfig = new IdempotencyConfig({
+  eventKeyJmesPath: "id",
+});
+
 function getOperationFromType(type: string): UpsertOperation {
   if (
     type.startsWith("uk.nhs.notify.letter-rendering.letter-request.prepared")
@@ -187,7 +191,7 @@ function parseQueueMessage(queueMessage: string): QueueMessage {
 export default function createUpsertLetterHandler(deps: Deps): SQSHandler {
   const processRecordIdempotently = makeIdempotentOnId(processRecord, deps);
   return metricScope((metrics: MetricsLogger) => {
-    return async (event: SQSEvent) => {
+    return async (event: SQSEvent, context: Context) => {
       const batchItemFailures: SQSBatchItemFailure[] = [];
       const perSupplierSuccess: Map<string, number> = new Map<string, number>();
       const perSupplierFailure: Map<string, number> = new Map<string, number>();
@@ -222,6 +226,7 @@ export default function createUpsertLetterHandler(deps: Deps): SQSHandler {
             supplier: supplierSpec,
           });
 
+          idempotencyConfig.registerLambdaContext(context);
           supplier = await processRecordIdempotently(
             letterEvent,
             supplierSpec,
@@ -283,8 +288,6 @@ async function processRecord(
 function makeIdempotentOnId(fn: AnyFunction, deps: Deps) {
   return makeIdempotent(fn, {
     persistenceStore: deps.idempotencyLayer,
-    config: new IdempotencyConfig({
-      eventKeyJmesPath: "id",
-    }),
+    config: idempotencyConfig,
   });
 }