CMS 3D CMS Logo

 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Groups Pages
List of all members | Public Member Functions | Private Member Functions | Private Attributes
edm::DelayedReaderInputProductResolver Class Reference

#include <ProductResolvers.h>

Inheritance diagram for edm::DelayedReaderInputProductResolver:
edm::MergeableInputProductResolver edm::DataManagingProductResolver edm::DataManagingOrAliasProductResolver edm::ProductResolverBase

Public Member Functions

 DelayedReaderInputProductResolver (std::shared_ptr< BranchDescription const > bd)
 
void setupUnscheduled (UnscheduledConfigurator const &) final
 
- Public Member Functions inherited from edm::MergeableInputProductResolver
 MergeableInputProductResolver (std::shared_ptr< BranchDescription const > bd, ProductStatus iDefaultStatus)
 
- Public Member Functions inherited from edm::DataManagingProductResolver
void connectTo (ProductResolverBase const &, Principal const *) final
 
 DataManagingProductResolver (std::shared_ptr< BranchDescription const > bd, ProductStatus iDefaultStatus)
 
void resetStatus ()
 
template<bool callResolver, typename FUNC >
ProductResolverBase::Resolution resolveProductImpl (FUNC resolver) const
 
- Public Member Functions inherited from edm::DataManagingOrAliasProductResolver
 DataManagingOrAliasProductResolver ()
 
- Public Member Functions inherited from edm::ProductResolverBase
BranchDescription const & branchDescription () const
 
std::string const & moduleLabel () const
 
ProductResolverBaseoperator= (ProductResolverBase const &)=delete
 
void prefetchAsync (WaitingTaskHolder waitTask, Principal const &principal, bool skipCurrentProcess, ServiceToken const &token, SharedResourcesAcquirer *sra, ModuleCallingContext const *mcc) const
 
std::string const & processName () const
 
ProductID const & productID () const
 
std::string const & productInstanceName () const
 
ProductProvenance const * productProvenancePtr () const
 
bool productResolved () const
 
 ProductResolverBase ()
 
 ProductResolverBase (ProductResolverBase const &)=delete
 
TypeID productType () const
 
bool productUnavailable () const
 
bool productWasDeleted () const
 
bool productWasFetchedAndIsValid (bool iSkipCurrentProcess) const
 
Provenance const * provenance () const
 
bool provenanceAvailable () const
 
void resetBranchDescription (std::shared_ptr< BranchDescription const > bd)
 
void resetProductData ()
 
std::string const & resolvedModuleLabel () const
 
Resolution resolveProduct (Principal const &principal, bool skipCurrentProcess, SharedResourcesAcquirer *sra, ModuleCallingContext const *mcc) const
 
void retrieveAndMerge (Principal const &principal, MergeableRunProductMetadata const *mergeableRunProductMetadata) const
 
void setMergeableRunProductMetadata (MergeableRunProductMetadata const *mrpm)
 
void setProductID (ProductID const &pid)
 
void setProductProvenanceRetriever (ProductProvenanceRetriever const *provRetriever)
 
bool singleProduct () const
 
StableProvenance const * stableProvenance () const
 
void unsafe_deleteProduct () const
 
bool unscheduledWasNotRun () const
 
void write (std::ostream &os) const
 
virtual ~ProductResolverBase ()
 

Private Member Functions

bool isFromCurrentProcess () const final
 
void prefetchAsync_ (WaitingTaskHolder waitTask, Principal const &principal, bool skipCurrentProcess, ServiceToken const &token, SharedResourcesAcquirer *sra, ModuleCallingContext const *mcc) const override
 
void resetProductData_ (bool deleteEarly) override
 
Resolution resolveProduct_ (Principal const &principal, bool skipCurrentProcess, SharedResourcesAcquirer *sra, ModuleCallingContext const *mcc) const override
 
void retrieveAndMerge_ (Principal const &principal, MergeableRunProductMetadata const *mergeableRunProductMetadata) const override
 
void setMergeableRunProductMetadata_ (MergeableRunProductMetadata const *) override
 
bool unscheduledWasNotRun_ () const final
 

Private Attributes

UnscheduledAuxiliary const * aux_
 
std::atomic< bool > m_prefetchRequested
 
WaitingTaskList m_waitingTasks
 

Additional Inherited Members

- Public Types inherited from edm::DataManagingProductResolver
enum  ProductStatus {
  ProductStatus::ProductSet, ProductStatus::NotPut, ProductStatus::ResolveFailed, ProductStatus::ResolveNotRun,
  ProductStatus::ProductDeleted
}
 
- Protected Member Functions inherited from edm::MergeableInputProductResolver
void mergeProduct (std::shared_ptr< WrapperBase > edp, MergeableRunProductMetadata const *) const
 
void setOrMergeProduct (std::shared_ptr< WrapperBase > prod, MergeableRunProductMetadata const *mergeableRunProductMetadata) const
 
- Protected Member Functions inherited from edm::DataManagingProductResolver
void checkType (WrapperBase const &prod) const
 
ProductStatus defaultStatus () const
 
ProductData const & getProductData () const final
 
template<bool callResolver, typename FUNC >
Resolution resolveProductImpl (FUNC resolver) const
 
void setFailedStatus () const
 
void setMergeableRunProductMetadataInProductData (MergeableRunProductMetadata const *)
 
void setProduct (std::unique_ptr< WrapperBase > edp) const
 
void setProduct (std::shared_ptr< WrapperBase > edp) const
 
ProductStatus status () const
 

Detailed Description

Definition at line 119 of file ProductResolvers.h.

Constructor & Destructor Documentation

edm::DelayedReaderInputProductResolver::DelayedReaderInputProductResolver ( std::shared_ptr< BranchDescription const >  bd)
inlineexplicit

Definition at line 121 of file ProductResolvers.h.

123  assert(bd->onDemand());
124  assert(not bd->produced());
125  }
MergeableInputProductResolver(std::shared_ptr< BranchDescription const > bd, ProductStatus iDefaultStatus)
assert(be >=bs)
UnscheduledAuxiliary const * aux_

Member Function Documentation

bool edm::DelayedReaderInputProductResolver::isFromCurrentProcess ( ) const
finalprivatevirtual

Implements edm::DataManagingProductResolver.

Definition at line 324 of file ProductResolvers.cc.

324 { return false; }
void edm::DelayedReaderInputProductResolver::prefetchAsync_ ( WaitingTaskHolder  waitTask,
Principal const &  principal,
bool  skipCurrentProcess,
ServiceToken const &  token,
SharedResourcesAcquirer sra,
ModuleCallingContext const *  mcc 
) const
overrideprivatevirtual

Implements edm::ProductResolverBase.

Definition at line 253 of file ProductResolvers.cc.

References edm::WaitingTaskList::add(), edm::ProductResolverBase::branchDescription(), CMS_SA_ALLOW, edm::WaitingTaskList::doneWaiting(), edm::WaitingTaskHolder::group(), edm::InEvent, edm::InProcess, edm::ServiceWeakToken::lock(), m_prefetchRequested, m_waitingTasks, edm::make_functor_task(), edm::ProductResolverBase::productResolved(), edm::SerialTaskQueueChain::push(), DQM::reader, alignCSCRings::s, edm::DataManagingProductResolver::setProduct(), submitPVValidationJobs::t, and unpackBuffers-CaloStage2::token.

258  {
259  //need to try changing m_prefetchRequested before adding to m_waitingTasks
260  bool expected = false;
261  bool prefetchRequested = m_prefetchRequested.compare_exchange_strong(expected, true);
262  m_waitingTasks.add(waitTask);
263 
264  if (prefetchRequested) {
265  ServiceWeakToken weakToken = token;
266  auto workToDo = [this, mcc, &principal, weakToken]() {
267  //need to make sure Service system is activated on the reading thread
268  ServiceRegistry::Operate operate(weakToken.lock());
269  // Caught exception is propagated via WaitingTaskList
270  CMS_SA_ALLOW try {
271  resolveProductImpl<true>([this, &principal, mcc]() {
272  if (principal.branchType() != InEvent && principal.branchType() != InProcess) {
273  return;
274  }
275  if (auto reader = principal.reader()) {
276  std::unique_lock<std::recursive_mutex> guard;
277  if (auto sr = reader->sharedResources().second) {
278  guard = std::unique_lock<std::recursive_mutex>(*sr);
279  }
280  if (not productResolved()) {
281  //another thread could have finished this while we were waiting
282  setProduct(reader->getProduct(branchDescription().branchID(), &principal, mcc));
283  }
284  }
285  });
286  } catch (...) {
287  this->m_waitingTasks.doneWaiting(std::current_exception());
288  return;
289  }
290  this->m_waitingTasks.doneWaiting(nullptr);
291  };
292 
293  SerialTaskQueueChain* queue = nullptr;
294  if (auto reader = principal.reader()) {
295  if (auto shared_res = reader->sharedResources().first) {
296  queue = &(shared_res->serialQueueChain());
297  }
298  }
299  if (queue) {
300  queue->push(*waitTask.group(), workToDo);
301  } else {
302  //Have to create a new task
303  auto t = make_functor_task(workToDo);
304  waitTask.group()->run([t]() {
305  TaskSentry s{t};
306  t->execute();
307  });
308  }
309  }
310  }
#define CMS_SA_ALLOW
void doneWaiting(std::exception_ptr iPtr)
Signals that the resource is now available and tasks should be spawned.
tuple reader
Definition: DQM.py:105
void add(tbb::task_group *, WaitingTask *)
Adds task to the waiting list.
void setProduct(std::unique_ptr< WrapperBase > edp) const
FunctorTask< F > * make_functor_task(F f)
Definition: FunctorTask.h:44
BranchDescription const & branchDescription() const
void edm::DelayedReaderInputProductResolver::resetProductData_ ( bool  deleteEarly)
overrideprivatevirtual

Reimplemented from edm::DataManagingProductResolver.

Definition at line 312 of file ProductResolvers.cc.

References m_prefetchRequested, m_waitingTasks, edm::WaitingTaskList::reset(), and edm::DataManagingProductResolver::resetProductData_().

312  {
313  if (not deleteEarly) {
314  m_prefetchRequested = false;
316  }
318  }
void resetProductData_(bool deleteEarly) override
void reset()
Resets access to the resource so that added tasks will wait.
ProductResolverBase::Resolution edm::DelayedReaderInputProductResolver::resolveProduct_ ( Principal const &  principal,
bool  skipCurrentProcess,
SharedResourcesAcquirer sra,
ModuleCallingContext const *  mcc 
) const
overrideprivatevirtual

Implements edm::ProductResolverBase.

Definition at line 161 of file ProductResolvers.cc.

References aux_, edm::ProductResolverBase::branchDescription(), edm::ModuleCallingContext::getStreamContext(), edm::InEvent, edm::InLumi, edm::InRun, edm::make_sentry(), edm::ProductResolverBase::productResolved(), DQM::reader, and edm::DataManagingProductResolver::setProduct().

162  {
163  return resolveProductImpl<true>([this, &principal, mcc]() {
164  auto branchType = principal.branchType();
165  if (branchType == InLumi || branchType == InRun) {
166  //delayed get has not been allowed with Run or Lumis
167  // The file may already be closed so the reader is invalid
168  return;
169  }
170  if (mcc and branchType == InEvent and aux_) {
171  aux_->preModuleDelayedGetSignal_.emit(*(mcc->getStreamContext()), *mcc);
172  }
173 
174  auto sentry(make_sentry(mcc, [this, branchType](ModuleCallingContext const* iContext) {
175  if (branchType == InEvent and aux_) {
176  aux_->postModuleDelayedGetSignal_.emit(*(iContext->getStreamContext()), *iContext);
177  }
178  }));
179 
180  if (auto reader = principal.reader()) {
181  std::unique_lock<std::recursive_mutex> guard;
182  if (auto sr = reader->sharedResources().second) {
183  guard = std::unique_lock<std::recursive_mutex>(*sr);
184  }
185  if (not productResolved()) {
186  //another thread could have beaten us here
187  setProduct(reader->getProduct(branchDescription().branchID(), &principal, mcc));
188  }
189  }
190  });
191  }
std::unique_ptr< T, F > make_sentry(T *iObject, F iFunc)
NOTE: if iObject is null, then iFunc will not be called.
Definition: make_sentry.h:30
UnscheduledAuxiliary const * aux_
tuple reader
Definition: DQM.py:105
void setProduct(std::unique_ptr< WrapperBase > edp) const
BranchDescription const & branchDescription() const
void edm::DelayedReaderInputProductResolver::retrieveAndMerge_ ( Principal const &  principal,
MergeableRunProductMetadata const *  mergeableRunProductMetadata 
) const
overrideprivatevirtual

Reimplemented from edm::ProductResolverBase.

Definition at line 193 of file ProductResolvers.cc.

References edm::ProductResolverBase::branchDescription(), edm::BranchDescription::branchName(), edm::BranchDescription::branchType(), edm::DataManagingProductResolver::defaultStatus(), Exception, edm::InRun, edm::errors::LogicError, edm::errors::MismatchedInputFiles, eostools::move(), edm::DataManagingProductResolver::ProductSet, DQM::reader, edm::Principal::reader(), edm::DataManagingProductResolver::ResolveFailed, edm::DataManagingProductResolver::setFailedStatus(), edm::MergeableInputProductResolver::setOrMergeProduct(), and edm::DataManagingProductResolver::status().

194  {
195  if (auto reader = principal.reader()) {
196  std::unique_lock<std::recursive_mutex> guard;
197  if (auto sr = reader->sharedResources().second) {
198  guard = std::unique_lock<std::recursive_mutex>(*sr);
199  }
200 
201  //Can't use resolveProductImpl since it first checks to see
202  // if the product was already retrieved and then returns if it is
203  auto edp(reader->getProduct(branchDescription().branchID(), &principal));
204 
205  if (edp.get() != nullptr) {
206  if (edp->isMergeable() && branchDescription().branchType() == InRun && !edp->hasSwap()) {
208  << "Missing definition of member function swap for branch name " << branchDescription().branchName()
209  << "\n"
210  << "Mergeable data types written to a Run must have the swap member function defined"
211  << "\n";
212  }
214  (status() == ProductStatus::ResolveFailed && !branchDescription().isMergeable())) {
215  setOrMergeProduct(std::move(edp), mergeableRunProductMetadata);
216  } else { // status() == ResolveFailed && branchDescription().isMergeable()
218  << "Merge of Run or Lumi product failed for branch " << branchDescription().branchName() << "\n"
219  << "The product branch was dropped in the first run or lumi fragment and present in a later one"
220  << "\n"
221  << "The solution is to drop the branch on input. Or better do not create inconsistent files\n"
222  << "that need to be merged in the first place.\n";
223  }
224  } else if (status() == defaultStatus()) {
225  setFailedStatus();
226  } else if (status() != ProductStatus::ResolveFailed && branchDescription().isMergeable()) {
228  << "Merge of Run or Lumi product failed for branch " << branchDescription().branchName() << "\n"
229  << "The product branch was present in first run or lumi fragment and dropped in a later one"
230  << "\n"
231  << "The solution is to drop the branch on input. Or better do not create inconsistent files\n"
232  << "that need to be merged in the first place.\n";
233  }
234  // Do nothing in other case. status is ResolveFailed already or
235  // it is not mergeable and the status is ProductSet
236  }
237  }
std::string const & branchName() const
BranchType const & branchType() const
ProductStatus status() const
void setOrMergeProduct(std::shared_ptr< WrapperBase > prod, MergeableRunProductMetadata const *mergeableRunProductMetadata) const
def move
Definition: eostools.py:511
tuple reader
Definition: DQM.py:105
ProductStatus defaultStatus() const
BranchDescription const & branchDescription() const
void edm::DelayedReaderInputProductResolver::setMergeableRunProductMetadata_ ( MergeableRunProductMetadata const *  mrpm)
overrideprivatevirtual

Reimplemented from edm::ProductResolverBase.

Definition at line 249 of file ProductResolvers.cc.

References edm::DataManagingProductResolver::setMergeableRunProductMetadataInProductData().

249  {
251  }
void setMergeableRunProductMetadataInProductData(MergeableRunProductMetadata const *)
void edm::DelayedReaderInputProductResolver::setupUnscheduled ( UnscheduledConfigurator const &  iConfigure)
finalvirtual

Reimplemented from edm::ProductResolverBase.

Definition at line 320 of file ProductResolvers.cc.

References aux_, and edm::UnscheduledConfigurator::auxiliary().

320  {
321  aux_ = iConfigure.auxiliary();
322  }
UnscheduledAuxiliary const * aux_
bool edm::DelayedReaderInputProductResolver::unscheduledWasNotRun_ ( ) const
inlinefinalprivatevirtual

Implements edm::ProductResolverBase.

Definition at line 148 of file ProductResolvers.h.

148 { return false; }

Member Data Documentation

UnscheduledAuxiliary const* edm::DelayedReaderInputProductResolver::aux_
private

Definition at line 154 of file ProductResolvers.h.

Referenced by resolveProduct_(), and setupUnscheduled().

std::atomic<bool> edm::DelayedReaderInputProductResolver::m_prefetchRequested
mutableprivate

Definition at line 152 of file ProductResolvers.h.

Referenced by prefetchAsync_(), and resetProductData_().

WaitingTaskList edm::DelayedReaderInputProductResolver::m_waitingTasks
mutableprivate

Definition at line 153 of file ProductResolvers.h.

Referenced by prefetchAsync_(), and resetProductData_().