@intelligems/sst 2.49.6-ig.4 → 2.49.6-ig.6

package/support/nodejs-runtime/index.mjs

@@ -6,9 +6,18 @@ import path from "path";
 import fs from "fs";
 import http from "http";
 import url from "url";
+import os from "os";
+try {
+  const mod = await import("node:module");
+  if (typeof mod.enableCompileCache === "function") {
+    const cacheDir = path.join(os.tmpdir(), "sst-compile-cache");
+    mod.enableCompileCache(cacheDir);
+  }
+} catch {
+}
 var input = workerData;
 var monoBundlePath = path.join(input.out, "index.mjs");
-var useMonoBundle = input.handler === "index.handler" && fs.existsSync(monoBundlePath);
+var useMonoBundle = input.isMonoBuild ?? (input.handler === "index.handler" && fs.existsSync(monoBundlePath));
 var file;
 var handlerName;
 if (useMonoBundle) {
@@ -62,6 +71,9 @@ try {
       useMonoBundle ? `Mono-bundle handler "${handlerName}" not found in "${file}". Found: ${Object.keys(mod).join(", ")}` : `Function "${handlerName}" not found in "${input.handler}". Found: ${Object.keys(mod).join(", ")}`
     );
   }
+  if (useMonoBundle && mod.warmUp) {
+    await mod.warmUp();
+  }
 } catch (ex) {
   await fetch({
     path: `/runtime/init/error`,
@@ -128,9 +140,10 @@ while (true) {
       identity: JSON.parse(result.headers["lambda-runtime-cognito-identity"]) ?? void 0,
       // If clientContext is null, we want to mimick AWS behavior and return undefined
       clientContext: JSON.parse(result.headers["lambda-runtime-client-context"]) ?? void 0,
-      functionName: process.env.AWS_LAMBDA_FUNCTION_NAME,
-      functionVersion: process.env.AWS_LAMBDA_FUNCTION_VERSION,
-      memoryLimitInMB: process.env.AWS_LAMBDA_FUNCTION_MEMORY_SIZE,
+      // Per-invocation function context from headers (essential for mono-build shared workers)
+      functionName: result.headers["lambda-runtime-function-name"] || process.env.AWS_LAMBDA_FUNCTION_NAME,
+      functionVersion: result.headers["lambda-runtime-function-version"] || process.env.AWS_LAMBDA_FUNCTION_VERSION,
+      memoryLimitInMB: result.headers["lambda-runtime-function-memory-size"] || process.env.AWS_LAMBDA_FUNCTION_MEMORY_SIZE,
       logGroupName: result.headers["lambda-runtime-log-group-name"],
       logStreamName: result.headers["lambda-runtime-log-stream-name"],
       callbackWaitsForEmptyEventLoop: {

package/support/python-runtime/runtime.py

@@ -130,33 +130,33 @@ if __name__ == '__main__':
             if hasattr(signal, 'SIGALRM'):
                 signal.alarm(0)
 
-            event = json.loads(r.read())
-            context = Context(
-                r.getheader('Lambda-Runtime-Invoked-Function-Arn'),
-                r.getheader('Lambda-Runtime-Aws-Request-Id'),
-                r.getheader('Lambda-Runtime-Deadline-Ms'),
-                r.getheader('Lambda-Runtime-Cognito-Identity'),
-                r.getheader('Lambda-Runtime-Client-Context'),
-                r.getheader('Lambda-Runtime-Log-Group-Name'),
-                r.getheader('Lambda-Runtime-Log-Stream-Name')
-            )
+    event = json.loads(r.read())
+    context = Context(
+        r.getheader('Lambda-Runtime-Invoked-Function-Arn'),
+        r.getheader('Lambda-Runtime-Aws-Request-Id'),
+        r.getheader('Lambda-Runtime-Deadline-Ms'),
+        r.getheader('Lambda-Runtime-Cognito-Identity'),
+        r.getheader('Lambda-Runtime-Client-Context'),
+        r.getheader('Lambda-Runtime-Log-Group-Name'),
+        r.getheader('Lambda-Runtime-Log-Stream-Name')
+    )
 
             # Invoke handler
-            try:
-                result = handler(event, context)
-                data = json.dumps(result, default=handleUnserializable).encode("utf-8")
+    try:
+        result = handler(event, context)
+        data = json.dumps(result, default=handleUnserializable).encode("utf-8")
                 url_destination = '/response'
-            except Exception as e:
+    except Exception as e:
                 # Handler error - report but keep worker alive
-                traceback.print_exc()
-                ex_type, ex_value, ex_traceback = sys.exc_info()
+        traceback.print_exc()
+        ex_type, ex_value, ex_traceback = sys.exc_info()
                 error_result = {
                     "errorType": ex_type.__name__ if ex_type else "Error",
-                    "errorMessage": str(ex_value),
+            "errorMessage": str(ex_value),
                     "trace": traceback.format_tb(ex_traceback) if ex_traceback else [],
-                }
+        }
                 data = json.dumps(error_result).encode("utf-8")
-                url_destination = '/error'
+        url_destination = '/error'
 
             # Send response
             response_url = "http://{}/2018-06-01/runtime/invocation/{}{}".format(
@@ -165,7 +165,7 @@ if __name__ == '__main__':
                 url_destination
             )
             req = request.Request(response_url, method="POST", data=data)
-            req.add_header('Content-Type', 'application/json')
+    req.add_header('Content-Type', 'application/json')
 
             # Retry sending response (matching Node.js behavior)
             max_retries = 3