SIGN IN SIGN UP
Unitech / pm2 UNCLAIMED

Node.js Production Process Manager with a built-in Load Balancer.

43013 0 18 JavaScript
2016-03-22 23:33:43 +01:00
/**
2022-01-19 11:41:33 +01:00
* Copyright 2013-2022 the PM2 project authors. All rights reserved.
2016-03-22 23:33:43 +01:00
* Use of this source code is governed by a license that
* can be found in the LICENSE file.
*/
2022-02-09 10:19:31 +01:00
const eachLimit = require('async/eachLimit');
const debug = require('debug')('pm2:worker');
const domain = require('domain');
const Cron = require('croner');
const pkg = require('../package.json');
var cst = require('../constants.js');
2019-12-10 16:42:31 +01:00
var vCheck = require('./VersionCheck.js')
2014-11-10 14:45:41 +01:00
module.exports = function(God) {
var timer = null;
God.CronJobs = new Map();
2014-11-10 14:45:41 +01:00
God.Worker = {};
2014-11-18 18:14:34 +01:00
God.Worker.is_running = false;
God.getCronID = function(pm_id) {
return `cron-${pm_id}`
}
2019-11-14 23:37:35 +01:00
God.registerCron = function(pm2_env) {
if (!pm2_env ||
pm2_env.pm_id === undefined ||
!pm2_env.cron_restart ||
2021-09-24 12:40:46 +02:00
pm2_env.cron_restart == '0' ||
God.CronJobs.has(God.getCronID(pm2_env.pm_id)))
2018-10-06 10:52:18 +02:00
return;
var pm_id = pm2_env.pm_id
console.log('[PM2][WORKER] Registering a cron job on:', pm_id);
2018-10-06 10:52:18 +02:00
2022-02-09 10:19:31 +01:00
var job = Cron(pm2_env.cron_restart, function() {
2021-09-24 22:57:02 +02:00
God.restartProcessId({id: pm_id}, function(err, data) {
if (err)
console.error(err.stack || err);
return;
});
2018-10-06 10:52:18 +02:00
});
God.CronJobs.set(God.getCronID(pm_id), job);
2019-11-14 23:37:35 +01:00
}
/**
* Deletes the cron job on deletion of process
*/
God.deleteCron = function(id) {
if (typeof(id) !== 'undefined' && God.CronJobs.has(God.getCronID(id)) === false)
2019-11-14 23:37:35 +01:00
return;
console.log('[PM2] Deregistering a cron job on:', id);
var job = God.CronJobs.get(God.getCronID(id));
2022-02-08 10:14:16 +01:00
if (job)
2022-02-09 10:19:31 +01:00
job.stop();
2022-02-08 10:14:16 +01:00
God.CronJobs.delete(God.getCronID(id));
2019-11-14 23:37:35 +01:00
};
var _getProcessById = function(pm_id) {
var proc = God.clusters_db[pm_id];
return proc ? proc : null;
2018-10-06 10:52:18 +02:00
};
2014-11-18 18:14:34 +01:00
var maxMemoryRestart = function(proc_key, cb) {
var proc = _getProcessById(proc_key.pm2_env.pm_id);
2014-11-18 15:41:20 +01:00
2014-11-18 18:14:34 +01:00
if (!(proc &&
proc.pm2_env &&
proc_key.monit))
2014-11-18 18:14:34 +01:00
return cb();
if (proc_key.monit.memory !== undefined &&
proc.pm2_env.max_memory_restart !== undefined &&
2016-11-28 16:14:20 +01:00
proc.pm2_env.max_memory_restart < proc_key.monit.memory &&
2016-11-30 10:53:30 +01:00
proc.pm2_env.axm_options &&
2016-11-28 16:14:20 +01:00
proc.pm2_env.axm_options.pid === undefined) {
console.log('[PM2][WORKER] Process %s restarted because it exceeds --max-memory-restart value (current_memory=%s max_memory_limit=%s [octets])', proc.pm2_env.pm_id, proc_key.monit.memory, proc.pm2_env.max_memory_restart);
God.reloadProcessId({
id : proc.pm2_env.pm_id
}, function(err, data) {
2014-11-18 18:14:34 +01:00
if (err)
console.error(err.stack || err);
2014-11-18 15:41:20 +01:00
return cb();
2014-11-18 18:14:34 +01:00
});
}
else {
2014-11-18 18:14:34 +01:00
return cb();
}
2014-11-17 17:47:52 +01:00
};
2014-11-10 14:45:41 +01:00
var tasks = function() {
2014-11-18 18:14:34 +01:00
if (God.Worker.is_running === true) {
debug('[PM2][WORKER] Worker is already running, skipping this round');
return false;
}
God.Worker.is_running = true;
2014-11-16 13:20:14 +01:00
2014-11-18 15:41:20 +01:00
God.getMonitorData(null, function(err, data) {
2014-11-18 18:14:34 +01:00
if (err || !data || typeof(data) !== 'object') {
God.Worker.is_running = false;
return console.error(err);
}
2014-11-17 17:47:52 +01:00
eachLimit(data, 1, function(proc, next) {
if (!proc || !proc.pm2_env || proc.pm2_env.pm_id === undefined)
2014-11-18 18:14:34 +01:00
return next();
2014-11-18 15:41:20 +01:00
debug('[PM2][WORKER] Processing proc id:', proc.pm2_env.pm_id);
// Reset restart delay if application has an uptime of more > 30secs
if (proc.pm2_env.exp_backoff_restart_delay !== undefined &&
proc.pm2_env.prev_restart_delay && proc.pm2_env.prev_restart_delay > 0) {
var app_uptime = Date.now() - proc.pm2_env.pm_uptime
if (app_uptime > cst.EXP_BACKOFF_RESET_TIMER) {
var ref_proc = _getProcessById(proc.pm2_env.pm_id);
ref_proc.pm2_env.prev_restart_delay = 0
console.log(`[PM2][WORKER] Reset the restart delay, as app ${proc.name} has been up for more than ${cst.EXP_BACKOFF_RESET_TIMER}ms`)
}
}
// Check if application has reached memory threshold
maxMemoryRestart(proc, function() {
return next();
2014-11-18 15:41:20 +01:00
});
2014-11-18 18:14:34 +01:00
}, function(err) {
God.Worker.is_running = false;
debug('[PM2][WORKER] My job here is done, next job in %d seconds', parseInt(cst.WORKER_INTERVAL / 1000));
2014-11-18 15:41:20 +01:00
});
});
2014-11-10 14:45:41 +01:00
};
2015-06-05 17:50:30 +02:00
var wrappedTasks = function() {
var d = domain.create();
d.once('error', function(err) {
console.error('[PM2][WORKER] Error caught by domain:\n' + (err.stack || err));
God.Worker.is_running = false;
});
d.run(function() {
tasks();
});
};
2014-11-10 14:45:41 +01:00
God.Worker.start = function() {
2015-06-05 17:50:30 +02:00
timer = setInterval(wrappedTasks, cst.WORKER_INTERVAL);
2019-12-10 16:42:31 +01:00
if (!process.env.PM2_DISABLE_VERSION_CHECK) {
setInterval(() => {
vCheck({
state: 'check',
version: pkg.version,
});
}, 1000 * 60 * 60 * 24);
}
2014-11-10 14:45:41 +01:00
};
God.Worker.stop = function() {
if (timer !== null)
clearInterval(timer);
};
};