WaitingTask.h

Go to the documentation of this file.

#ifndef FWCore_Concurrency_WaitingTask_h
#define FWCore_Concurrency_WaitingTask_h
// -*- C++ -*-
//
// Package:     Concurrency
// Class  :     WaitingTask
//
//
// Original Author:  Chris Jones
//         Created:  Thu Feb 21 13:46:31 CST 2013
// $Id$
//

// system include files
#include <atomic>
#include <exception>
#include <memory>
#include "tbb/task.h"

// user include files

// forward declarations

namespace edm {
  class WaitingTaskList;
  class WaitingTaskHolder;
  class WaitingTaskWithArenaHolder;

  class WaitingTask : public tbb::task {
  public:
    friend class WaitingTaskList;
    friend class WaitingTaskHolder;
    friend class WaitingTaskWithArenaHolder;

    WaitingTask() : m_ptr{nullptr} {}
    ~WaitingTask() override { delete m_ptr.load(); };

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


    std::exception_ptr const* exceptionPtr() const { return m_ptr.load(); }

  private:

    void dependentTaskFailed(std::exception_ptr iPtr) {
      if (iPtr and not m_ptr) {
        auto temp = std::make_unique<std::exception_ptr>(iPtr);
        std::exception_ptr* expected = nullptr;
        if (m_ptr.compare_exchange_strong(expected, temp.get())) {
          temp.release();
        }
      }
    }

    std::atomic<std::exception_ptr*> m_ptr;
  };

  template <typename F>
  class FunctorWaitingTask : public WaitingTask {
  public:
    explicit FunctorWaitingTask(F f) : func_(std::move(f)) {}

    task* execute() override {
      func_(exceptionPtr());
      return nullptr;
    };

  private:
    F func_;
  };

  template <typename ALLOC, typename F>
  FunctorWaitingTask<F>* make_waiting_task(ALLOC&& iAlloc, F f) {
    return new (iAlloc) FunctorWaitingTask<F>(std::move(f));
  }

}  // namespace edm

#endif