ReusableThread.hpp

Go to the documentation of this file.

#ifndef UTILITIES_INCLUDE_UTILITIES_REUSABLETHREAD_HPP_
#define UTILITIES_INCLUDE_UTILITIES_REUSABLETHREAD_HPP_
 
#include <atomic>
#include <chrono>
#include <condition_variable>
#include <functional>
#include <mutex>
#include <string>
#include <thread>
 
namespace dunedaq::utilities {
 
class ReusableThread
{
public:
  explicit ReusableThread(int threadid = 0);
 
  ~ReusableThread();
 
  ReusableThread(const ReusableThread&) = delete;            
  ReusableThread& operator=(const ReusableThread&) = delete; 
  ReusableThread(ReusableThread&&) = delete;                 
  ReusableThread& operator=(ReusableThread&&) = delete;      
 
  // Set thread ID
  void set_thread_id(int tid) { m_thread_id = tid; }
 
  // Get thread ID
  int get_thread_id() const { return m_thread_id; }
 
  // Set name for pthread handle
  void set_name(const std::string& name, int tid);
 
  // Pin thread to CPU
  void set_pin(int cpuid);
 
  // Check for completed task execution
  bool get_readiness() const { return m_task_executed; }
 
  // Set task to be executed
  template<typename Function, typename... Args>
  bool set_work(Function&& f, Args&&... args)
  {
    if (!m_task_assigned && m_task_executed.exchange(false)) {
      m_task = std::bind(f, args...); // NOLINT
      m_task_assigned = true;
      m_cv.notify_all();
      return true;
    }
    return false;
  }
 
private:
  // Internals
  int m_thread_id;
  std::atomic<bool> m_task_executed;
  std::atomic<bool> m_task_assigned;
  std::atomic<bool> m_thread_quit;
  std::atomic<bool> m_worker_done;
  std::atomic<bool> m_named;
  std::function<void()> m_task;
 
  // Locks
  std::mutex m_mtx;
  std::condition_variable m_cv;
  std::thread m_thread;
 
  // Actual worker thread
  void thread_worker();
};
 
} // namespace dunedaq::utilities
 
#endif // UTILITIES_INCLUDE_UTILITIES_REUSABLETHREAD_HPP_