WaitingTaskHolder.h

Go to the documentation of this file.

#ifndef FWCore_Concurrency_WaitingTaskHolder_h
#define FWCore_Concurrency_WaitingTaskHolder_h
// -*- C++ -*-
//
// Package:     FWCore/Concurrency
// Class  :     WaitingTaskHolder
//
//
// Original Author:  FWCore
//         Created:  Fri, 18 Nov 2016 20:30:42 GMT
//

// system include files
#include <cassert>
#include "oneapi/tbb/task_group.h"

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

// forward declarations

namespace edm {
  class WaitingTaskHolder {
  public:
    friend class WaitingTaskList;
    friend class WaitingTaskWithArenaHolder;

    WaitingTaskHolder() : m_task(nullptr), m_group(nullptr) {}

    explicit WaitingTaskHolder(oneapi::tbb::task_group& iGroup, edm::WaitingTask* iTask)
        : m_task(iTask), m_group(&iGroup) {
      m_task->increment_ref_count();
    }
    ~WaitingTaskHolder() {
      if (m_task) {
        doneWaiting(std::exception_ptr{});
      }
    }

    WaitingTaskHolder(const WaitingTaskHolder& iHolder) : m_task(iHolder.m_task), m_group(iHolder.m_group) {
      m_task->increment_ref_count();
    }

    WaitingTaskHolder(WaitingTaskHolder&& iOther) : m_task(iOther.m_task), m_group(iOther.m_group) {
      iOther.m_task = nullptr;
    }

    WaitingTaskHolder& operator=(const WaitingTaskHolder& iRHS) {
      WaitingTaskHolder tmp(iRHS);
      std::swap(m_task, tmp.m_task);
      std::swap(m_group, tmp.m_group);
      return *this;
    }

    WaitingTaskHolder& operator=(WaitingTaskHolder&& iRHS) {
      WaitingTaskHolder tmp(std::move(iRHS));
      std::swap(m_task, tmp.m_task);
      std::swap(m_group, tmp.m_group);
      return *this;
    }

    // ---------- const member functions ---------------------
    bool taskHasFailed() const noexcept { return static_cast<bool>(m_task->exceptionPtr()); }

    bool hasTask() const noexcept { return m_task != nullptr; }
    CMS_SA_ALLOW oneapi::tbb::task_group* group() const noexcept { return m_group; }
    // ---------- static member functions --------------------

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

    void presetTaskAsFailed(std::exception_ptr iExcept) {
      if (iExcept) {
        m_task->dependentTaskFailed(iExcept);
      }
    }

    void doneWaiting(std::exception_ptr iExcept) {
      if (iExcept) {
        m_task->dependentTaskFailed(iExcept);
      }
      //task_group::run can run the task before we finish
      // doneWaiting and some other thread might
      // try to reuse this object. Resetting
      // before spawn avoids problems
      auto task = m_task;
      m_task = nullptr;
      if (0 == task->decrement_ref_count()) {
        m_group->run([task]() {
          TaskSentry s{task};
          task->execute();
        });
      }
    }

  private:
    WaitingTask* release_no_decrement() noexcept {
      auto t = m_task;
      m_task = nullptr;
      return t;
    }
    // ---------- member data --------------------------------
    WaitingTask* m_task;
    oneapi::tbb::task_group* m_group;
  };
}  // namespace edm

#endif