pm2-perfmonitor 1.1.2 → 1.2.1

package/README.md

@@ -17,15 +17,16 @@ $ pm2 install pm2-perfmonitor
 
 # Configure
 
-|            Property             | Default Value |                               Description                                |
-| :-----------------------------: | :-----------: | :----------------------------------------------------------------------: |
-|            `enabled`            |    `true`     |                  Specify whether to enable this module                   |
-|          `excludeApps`          |       -       | Specify the application name that needs to be excluded from guardianship |
-|          `includeApps`          |       -       |          Specify the application name that needs to be guarded           |
-|        `workerInterval`         |    `60000`    |                    Timed task execution interval (ms)                    |
-|        `zombieDetection`        |    `true`     |           Specify whether to enable zombie process protection            |
-|         `zombieMaxHits`         |     `10`      |        Specify the maximum occurrence frequency of zombie status         |
-| `autoRestartWhenZombieDetected` |    `true`     |        Specify whether to automatically restart zombie processes         |
+|            Property             | Default Value |                                     Description                                      |
+| :-----------------------------: | :-----------: | :----------------------------------------------------------------------------------: |
+|            `enabled`            |    `true`     |                        Specify whether to enable this module                         |
+|          `excludeApps`          |       -       |       Specify the application name that needs to be excluded from guardianship       |
+|          `includeApps`          |       -       |                Specify the application name that needs to be guarded                 |
+|        `workerInterval`         |    `60000`    |                          Timed task execution interval (ms)                          |
+|        `zombieDetection`        |    `true`     |                 Specify whether to enable zombie process protection                  |
+|         `zombieMaxHits`         |     `10`      |              Specify the maximum occurrence frequency of zombie status               |
+| `autoRestartWhenZombieDetected` |    `true`     |              Specify whether to automatically restart zombie processes               |
+|       `zombieMaxRestarts`       |      `0`      | The maximum number of zombie process restarts can be set to `0` to indicate no limit |
 
 # How to set these values ?
 

package/lib/app.js

@@ -1,5 +1,6 @@
 const pmx = require('pmx')
 const pm2 = require('pm2')
+const { getPm2ListAsync } = require('./pm2-extra')
 const { parseParamToArray, parseParamToNumber, parseBool } = require('./utils')
 
 const defaultOptions = {
@@ -28,6 +29,10 @@ const defaultOptions = {
    * 僵尸状态达到最大容忍度时，是否自动重启僵尸进程
    */
   autoRestartWhenZombieDetected: true,
+  /**
+   * 僵尸进程最大重启次数，设置为0表示不限制
+   */
+  zombieMaxRestarts: 0,
 }
 
 const conf = pmx.initModule({}, (err, incomingConf) => {
@@ -53,6 +58,7 @@ const AUTO_RESTART_WHEN_ZOMBIE_DETECTED = parseBool(
   conf.autoRestartWhenZombieDetected,
 )
 const ZOMBIE_MAX_HITS = parseParamToNumber(conf.zombieMaxHits)
+const ZOMBIE_MAX_RESTARTS = parseParamToNumber(conf.zombieMaxRestarts)
 
 // 存储每个进程的 CPU 采样历史（pm_id -> [cpu1, cpu2, ...]）
 const cpuHistory = new Map()
@@ -69,6 +75,7 @@ const logger = (type, ...args) => {
 
 /**
  * 判断是否为僵尸进程：最近 ZOMBIE_MAX_HITS 次全是 0%
+ * @param { number[] } history
  */
 const isZombie = (history) => {
   return history.length >= ZOMBIE_MAX_HITS && history.every((v) => v === 0)
@@ -77,10 +84,12 @@ const isZombie = (history) => {
 /**
  * check zombie process
  */
-const zombieProcessChecker = () => {
+const zombieProcessChecker = async () => {
   if (!ZOMBIE_DETECTION) return
 
-  pm2.list((err, apps) => {
+  try {
+    const apps = await getPm2ListAsync()
+
     apps.forEach((app) => {
       const { name, pm_id, monit, pm2_env } = app
 
@@ -121,6 +130,13 @@ const zombieProcessChecker = () => {
         logger('info', `Zombie detected: ${name} (pm_id: ${pm_id})`)
 
         if (AUTO_RESTART_WHEN_ZOMBIE_DETECTED) {
+          if (
+            ZOMBIE_MAX_RESTARTS > 0 &&
+            zombieRestartHistory.get(pm_id) >= ZOMBIE_MAX_RESTARTS
+          ) {
+            return
+          }
+
           logger('info', 'restarting...')
 
           pm2.restart(pm_id, (restartErr) => {
@@ -162,7 +178,9 @@ const zombieProcessChecker = () => {
         }
       }
     })
-  })
+  } catch (err) {
+    logger('error', err)
+  }
 }
 
 const runModule = () => {

package/lib/pm2-extra.js

@@ -0,0 +1,37 @@
+const pm2 = require('pm2')
+
+/**
+ * @returns { Promise<pm2.ProcessDescription[]> }
+ */
+const getPm2ListAsync = () => {
+  return new Promise((resolve, reject) => {
+    pm2.list((err, apps) => {
+      if (err) {
+        return reject(err)
+      }
+
+      resolve(apps)
+    })
+  })
+}
+
+/**
+ * @param { string | number} pm_id
+ * @returns { Promise<void> }
+ */
+const pm2StopAsync = (pm_id) => {
+  return new Promise((resolve, reject) => {
+    pm2.stop(pm_id, (err) => {
+      if (err) {
+        return reject(err)
+      }
+
+      resolve()
+    })
+  })
+}
+
+module.exports = {
+  getPm2ListAsync,
+  pm2StopAsync,
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pm2-perfmonitor",
-  "version": "1.1.2",
+  "version": "1.2.1",
   "description": "A pm2 module for performance monitoring. Automatically detect zombie processes and restart it",
   "author": {
     "name": "elenh",