UnixSignalService

Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64
/*----------------------------------------------------------------------
 
 UnixSignalService: At present, this defines a SIGUSR2 handler and
 sets the shutdown flag when that signal has been raised.
 
 This service is instantiated at job startup.
 
 ----------------------------------------------------------------------*/
#include <iostream>
#include <cstdlib>

#include "FWCore/ServiceRegistry/interface/ServiceMaker.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "FWCore/Utilities/interface/UnixSignalHandlers.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"

namespace edm {
  namespace service {
    class UnixSignalService {
    public:
      explicit UnixSignalService(ParameterSet const& ps);
      ~UnixSignalService();

      static void fillDescriptions(ConfigurationDescriptions& descriptions);

    private:
      bool enableSigInt_;
    };  // class UnixSignalService
  }  // end of namespace service
}  // end of namespace edm

namespace edm {

  namespace service {

    UnixSignalService::UnixSignalService(ParameterSet const& pset)
        : enableSigInt_(pset.getUntrackedParameter<bool>("EnableCtrlC")) {
      installCustomHandler(SIGUSR2, ep_sigusr2);
      if (enableSigInt_)
        installCustomHandler(SIGINT, ep_sigusr2);
    }

    UnixSignalService::~UnixSignalService() {}

    void UnixSignalService::fillDescriptions(ConfigurationDescriptions& descriptions) {
      ParameterSetDescription desc;
      desc.addUntracked<bool>("EnableCtrlC", true)
          ->setComment(
              "If 'true', you can stop a cmsRun job gracefully by sending it a '<control> c' keyboard interrupt (i.e. "
              "SIGINT).");
      descriptions.add("UnixSignalService", desc);
      descriptions.setComment(
          "This service sets up unix signal handlers for the unix signal SIGUSR2 and optionally SIGINT"
          " so that when cmsRun is sent a signal the application will stop processing and shut down gracefully.");
    }
  }  // end of namespace service
}  // end of namespace edm

using edm::service::UnixSignalService;
typedef edm::serviceregistry::ParameterSetMaker<UnixSignalService> UnixSignalMaker;
DEFINE_FWK_SERVICE_MAKER(UnixSignalService, UnixSignalMaker);