accept script in base64 encoding

Author: akashjain971

ecosystem.config.js

@@ -1,15 +1,22 @@
 module.exports = {
-    apps : [{
-        name: "dgraph-lambda",
-        script: "./dist/index.js",
-        instances: 4,
-        watch: ["script.js"],
-        exec_mode  : "cluster",
-        env: {
-            NODE_ENV: "development",
-        },
-        env_production: {
-            NODE_ENV: "production",
-        }
-    }]
-}
+  apps: [
+    {
+      name: "dgraph-lambda",
+      script: "./dist/index.js",
+      instances: 4,
+      exp_backoff_restart_delay: 100,
+      max_memory_restart: "32M",
+      watch: ["./script/script.js"],
+      watch_options: {
+        followSymlinks: false,
+      },
+      exec_mode: "cluster",
+      env: {
+        NODE_ENV: "development",
+      },
+      env_production: {
+        NODE_ENV: "production",
+      },
+    },
+  ],
+};

script.js

@@ -0,0 +1,24 @@
+const fullName = ({ parent: { firstName, lastName } }) =>
+  `${firstName} ${lastName}`;
+
+async function todoTitles({ graphql }) {
+  const results = await graphql("{ queryTodo { title } }");
+  return results.data.queryTodo.map((t) => t.title);
+}
+
+self.addGraphQLResolvers({
+  "User.fullName": fullName,
+  "Query.todoTitles": todoTitles,
+});
+
+async function reallyComplexDql({ parents, dql }) {
+  const ids = parents.map((p) => p.id);
+  const someComplexResults = await dql.query(
+    `really-complex-query-here with ${ids}`
+  );
+  return parents.map((parent) => someComplexResults[parent.id]);
+}
+
+self.addMultiParentGraphQLResolvers({
+  "User.reallyComplexProperty": reallyComplexDql,
+});

script/script.js

@@ -1,14 +1,37 @@
-import cluster from 'cluster';
-import fs from 'fs'
-import { scriptToExpress } from './script-to-express';
+import cluster from "cluster";
+import fs from "fs";
+import { scriptToExpress } from "./script-to-express";
+import atob from "atob";
+import btoa from "btoa";
+
+function base64Decode(str: string) {
+  try {
+    const original = str.trim();
+    const decoded = atob(original);
+    return btoa(decoded) === original ? decoded : "";
+  } catch (err) {
+    console.error(err);
+    return "";
+  }
+}
 
 async function startServer() {
-  const source = (await fs.promises.readFile(process.env.SCRIPT_PATH || "./script.js")).toString()
-  const app = scriptToExpress(source);
+  const source = (
+    await fs.promises.readFile(process.env.SCRIPT_PATH || "./script/script.js")
+  ).toString();
+  const script = base64Decode(source) || source;
+
+  const app = scriptToExpress(script);
   const port = process.env.PORT || "8686";
-  const server = app.listen(port, () => console.log("Server Listening on port " + port + "!"))
-  cluster.on('disconnect', () => server.close())
+  const server = app.listen(port, () =>
+    console.log("Server Listening on port " + port + "!")
+  );
+  cluster.on("disconnect", () => server.close());
+
+  process.on("SIGINT", () => {
+    server.close();
+    process.exit(0);
+  });
 }
 
 startServer();
-