[QC-156] Access to conditions in CCDB inside Tasks (#224)

Author: knopers8

Framework/advanced.json

@@ -17,6 +17,9 @@
       },
       "consul": {
         "url": "http://consul-test.cern.ch:8500"
+      },
+      "conditionDB": {
+        "url": "ccdb-test.cern.ch:8080"
       }
     },
     "tasks": {

Framework/basic-no-sampling.json

@@ -17,6 +17,9 @@
       },
       "consul": {
         "url": "http://consul-test.cern.ch:8500"
+      },
+      "conditionDB": {
+        "url": "ccdb-test.cern.ch:8080"
       }
     },
     "tasks": {

Framework/basic.json

@@ -17,6 +17,9 @@
       },
       "consul": {
         "url": "http://consul-test.cern.ch:8500"
+      },
+      "conditionDB": {
+        "url": "ccdb-test.cern.ch:8080"
       }
     },
     "tasks": {
@@ -43,7 +46,7 @@
       "id": "tst-raw",
       "active": "true",
       "machines": [],
-      "query" : "random:TST/RAWDATA/0",
+      "query": "random:TST/RAWDATA/0",
       "samplingConditions": [
         {
           "condition": "random",

Framework/example-default.json

@@ -17,6 +17,9 @@
       },
       "consul": {
         "url": "http://consul-test.cern.ch:8500"
+      },
+      "conditionDB": {
+        "url": "ccdb-test.cern.ch:8080"
       }
     },
     "tasks": {

Framework/include/QualityControl/TaskConfig.h

@@ -30,6 +30,7 @@ struct TaskConfig {
   int cycleDurationSeconds;
   int maxNumberCycles;
   std::string consulUrl;
+  std::string conditionUrl = "";
   std::unordered_map<std::string, std::string> customParameters = {};
 };
 

Framework/include/QualityControl/TaskInterface.h

@@ -17,6 +17,7 @@
 #ifndef QC_CORE_TASKINTERFACE_H
 #define QC_CORE_TASKINTERFACE_H
 
+#include <map>
 #include <memory>
 #include <string>
 #include <unordered_map>
@@ -27,6 +28,13 @@
 #include "QualityControl/Activity.h"
 #include "QualityControl/ObjectsManager.h"
 
+namespace o2::ccdb
+{
+class CcdbApi;
+}
+
+class TObject;
+
 namespace o2::quality_control::core
 {
 
@@ -62,6 +70,8 @@ class TaskInterface
   /// Move assignment operator
   TaskInterface& operator=(TaskInterface&& other) /* noexcept */ = default; // error with gcc if noexcept
 
+  virtual void loadCcdb(std::string url) final;
+
   // Definition of the methods for the template method pattern
   virtual void initialize(o2::framework::InitContext& ctx) = 0;
   virtual void startOfActivity(Activity& activity) = 0;
@@ -79,12 +89,15 @@ class TaskInterface
 
  protected:
   std::shared_ptr<ObjectsManager> getObjectsManager();
+  TObject* retrieveCondition(std::string path, std::map<std::string, std::string> metadata = {}, long timestamp = -1);
+
   std::unordered_map<std::string, std::string> mCustomParameters;
 
  private:
   // TODO should we rather have a global/singleton for the objectsManager ?
   std::shared_ptr<ObjectsManager> mObjectsManager;
   std::string mName;
+  std::shared_ptr<o2::ccdb::CcdbApi> mCcdbApi;
 };
 
 } // namespace o2::quality_control::core

Framework/readout-no-sampling.json

@@ -17,6 +17,9 @@
       },
       "consul": {
         "url": "http://consul-test.cern.ch:8500"
+      },
+      "conditionDB": {
+        "url": "ccdb-test.cern.ch:8080"
       }
     },
     "tasks": {

Framework/readout.json

@@ -17,6 +17,9 @@
       },
       "consul": {
         "url": "http://consul-test.cern.ch:8500"
+      },
+      "conditionDB": {
+        "url": "ccdb-test.cern.ch:8080"
       }
     },
     "tasks": {

Framework/src/TaskInterface.cxx

@@ -15,11 +15,16 @@
 ///
 
 #include "QualityControl/TaskInterface.h"
+#include <CCDB/CcdbApi.h>
+
+using namespace o2::ccdb;
 
 namespace o2::quality_control::core
 {
 
-TaskInterface::TaskInterface(ObjectsManager* objectsManager) : mObjectsManager(objectsManager) {}
+TaskInterface::TaskInterface(ObjectsManager* objectsManager) : mObjectsManager(objectsManager)
+{
+}
 
 const std::string& TaskInterface::getName() const { return mName; }
 
@@ -30,11 +35,33 @@ void TaskInterface::setObjectsManager(std::shared_ptr<ObjectsManager> objectsMan
   mObjectsManager = objectsManager;
 }
 
+void TaskInterface::loadCcdb(std::string url)
+{
+  if (!mCcdbApi) {
+    mCcdbApi = std::make_shared<CcdbApi>();
+  }
+
+  mCcdbApi->init(url);
+  if (!mCcdbApi->isHostReachable()) {
+    LOG(WARN) << "CCDB at URL '" << url << "' is not reachable.";
+  }
+}
+
 void TaskInterface::setCustomParameters(const std::unordered_map<std::string, std::string>& parameters)
 {
   mCustomParameters = parameters;
 }
 
+TObject* TaskInterface::retrieveCondition(std::string path, std::map<std::string, std::string> metadata, long timestamp)
+{
+  if (mCcdbApi) {
+    return mCcdbApi->retrieve(path, metadata, timestamp);
+  } else {
+    LOG(ERROR) << "Trying to retrieve a condition, but CCDB API is not constructed.";
+    return nullptr;
+  }
+}
+
 std::shared_ptr<ObjectsManager> TaskInterface::getObjectsManager() { return mObjectsManager; }
 
 } // namespace o2::quality_control::core

Framework/src/TaskRunner.cxx

@@ -79,6 +79,7 @@ void TaskRunner::init(InitContext& iCtx)
   mTask.reset(f.create(mTaskConfig, mObjectsManager));
 
   // init user's task
+  mTask->loadCcdb(mTaskConfig.conditionUrl);
   mTask->initialize(iCtx);
 }
 
@@ -253,6 +254,7 @@ void TaskRunner::populateConfig(std::string taskName)
     mTaskConfig.cycleDurationSeconds = taskConfigTree->second.get<int>("cycleDurationSeconds", 10);
     mTaskConfig.maxNumberCycles = taskConfigTree->second.get<int>("maxNumberCycles", -1);
     mTaskConfig.consulUrl = mConfigFile->get<std::string>("qc.config.consul.url", "http://consul-test.cern.ch:8500");
+    mTaskConfig.conditionUrl = mConfigFile->get<std::string>("qc.config.conditionDB.url", "http://ccdb-test.cern.ch:8080");
     try {
       mTaskConfig.customParameters = mConfigFile->getRecursiveMap("qc.tasks." + taskName + ".taskParameters");
     } catch (...) {