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>

// user include files
#include "FWCore/Concurrency/interface/TaskBase.h"

// forward declarations

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

  class WaitingTask : public TaskBase {
  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;
  };

  class FinalWaitingTask : public WaitingTask {
  public:
    FinalWaitingTask() : m_done{false} {}

    void execute() final { m_done = true; }

    bool done() const { return m_done.load(); }

  private:
    void recycle() final {}
    std::atomic<bool> m_done;
  };

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

    void execute() final { func_(exceptionPtr()); };

  private:
    F func_;
  };

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

}  // namespace edm

#endif