@sentio/runtime 2.58.0-rc.7 → 2.58.0-rc.8

package/lib/processor-runner.js

@@ -26493,15 +26493,7 @@ var httpServer = http2.createServer(async function(req, res) {
       case "/heap": {
         try {
           const file = "/tmp/" + Date.now() + ".heapsnapshot";
-          const session = new Session();
-          const fd = import_fs_extra2.default.openSync(file, "w");
-          session.connect();
-          session.on("HeapProfiler.addHeapSnapshotChunk", (m) => {
-            import_fs_extra2.default.writeSync(fd, m.params.chunk);
-          });
-          await session.post("HeapProfiler.takeHeapSnapshot");
-          session.disconnect();
-          import_fs_extra2.default.closeSync(fd);
+          await dumpHeap(file);
           const readStream = import_fs_extra2.default.createReadStream(file);
           res.writeHead(200, { "Content-Type": "application/json" });
           readStream.pipe(res);
@@ -26554,6 +26546,36 @@ process.on("SIGINT", function() {
     baseService.unhandled = reason;
   }
 });
+if (process.env["OOM_DUMP_MEMORY_SIZE_GB"]) {
+  let dumping = false;
+  const memorySize = parseInt(process.env["OOM_DUMP_MEMORY_SIZE_GB"], 10);
+  console.log("heap dumping is enabled, limit set to ", memorySize, "gb");
+  setInterval(async () => {
+    const mem = process.memoryUsage();
+    if (mem.heapTotal > memorySize * 1024 * 1024 * 1024 && !dumping) {
+      const file = "/tmp/" + Date.now() + ".heapsnapshot";
+      dumping = true;
+      await dumpHeap(file);
+      process.exit(11);
+    }
+  }, 1e3 * 60);
+}
+async function dumpHeap(file) {
+  console.log("Heap dumping to", file);
+  const session = new Session();
+  const fd = import_fs_extra2.default.openSync(file, "w");
+  try {
+    session.connect();
+    session.on("HeapProfiler.addHeapSnapshotChunk", (m) => {
+      import_fs_extra2.default.writeSync(fd, m.params.chunk);
+    });
+    await session.post("HeapProfiler.takeHeapSnapshot");
+    console.log("Heap dumped to", file);
+  } finally {
+    session.disconnect();
+    import_fs_extra2.default.closeSync(fd);
+  }
+}
 function shutdownServers(exitCode) {
   server?.forceShutdown();
   console.log("RPC server shut down");