WorkerMonitorThread.h

Go to the documentation of this file.

#ifndef FWCore_SharedMemory_WorkerMonitorThread_h
#define FWCore_SharedMemory_WorkerMonitorThread_h
// -*- C++ -*-
//
// Package:     FWCore/SharedMemory
// Class  :     WorkerMonitorThread
//
//
// Original Author:  Chris Jones
//         Created:  21/01/2020
//

// system include files
#include <atomic>
#include <csignal>
#include <functional>
#include <thread>

// user include files
#include "FWCore/Utilities/interface/thread_safety_macros.h"

// forward declarations

namespace edm::shared_memory {
  class WorkerMonitorThread {
  public:
    WorkerMonitorThread() {}
    ~WorkerMonitorThread() {
      if (not stopRequested_.load()) {
        stop();
      }
    }
    WorkerMonitorThread(const WorkerMonitorThread&) = delete;
    const WorkerMonitorThread& operator=(const WorkerMonitorThread&) = delete;
    WorkerMonitorThread(WorkerMonitorThread&&) = delete;
    const WorkerMonitorThread& operator=(WorkerMonitorThread&&) = delete;

    // ---------- const member functions ---------------------

    // ---------- member functions ---------------------------
    void setAction(std::function<void()> iFunc) {
      action_ = std::move(iFunc);
      actionSet_.store(true);
    }

    void startThread();

    void setupSignalHandling();

    void stop();

  private:
    static void sig_handler(int sig, siginfo_t*, void*);
    void run();
    // ---------- member data --------------------------------
    std::atomic<bool> stopRequested_ = false;
    std::atomic<bool> helperReady_ = false;
    std::atomic<bool> actionSet_ = false;
    CMS_THREAD_GUARD(actionSet_) std::function<void()> action_;

    static std::atomic<int> s_pipeReadEnd;
    static std::atomic<int> s_pipeWriteEnd;
    static std::atomic<bool> s_helperThreadDone;

    std::thread helperThread_;
  };
}  // namespace edm::shared_memory

#endif