ReusableThread.cpp

Go to the documentation of this file.

 
#include "utilities/ReusableThread.hpp"
#include "utilities/Issues.hpp"
 
#include <string>
 
dunedaq::utilities::ReusableThread::ReusableThread(int threadid)
  : m_thread_id(threadid)
  , m_task_executed(true)
  , m_task_assigned(false)
  , m_thread_quit(false)
  , m_worker_done(false)
  , m_thread(&ReusableThread::thread_worker, this)
{
}
 
dunedaq::utilities::ReusableThread::~ReusableThread()
{
  while (m_task_assigned) {
    std::this_thread::sleep_for(std::chrono::milliseconds(1));
  }
  m_thread_quit = true;
  while (!m_worker_done) {
    std::this_thread::sleep_for(std::chrono::milliseconds(1));
    m_cv.notify_all();
  }
  m_thread.join();
}
 
void
dunedaq::utilities::ReusableThread::set_name(const std::string& name, int tid)
{
  set_thread_id(tid);
  char tname[16];                                  // NOLINT
  snprintf(tname, 16, "%s-%d", name.c_str(), tid); // NOLINT
  auto handle = m_thread.native_handle();
  pthread_setname_np(handle, tname);
 
  m_named = true;
}
 
void
dunedaq::utilities::ReusableThread::set_pin(int cpuid)
{
  // Require that the thread has been named
  if (!m_named) {
    ers::warning(ThreadingIssue(ERS_HERE, "May not set CPU affinity for un-named thread"));
  }
 
  auto handle = m_thread.native_handle();
  cpu_set_t cpuset;
  CPU_ZERO(&cpuset);
  CPU_SET(cpuid, &cpuset);
  int rc = pthread_setaffinity_np(handle, sizeof(cpu_set_t), &cpuset);
 
  if (rc != 0) {
    ers::warning(ThreadingIssue(ERS_HERE, "Error calling pthread_setaffinity_np: " + std::to_string(rc)));
  }
}
 
void
dunedaq::utilities::ReusableThread::thread_worker()
{
  std::unique_lock<std::mutex> lock(m_mtx);
 
  while (!m_thread_quit) {
    if (!m_task_executed && m_task_assigned) {
      m_task();
      m_task_executed = true;
      m_task_assigned = false;
    } else {
      m_cv.wait(lock);
    }
  }
 
  m_worker_done = true;
}