DataProxyTemplate.h

Go to the documentation of this file.

#ifndef FWCore_Framework_DataProxyTemplate_h
#define FWCore_Framework_DataProxyTemplate_h
// -*- C++ -*-
//
// Package:     Framework
// Class  :     DataProxyTemplate
//
//
// Author:      Chris Jones
// Created:     Thu Mar 31 12:45:32 EST 2005
//
 
// system include files
 
// user include files
#include "FWCore/Framework/interface/DataProxy.h"
#include "FWCore/Framework/interface/EventSetupRecord.h"
#include "FWCore/ServiceRegistry/interface/ServiceRegistry.h"
#include "FWCore/Concurrency/interface/WaitingTaskList.h"
#include <cassert>
#include <limits>
#include <atomic>
 
// forward declarations
 
namespace edm {
 
  class EventSetupImpl;
 
  namespace eventsetup {
 
    template <class RecordT, class DataT>
    class DataProxyTemplate : public DataProxy {
    public:
      typedef DataT value_type;
      typedef RecordT record_type;
 
      DataProxyTemplate() {}
 
      void prefetchAsyncImpl(WaitingTask* iTask,
                             const EventSetupRecordImpl& iRecord,
                             const DataKey& iKey,
                             EventSetupImpl const* iEventSetupImpl,
                             edm::ServiceToken const& iToken) override {
        assert(iRecord.key() == RecordT::keyForClass());
        bool expected = false;
        bool doPrefetch = prefetching_.compare_exchange_strong(expected, true);
        taskList_.add(iTask);
 
        if (doPrefetch) {
          tbb::task::spawn(*edm::make_waiting_task(
              tbb::task::allocate_root(), [this, &iRecord, iKey, iEventSetupImpl, iToken](std::exception_ptr const*) {
                try {
                  RecordT rec;
                  rec.setImpl(&iRecord, std::numeric_limits<unsigned int>::max(), nullptr, iEventSetupImpl, true);
                  ServiceRegistry::Operate operate(iToken);
                  this->make(rec, iKey);
                } catch (...) {
                  this->taskList_.doneWaiting(std::current_exception());
                  return;
                }
                this->taskList_.doneWaiting(std::exception_ptr{});
              }));
        }
      }
 
    protected:
      void invalidateCache() override {
        taskList_.reset();
        prefetching_ = false;
      }
 
      virtual const DataT* make(const RecordT&, const DataKey&) = 0;
 
    private:
      WaitingTaskList taskList_;
      std::atomic<bool> prefetching_{false};
    };
 
  }  // namespace eventsetup
}  // namespace edm
#endif