Account getRunDuration in the cache report of BasicCCDBManager

AliceO2Group · Mar 5, 2024 · f8e40c9 · f8e40c9
1 parent 14189ae
commit f8e40c9
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 3 deletions.
diff --git a/CCDB/include/CCDB/BasicCCDBManager.h b/CCDB/include/CCDB/BasicCCDBManager.h
@@ -183,7 +183,7 @@ class CCDBManagerInstance
   /// A convenience function for MC to fetch
   /// valid start and end timestamps given an ALICE run number.
   /// On error it fatals (if fatal == true) or else returns the pair -1, -1.
-  std::pair<int64_t, int64_t> getRunDuration(int runnumber, bool fatal = true) const;
+  std::pair<int64_t, int64_t> getRunDuration(int runnumber, bool fatal = true);
 
   /// A convenience function for MC to fetch
   /// valid start and end timestamps given an ALICE run number.

diff --git a/CCDB/src/BasicCCDBManager.cxx b/CCDB/src/BasicCCDBManager.cxx
@@ -47,9 +47,26 @@ std::pair<int64_t, int64_t> CCDBManagerInstance::getRunDuration(o2::ccdb::CcdbAp
   return std::make_pair(sor, eor);
 }
 
-std::pair<int64_t, int64_t> CCDBManagerInstance::getRunDuration(int runnumber, bool fatal) const
+std::pair<int64_t, int64_t> CCDBManagerInstance::getRunDuration(int runnumber, bool fatal)
 {
-  return CCDBManagerInstance::getRunDuration(mCCDBAccessor, runnumber, fatal);
+  mQueries++;
+  if (!isCachingEnabled()) {
+    return CCDBManagerInstance::getRunDuration(mCCDBAccessor, runnumber, fatal);
+  }
+  auto& cached = mCache["RCT-Run-Info HeaderOnly"];
+  std::pair<int64_t, int64_t> rd;
+  cached.queries++;
+  if (cached.startvalidity != runnumber) { // need to fetch
+    rd = CCDBManagerInstance::getRunDuration(mCCDBAccessor, runnumber, fatal);
+    cached.objPtr = std::make_shared<std::pair<int64_t, int64_t>>(rd);
+    cached.startvalidity = runnumber;
+    cached.endvalidity = runnumber + 1;
+    cached.minSize = cached.maxSize = 0;
+    cached.fetches++;
+  } else {
+    rd = *reinterpret_cast<std::pair<int64_t, int64_t>*>(cached.objPtr.get());
+  }
+  return rd;
 }
 
 std::string CCDBManagerInstance::getSummaryString() const