@teamkeel/functions-runtime 0.393.2 → 0.393.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@teamkeel/functions-runtime",
-  "version": "0.393.2",
+  "version": "0.393.3",
   "description": "Internal package used by @teamkeel/sdk",
   "main": "src/index.js",
   "scripts": {

package/src/database.js

@@ -43,10 +43,11 @@ async function withDatabase(db, actionType, cb) {
   });
 }
 
-let db = null;
-
 const dbInstance = new AsyncLocalStorage();
 
+// used to establish a singleton for our vitest environment
+let vitestDb = null;
+
 // useDatabase will retrieve the database client set by withDatabase from the local storage
 function useDatabase() {
   // retrieve the instance of the database client from the store which is aware of
@@ -61,7 +62,10 @@ function useDatabase() {
   // which covers any test files ending in *.test.ts. Custom function code runs in a different node process which will not have this environment variable. Tests written using our testing
   // framework call actions (and in turn custom function code) over http using the ActionExecutor class
   if ("NODE_ENV" in process.env && process.env.NODE_ENV == "test") {
-    return getDatabaseClient();
+    if (!vitestDb) {
+      vitestDb = createDatabaseClient();
+    }
+    return vitestDb;
   }
 
   // If we've gotten to this point, then we know that we are in a custom function runtime server
@@ -69,17 +73,11 @@ function useDatabase() {
   throw new Error("useDatabase must be called within a function");
 }
 
-// getDatabaseClient will return a brand new instance of Kysely. Every instance of Kysely
+// createDatabaseClient will return a brand new instance of Kysely. Every instance of Kysely
 // represents an individual connection to the database.
 // not to be exported externally from our sdk - consumers should use useDatabase
-function getDatabaseClient() {
-  // 'db' represents the singleton connection to the database which is stored
-  // as a module scope variable.
-  if (db) {
-    return db;
-  }
-
-  db = new Kysely({
+function createDatabaseClient() {
+  const db = new Kysely({
     dialect: getDialect(),
     plugins: [
       // ensures that the audit context data is written to Postgres configuration parameters
@@ -109,7 +107,6 @@ class InstrumentedPool extends pg.Pool {
     const _super = super.connect.bind(this);
     return withSpan("Database Connect", function (span) {
       span.setAttribute("dialect", process.env["KEEL_DB_CONN_TYPE"]);
-      span.setAttribute("timeout", connectionTimeout());
       return _super(...args);
     });
   }
@@ -120,7 +117,6 @@ class InstrumentedNeonServerlessPool extends neonserverless.Pool {
     const _super = super.connect.bind(this);
     return withSpan("Database Connect", function (span) {
       span.setAttribute("dialect", process.env["KEEL_DB_CONN_TYPE"]);
-      span.setAttribute("timeout", connectionTimeout());
       return _super(...args);
     });
   }
@@ -179,7 +175,7 @@ function getDialect() {
           // Although I doubt we will run into these freeze/thaw issues if idleTimeoutMillis is always shorter than the
           // time is takes for a lambda to freeze (which is not a constant, but could be as short as several minutes,
           // https://www.pluralsight.com/resources/blog/cloud/how-long-does-aws-lambda-keep-your-idle-functions-around-before-a-cold-start)
-          idleTimeoutMillis: connectionTimeout(),
+          idleTimeoutMillis: 50000,
           connectionString: mustEnv("KEEL_DB_CONN"),
         }),
       });
@@ -187,7 +183,6 @@ function getDialect() {
       neonserverless.neonConfig.webSocketConstructor = ws;
 
       const pool = new InstrumentedNeonServerlessPool({
-        idleTimeoutMillis: connectionTimeout(),
         connectionString: mustEnv("KEEL_DB_CONN"),
       });
 
@@ -229,17 +224,6 @@ function mustEnv(key) {
   return v;
 }
 
-function connectionTimeout() {
-  const v = Number(process.env["KEEL_DB_CONN_TIMEOUT"]);
-  if (!v || isNaN(v)) {
-    return 45000; // 60s is our current neon suspend default
-  }
-  return v;
-}
-
-// initialise the database client at module scope level so the db variable is set
-getDatabaseClient();
-
-module.exports.getDatabaseClient = getDatabaseClient;
+module.exports.createDatabaseClient = createDatabaseClient;
 module.exports.useDatabase = useDatabase;
 module.exports.withDatabase = withDatabase;

package/src/handleJob.js

@@ -3,7 +3,7 @@ const {
   createJSONRPCSuccessResponse,
   JSONRPCErrorCode,
 } = require("json-rpc-2.0");
-const { getDatabaseClient } = require("./database");
+const { createDatabaseClient } = require("./database");
 const { errorToJSONRPCResponse, RuntimeErrors } = require("./errors");
 const opentelemetry = require("@opentelemetry/api");
 const { withSpan } = require("./tracing");
@@ -25,6 +25,8 @@ async function handleJob(request, config) {
   return opentelemetry.context.with(activeContext, () => {
     // Wrapping span for the whole request
     return withSpan(request.method, async (span) => {
+      let db = null;
+
       try {
         const { createJobContextAPI, jobs } = config;
 
@@ -51,7 +53,7 @@ async function handleJob(request, config) {
             ? true
             : null;
 
-        const db = getDatabaseClient();
+        db = createDatabaseClient();
         const jobFunction = jobs[request.method];
         const actionType = PROTO_ACTION_TYPES.JOB;
 
@@ -89,6 +91,10 @@ async function handleJob(request, config) {
           RuntimeErrors.UnknownError,
           message
         );
+      } finally {
+        if (db) {
+          await db.destroy();
+        }
       }
     });
   });

package/src/handleRequest.js

@@ -3,7 +3,7 @@ const {
   createJSONRPCSuccessResponse,
   JSONRPCErrorCode,
 } = require("json-rpc-2.0");
-const { getDatabaseClient } = require("./database");
+const { createDatabaseClient } = require("./database");
 const { tryExecuteFunction } = require("./tryExecuteFunction");
 const { errorToJSONRPCResponse, RuntimeErrors } = require("./errors");
 const opentelemetry = require("@opentelemetry/api");
@@ -28,6 +28,8 @@ async function handleRequest(request, config) {
   return opentelemetry.context.with(activeContext, () => {
     // Wrapping span for the whole request
     return withSpan(request.method, async (span) => {
+      let db = null;
+
       try {
         const { createContextAPI, functions, permissionFns, actionTypes } =
           config;
@@ -62,7 +64,7 @@ async function handleRequest(request, config) {
             ? true
             : null;
 
-        const db = getDatabaseClient();
+        db = createDatabaseClient();
         const customFunction = functions[request.method];
         const actionType = actionTypes[request.method];
 
@@ -122,6 +124,10 @@ async function handleRequest(request, config) {
           RuntimeErrors.UnknownError,
           message
         );
+      } finally {
+        if (db) {
+          await db.destroy();
+        }
       }
     });
   });

package/src/handleSubscriber.js

@@ -3,7 +3,7 @@ const {
   createJSONRPCSuccessResponse,
   JSONRPCErrorCode,
 } = require("json-rpc-2.0");
-const { getDatabaseClient } = require("./database");
+const { createDatabaseClient } = require("./database");
 const { errorToJSONRPCResponse, RuntimeErrors } = require("./errors");
 const opentelemetry = require("@opentelemetry/api");
 const { withSpan } = require("./tracing");
@@ -24,6 +24,8 @@ async function handleSubscriber(request, config) {
   return opentelemetry.context.with(activeContext, () => {
     // Wrapping span for the whole request
     return withSpan(request.method, async (span) => {
+      let db = null;
+
       try {
         const { createSubscriberContextAPI, subscribers } = config;
 
@@ -45,7 +47,7 @@ async function handleSubscriber(request, config) {
           meta: request.meta,
         });
 
-        const db = getDatabaseClient();
+        db = createDatabaseClient();
         const subscriberFunction = subscribers[request.method];
         const actionType = PROTO_ACTION_TYPES.SUBSCRIBER;
 
@@ -77,6 +79,10 @@ async function handleSubscriber(request, config) {
           RuntimeErrors.UnknownError,
           message
         );
+      } finally {
+        if (db) {
+          await db.destroy();
+        }
       }
     });
   });