actionhero/initializers/pids.js

actionhero.js is a multi-transport API Server with integrated cluster capabilities and delayed tasks

'use strict';

var fs = require('fs');
var cluster = require('cluster');

module.exports = {
  startPriority: 1,
  loadPriority:  110,
  initialize: function(api, next){

    api.pids = {};
    api.pids.pid = process.pid;
    api.pids.path = api.config.general.paths.pid[0]; // it would be silly to have more than one pid

    api.pids.sanitizeId = function(){
      var pidfile = api.id;
      pidfile = pidfile.replace(new RegExp(':', 'g'), '-');
      pidfile = pidfile.replace(new RegExp(' ', 'g'), '_');
      pidfile = pidfile.replace(new RegExp('\r', 'g'), '');
      pidfile = pidfile.replace(new RegExp('\n', 'g'), '');

      return pidfile;
    };

    if(cluster.isMaster){
      api.pids.title = 'actionhero-' + api.pids.sanitizeId();
    }else{
      api.pids.title = api.pids.sanitizeId();
    }

    try{ fs.mkdirSync(api.pids.path); }catch(e){};

    api.pids.writePidFile = function(){
      fs.writeFileSync(api.pids.path + '/' + api.pids.title, api.pids.pid.toString(), 'ascii');
    };

    api.pids.clearPidFile = function(){
      try{
        fs.unlinkSync(api.pids.path + '/' + api.pids.title);
      }catch(e){
        api.log('Unable to remove pidfile', 'error', e);
      }
    };

    next();
  },

  start: function(api, next){
    api.pids.writePidFile();
    api.log(['pid: %s', process.pid], 'notice');
    next();
  }
};