fix(middleware-recursion-detection): handle case insensitive matching of X-Amzn-Trace-Id header

Author: smilkuri

packages/middleware-recursion-detection/src/index.spec.ts

@@ -70,6 +70,27 @@ describe(recursionDetectionMiddleware.name, () => {
     expect(request.headers[TRACE_ID_HEADER_NAME]).toBe("some-real-trace-id");
   });
 
+  it(`should NOT set ${TRACE_ID_HEADER_NAME} header when the header is already set in all lowercase`, async () => {
+    process.env = {
+      AWS_LAMBDA_FUNCTION_NAME: "some-function",
+      _X_AMZN_TRACE_ID: "some-trace-id",
+    };
+    const handler = recursionDetectionMiddleware({ runtime: "node" })(mockNextHandler, {} as any);
+    await handler({
+      input: {},
+      request: new HttpRequest({
+        headers: {
+          [TRACE_ID_HEADER_NAME.toLowerCase()]: "some-real-trace-id",
+        },
+      }),
+    });
+
+    const { calls } = (mockNextHandler as any).mock;
+    expect(calls.length).toBe(1);
+    const { request } = mockNextHandler.mock.calls[0][0];
+    expect(request.headers[TRACE_ID_HEADER_NAME.toLowerCase()]).toBe("some-real-trace-id");
+  });
+
   it("has no effect for browser runtime", async () => {
     process.env = {
       AWS_LAMBDA_FUNCTION_NAME: "some-function",

packages/middleware-recursion-detection/src/index.ts

@@ -27,14 +27,16 @@ export const recursionDetectionMiddleware =
   <Output extends MetadataBearer>(next: BuildHandler<any, Output>): BuildHandler<any, Output> =>
   async (args: BuildHandlerArguments<any>): Promise<BuildHandlerOutput<Output>> => {
     const { request } = args;
-    if (
-      !HttpRequest.isInstance(request) ||
-      options.runtime !== "node" ||
-      request.headers.hasOwnProperty(TRACE_ID_HEADER_NAME)
-    ) {
+    if (!HttpRequest.isInstance(request) || options.runtime !== "node") {
       return next(args);
     }
+    const traceIdHeader =
+      Object.keys(request.headers ?? {}).find((h) => h.toLowerCase() === TRACE_ID_HEADER_NAME.toLowerCase()) ??
+      TRACE_ID_HEADER_NAME;
 
+    if (request.headers.hasOwnProperty(traceIdHeader)) {
+      return next(args);
+    }
     const functionName = process.env[ENV_LAMBDA_FUNCTION_NAME];
     const traceId = process.env[ENV_TRACE_ID];
     const nonEmptyString = (str: unknown): str is string => typeof str === "string" && str.length > 0;