add setGC

cenfun · Oct 6, 2024 · e562723 · e562723
1 parent d1e66e9
commit e562723
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 3 deletions.
diff --git a/lib/index.js b/lib/index.js
@@ -121,6 +121,10 @@ class CoverageReport {
         return results;
     }
 
+    setGC(value) {
+        Util.setGC(value);
+    }
+
     // add coverage from dir
     async addFromDir(dir) {
         const time_start = Date.now();

diff --git a/lib/utils/util.js b/lib/utils/util.js
@@ -869,18 +869,36 @@ const Util = {
         console.log(`[MCR] ${message} before ${lengthBefore} => after ${lengthAfter}`);
     },
 
+    setGC: (value) => {
+        if (!Util.isNum(value)) {
+            value = 1024;
+        }
+        Util.gcValue = value;
+    },
+
+    getMemory: () => {
+        const { heapUsed } = process.memoryUsage();
+        const memory = parseFloat((heapUsed / 1024 ** 2).toFixed(1));
+        return memory;
+    },
+
     // time is debug level, fix the log info, for checking snapshot
     logTime: (message, time_start) => {
+
+        // memory gc
+        let memory = Util.getMemory();
+        if (Util.gcValue && Util.gcValue < memory) {
+            Util.forceGC();
+        }
+
         if (Util.loggingLevel < Util.loggingLevels.debug && !process.env.MCR_LOG_TIME) {
             return;
         }
         const duration = Date.now() - time_start;
         const durationH = Util.TF(duration);
         const ls = [`[MCR] ${message}`];
 
-        // memory
-        const { heapUsed } = process.memoryUsage();
-        const memory = parseFloat((heapUsed / 1024 ** 2).toFixed(1));
+        memory = Util.getMemory();
         ls.push(` (memory: ${memory}MB)`);
 
         // time