feat(logger): add context for low prio logs

Author: zirkelc

packages/logger/src/Logger.ts

@@ -139,6 +139,11 @@ class Logger extends Utility implements LoggerInterface {
    * immediately because the logger is not ready yet. This buffer stores those logs until the logger is ready.
    */
   #buffer: [number, Parameters<Logger['createAndPopulateLogItem']>][] = [];
+
+
+  #context: Record<string, Array<Parameters<Logger['createAndPopulateLogItem']>>> = {}
+
+
   /**
    * Flag used to determine if the logger is initialized.
    */
@@ -183,6 +188,7 @@ class Logger extends Utility implements LoggerInterface {
       this.printLog(level, this.createAndPopulateLogItem(...log));
     }
     this.#buffer = [];
+    this.#context = {};
   }
 
   /**
@@ -868,6 +874,22 @@ class Logger extends Utility implements LoggerInterface {
     extraInput: LogItemExtraInput
   ): void {
     if (logLevel >= this.logLevel) {
+      const xRayTraceId = this.envVarsService.getXrayTraceId();
+      
+      // Print all log items in the context
+      if (this.#context[xRayTraceId]) {
+        for (const contextItem of this.#context[xRayTraceId]) {
+          this.printLog(
+            logLevel,
+            this.createAndPopulateLogItem(...contextItem)
+          );
+        }
+        
+        // Clear the context after flushing
+        // This also removes entries from other X-Ray trace IDs
+        this.#context = {};
+      }
+
       if (this.#isInitialized) {
         this.printLog(
           logLevel,
@@ -876,6 +898,18 @@ class Logger extends Utility implements LoggerInterface {
       } else {
         this.#buffer.push([logLevel, [logLevel, input, extraInput]]);
       }
+    } else {
+      const xRayTraceId = this.envVarsService.getXrayTraceId() as string;
+
+      // Add the log item to the context
+      const context = this.#context[xRayTraceId] ?? [];
+      context.push([logLevel, input, extraInput]);
+
+      // Assign the updated context to the context property 
+      // This also removes other X-Ray trace IDs from the context
+      this.#context = {
+        [xRayTraceId]: context,
+      };
     }
   }