rotor: increase isolate-vm pool

jitsucom · Dec 15, 2023 · 48bcee3 · 48bcee3 · vercel · Dec 15, 2023
1 parent 834739c
commit 48bcee3
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 13 deletions.
diff --git a/libs/core-functions/src/functions/lib/mongodb.ts b/libs/core-functions/src/functions/lib/mongodb.ts
@@ -17,7 +17,7 @@ async function createClient() {
   log.atInfo().log(`Connecting to MongoDB server...`);
 
   // Create a new MongoClient
-  const client = new MongoClient(mongodbURL);
+  const client = new MongoClient(mongodbURL, { readPreference: "nearest" });
   // Connect the client to the server (optional starting in v4.7)
   await client.connect();
   // Establish and verify connection

diff --git a/libs/core-functions/src/functions/lib/store.ts b/libs/core-functions/src/functions/lib/store.ts
@@ -87,29 +87,41 @@ export const createMongoStore = (namespace: string, mongo: MongoClient, defaultT
 
   return {
     get: async (key: string) => {
-      const res = await mongo.db(dbName).collection<StoreValue>(namespace).findOne({ _id: key });
+      const res = await mongo
+        .db(dbName)
+        .collection<StoreValue>(namespace)
+        .findOne({ _id: key }, { readPreference: "nearest" });
       return res ? res.value : undefined;
     },
     set: async (key: string, obj: any, opts?: SetOpts) => {
       await ensureCollection();
       const ttl = getTtlSec(opts);
       const expireAt = new Date();
       expireAt.setSeconds(expireAt.getSeconds() + ttl);
-      await mongo.db(dbName).collection<StoreValue>(namespace).replaceOne(
-        { _id: key },
-        {
-          value: obj,
-          expireAt,
-        },
-        { upsert: true }
-      );
+      await mongo
+        .db(dbName)
+        .collection<StoreValue>(namespace)
+        .replaceOne(
+          { _id: key },
+          {
+            value: obj,
+            expireAt,
+          },
+          { upsert: true, writeConcern: { w: 1, journal: false } }
+        );
     },
     del: async (key: string) => {
       await ensureCollection();
-      await mongo.db(dbName).collection<StoreValue>(namespace).deleteOne({ _id: key });
+      await mongo
+        .db(dbName)
+        .collection<StoreValue>(namespace)
+        .deleteOne({ _id: key }, { writeConcern: { w: 1, journal: false } });
     },
     ttl: async (key: string) => {
-      const res = await mongo.db(dbName).collection<StoreValue>(namespace).findOne({ _id: key });
+      const res = await mongo
+        .db(dbName)
+        .collection<StoreValue>(namespace)
+        .findOne({ _id: key }, { readPreference: "nearest" });
       return res ? Math.max(Math.floor((res.expireAt.getTime() - new Date().getTime()) / 1000), 0) : -2;
     },
   };

diff --git a/libs/core-functions/src/functions/udf_wrapper.ts b/libs/core-functions/src/functions/udf_wrapper.ts
@@ -29,7 +29,7 @@ export const UDFWrapper = (functionId, name, functionCode: string): UDFWrapperRe
   let isolate: Isolate;
   try {
     //const wrappedCode = `let exports = {}\n${functionCode}\n${wrapperJs}`;
-    isolate = new Isolate({ memoryLimit: 10 });
+    isolate = new Isolate({ memoryLimit: 128 });
     const context = isolate.createContextSync();
     const jail = context.global;