feature: #3485 add cleardump command

Unitech · Mar 2, 2018 · ced2835 · ced2835
1 parent f2523f6
commit ced2835
Show file tree

Hide file tree

Showing 4 changed files with 36 additions and 3 deletions.
diff --git a/bin/pm2 b/bin/pm2
@@ -612,6 +612,15 @@ commander.command('dump')
     pm2.dump();
   }));
 
+//
+// Delete dump file
+//
+commander.command('cleardump')
+  .description('Create empty dump file')
+  .action(failOnUnknown(function() {
+    pm2.clearDump();
+  }));
+
 //
 // Save processes to file
 //

diff --git a/lib/API/Startup.js b/lib/API/Startup.js
@@ -404,7 +404,9 @@ module.exports = function(CLI) {
           console.error(e.stack || e);
           try {
             // try to backup file
-            fs.copyFileSync(cst.DUMP_BACKUP_FILE_PATH, cst.DUMP_FILE_PATH);
+            if(fs.existsSync(cst.DUMP_BACKUP_FILE_PATH)) {
+              fs.copyFileSync(cst.DUMP_BACKUP_FILE_PATH, cst.DUMP_FILE_PATH);
+            }
           } catch (e) {
             // don't keep broken file
             fs.unlinkSync(cst.DUMP_FILE_PATH);
@@ -431,6 +433,21 @@ module.exports = function(CLI) {
     });
   };
 
+  /**
+   * Remove DUMP_FILE_PATH file and DUMP_BACKUP_FILE_PATH file
+   * @method dump
+   * @param {} cb
+   * @return
+   */
+  CLI.prototype.clearDump = function(cb) {
+    fs.writeFileSync(cst.DUMP_FILE_PATH, JSON.stringify([]));
+
+    if(cb && typeof cb === 'function') return cb();
+
+    Common.printOut(cst.PREFIX_MSG + 'Successfully created %s', cst.DUMP_FILE_PATH);
+    return this.exitCli(cst.SUCCESS_EXIT);
+  };
+
   /**
    * Resurrect processes
    * @method resurrect

diff --git a/lib/God/ActionMethods.js b/lib/God/ActionMethods.js
@@ -162,7 +162,9 @@ module.exports = function(God) {
         console.error(e.stack || e);
         try {
           // try to backup file
-          fs.copyFileSync(cst.DUMP_BACKUP_FILE_PATH, cst.DUMP_FILE_PATH);
+          if(fs.existsSync(cst.DUMP_BACKUP_FILE_PATH)) {
+            fs.copyFileSync(cst.DUMP_BACKUP_FILE_PATH, cst.DUMP_FILE_PATH);
+          }
         } catch (e) {
           // don't keep broken file
           fs.unlinkSync(cst.DUMP_FILE_PATH);

diff --git a/test/programmatic/programmatic.js b/test/programmatic/programmatic.js
@@ -18,7 +18,12 @@ describe('PM2 programmatic calls', function() {
   });
 
   after(function(done) {
-    pm2.kill(done);
+    pm2.delete('all', function(err, ret) {
+      // clean dump file
+      pm2.clearDump(function(err) {
+        pm2.kill(done);
+      });
+    });
   });
 
   before(function(done) {