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 348 of file ProductResolvers.cc.

348 { 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 270 of file ProductResolvers.cc.

References edm::WaitingTaskList::add(), edm::ProductResolverBase::branchDescription(), CMS_SA_ALLOW, edm::WaitingTaskList::doneWaiting(), cppFunctionSkipper::exception, Exception, 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(), edm::errors::StdException, submitPVValidationJobs::t, and unpackBuffers-CaloStage2::token.

275  {
276  //need to try changing m_prefetchRequested before adding to m_waitingTasks
277  bool expected = false;
278  bool prefetchRequested = m_prefetchRequested.compare_exchange_strong(expected, true);
279  m_waitingTasks.add(waitTask);
280 
281  if (prefetchRequested) {
282  ServiceWeakToken weakToken = token;
283  auto workToDo = [this, mcc, &principal, weakToken]() {
284  //need to make sure Service system is activated on the reading thread
285  ServiceRegistry::Operate operate(weakToken.lock());
286  // Caught exception is propagated via WaitingTaskList
287  CMS_SA_ALLOW try {
288  resolveProductImpl<true>([this, &principal, mcc]() {
289  if (principal.branchType() != InEvent && principal.branchType() != InProcess) {
290  return;
291  }
292  if (auto reader = principal.reader()) {
293  std::unique_lock<std::recursive_mutex> guard;
294  if (auto sr = reader->sharedResources().second) {
295  guard = std::unique_lock<std::recursive_mutex>(*sr);
296  }
297  if (not productResolved()) {
298  try {
299  //another thread could have finished this while we were waiting
300  setProduct(reader->getProduct(branchDescription().branchID(), &principal, mcc));
301  } catch (cms::Exception& e) {
302  throw extendException(e, branchDescription(), mcc);
303  } catch (std::exception const& e) {
304  auto newExcept = edm::Exception(errors::StdException) << e.what();
305  throw extendException(newExcept, branchDescription(), mcc);
306  }
307  }
308  }
309  });
310  } catch (...) {
311  this->m_waitingTasks.doneWaiting(std::current_exception());
312  return;
313  }
314  this->m_waitingTasks.doneWaiting(nullptr);
315  };
316 
317  SerialTaskQueueChain* queue = nullptr;
318  if (auto reader = principal.reader()) {
319  if (auto shared_res = reader->sharedResources().first) {
320  queue = &(shared_res->serialQueueChain());
321  }
322  }
323  if (queue) {
324  queue->push(*waitTask.group(), workToDo);
325  } else {
326  //Have to create a new task
327  auto t = make_functor_task(workToDo);
328  waitTask.group()->run([t]() {
329  TaskSentry s{t};
330  t->execute();
331  });
332  }
333  }
334  }
#define CMS_SA_ALLOW
void doneWaiting(std::exception_ptr iPtr)
Signals that the resource is now available and tasks should be spawned.
void add(oneapi::tbb::task_group *, WaitingTask *)
Adds task to the waiting list.
tuple reader
Definition: DQM.py:105
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 336 of file ProductResolvers.cc.

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

336  {
337  if (not deleteEarly) {
338  m_prefetchRequested = false;
340  }
342  }
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 171 of file ProductResolvers.cc.

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

172  {
173  return resolveProductImpl<true>([this, &principal, mcc]() {
174  auto branchType = principal.branchType();
175  if (branchType == InLumi || branchType == InRun) {
176  //delayed get has not been allowed with Run or Lumis
177  // The file may already be closed so the reader is invalid
178  return;
179  }
180  if (mcc and branchType == InEvent and aux_) {
181  aux_->preModuleDelayedGetSignal_.emit(*(mcc->getStreamContext()), *mcc);
182  }
183 
184  auto sentry(make_sentry(mcc, [this, branchType](ModuleCallingContext const* iContext) {
185  if (branchType == InEvent and aux_) {
186  aux_->postModuleDelayedGetSignal_.emit(*(iContext->getStreamContext()), *iContext);
187  }
188  }));
189 
190  if (auto reader = principal.reader()) {
191  std::unique_lock<std::recursive_mutex> guard;
192  if (auto sr = reader->sharedResources().second) {
193  guard = std::unique_lock<std::recursive_mutex>(*sr);
194  }
195  if (not productResolved()) {
196  try {
197  //another thread could have beaten us here
198  setProduct(reader->getProduct(branchDescription().branchID(), &principal, mcc));
199  } catch (cms::Exception& e) {
200  throw extendException(e, branchDescription(), mcc);
201  } catch (std::exception const& e) {
202  auto newExcept = edm::Exception(errors::StdException) << e.what();
203  throw extendException(newExcept, branchDescription(), mcc);
204  }
205  }
206  }
207  });
208  }
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 210 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().

211  {
212  if (auto reader = principal.reader()) {
213  std::unique_lock<std::recursive_mutex> guard;
214  if (auto sr = reader->sharedResources().second) {
215  guard = std::unique_lock<std::recursive_mutex>(*sr);
216  }
217 
218  //Can't use resolveProductImpl since it first checks to see
219  // if the product was already retrieved and then returns if it is
220  auto edp(reader->getProduct(branchDescription().branchID(), &principal));
221 
222  if (edp.get() != nullptr) {
223  if (edp->isMergeable() && branchDescription().branchType() == InRun && !edp->hasSwap()) {
225  << "Missing definition of member function swap for branch name " << branchDescription().branchName()
226  << "\n"
227  << "Mergeable data types written to a Run must have the swap member function defined"
228  << "\n";
229  }
231  (status() == ProductStatus::ResolveFailed && !branchDescription().isMergeable())) {
232  setOrMergeProduct(std::move(edp), mergeableRunProductMetadata);
233  } else { // status() == ResolveFailed && branchDescription().isMergeable()
235  << "Merge of Run or Lumi product failed for branch " << branchDescription().branchName() << "\n"
236  << "The product branch was dropped in the first run or lumi fragment and present in a later one"
237  << "\n"
238  << "The solution is to drop the branch on input. Or better do not create inconsistent files\n"
239  << "that need to be merged in the first place.\n";
240  }
241  } else if (status() == defaultStatus()) {
242  setFailedStatus();
243  } else if (status() != ProductStatus::ResolveFailed && branchDescription().isMergeable()) {
245  << "Merge of Run or Lumi product failed for branch " << branchDescription().branchName() << "\n"
246  << "The product branch was present in first run or lumi fragment and dropped in a later one"
247  << "\n"
248  << "The solution is to drop the branch on input. Or better do not create inconsistent files\n"
249  << "that need to be merged in the first place.\n";
250  }
251  // Do nothing in other case. status is ResolveFailed already or
252  // it is not mergeable and the status is ProductSet
253  }
254  }
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 266 of file ProductResolvers.cc.

References edm::DataManagingProductResolver::setMergeableRunProductMetadataInProductData().

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

Reimplemented from edm::ProductResolverBase.

Definition at line 344 of file ProductResolvers.cc.

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

344  {
345  aux_ = iConfigure.auxiliary();
346  }
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_().