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{} {}
    ~WaitingTask() override{};

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


    std::exception_ptr exceptionPtr() const {
      if (m_ptrSet == static_cast<unsigned char>(State::kSet)) {
        return m_ptr;
      }
      return std::exception_ptr{};
    }

  protected:
    std::exception_ptr const& uncheckedExceptionPtr() const { return m_ptr; }

  private:
    enum class State : unsigned char { kUnset = 0, kSetting = 1, kSet = 2 };

    void dependentTaskFailed(std::exception_ptr iPtr) {
      unsigned char isSet = static_cast<unsigned char>(State::kUnset);
      if (iPtr and m_ptrSet.compare_exchange_strong(isSet, static_cast<unsigned char>(State::kSetting))) {
        m_ptr = iPtr;
        m_ptrSet = static_cast<unsigned char>(State::kSet);
      }
    }

    std::exception_ptr m_ptr;
    std::atomic<unsigned char> m_ptrSet = static_cast<unsigned char>(State::kUnset);
  };

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

    void execute() final { func_(uncheckedExceptionPtr() ? &uncheckedExceptionPtr() : nullptr); };

  private:
    F func_;
  };

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

}  // namespace edm

#endif