implement CCDB and QCDB service access in Actor

Author: knopers8

Framework/include/QualityControl/Actor.h

@@ -30,6 +30,15 @@ namespace o2::bkp {
 enum class DplProcessType;
 }
 
+
+namespace o2::quality_control::repository {
+class DatabaseInterface;
+}
+
+namespace o2::ccdb {
+class CCDBManagerInstance;
+}
+
 namespace o2::quality_control::core {
 
 
@@ -46,6 +55,11 @@ void initBookkeeping(std::string_view url);
 void startBookkeeping(int runNumber, std::string_view actorName, std::string_view detectorName, const o2::bkp::DplProcessType& processType, std::string_view args);
 Bookkeeping& getBookkeeping();
 
+std::shared_ptr<repository::DatabaseInterface> initRepository(const std::unordered_map<std::string, std::string>& config);
+
+void initCCDB(const std::string& url);
+ccdb::CCDBManagerInstance& getCCDB();
+
 }
 
 
@@ -132,7 +146,6 @@ class Actor
       detectorName = std::string{concreteActor().getDetectorName()};
     }
 
-    // init services
     if constexpr (requiresService<Service::InfoLogger>()) {
       std::string facility;
       if constexpr (runsUserCode<traits>()) {
@@ -150,7 +163,12 @@ class Actor
     if constexpr (requiresService<Service::Bookkeeping>()) {
       impl::initBookkeeping(mServicesConfig.bookkeepingUrl);
     }
-    // todo: init the rest, such as QCDB
+    if constexpr (requiresService<Service::QCDB>()) {
+      mRepository = impl::initRepository(mServicesConfig.database);
+    }
+    if constexpr (requiresService<Service::CCDB>()) {
+      impl::initCCDB(mServicesConfig.conditionDBUrl);
+    }
   }
 
   void initDplCallbacks(framework::InitContext& ictx)
@@ -272,13 +290,24 @@ class Actor
     return impl::getBookkeeping();
   }
 
+  repository::DatabaseInterface& getRepository() requires (requiresService<Service::QCDB>())
+  {
+    return *mRepository;
+  }
+
+  ccdb::CCDBManagerInstance& getCCDB() requires (requiresService<Service::QCDB>())
+  {
+    return impl::getCCDB();
+  }
+
   const Activity& getActivity() const { return mActivity; }
 
   private:
   Activity mActivity;
   const ServicesConfig mServicesConfig;
 
   std::shared_ptr<monitoring::Monitoring> mMonitoring;
+  std::shared_ptr<repository::DatabaseInterface> mRepository;
 };
 
 

Framework/include/QualityControl/ActorTraits.h

@@ -52,7 +52,8 @@ enum class Service {
   Monitoring,
   InfoLogger,
   CCDB,
-  Bookkeeping
+  Bookkeeping,
+  QCDB
 };
 
 template<typename ActorTraitsT, Service S>

Framework/src/Actor.cxx

@@ -6,14 +6,14 @@
 #include "QualityControl/Actor.h"
 
 #include "QualityControl/Bookkeeping.h"
+#include "QualityControl/DatabaseFactory.h"
 #include <Monitoring/Monitoring.h>
 #include <Monitoring/MonitoringFactory.h>
-
+#include <CCDB/BasicCCDBManager.h>
 
 namespace o2::quality_control::core
 {
 
-
 namespace impl
 {
 std::shared_ptr<monitoring::Monitoring> initMonitoring(std::string_view url, std::string_view detector)
@@ -48,6 +48,28 @@ Bookkeeping& getBookkeeping()
   return Bookkeeping::getInstance();
 }
 
+std::shared_ptr<quality_control::repository::DatabaseInterface> initRepository(const std::unordered_map<std::string, std::string>& config)
+{
+  auto db = quality_control::repository::DatabaseFactory::create(config.at("implementation"));
+  assert(db != nullptr);
+  db->connect(config);
+  ILOG(Info, Devel) << "Database that is going to be used > Implementation : " << config.at("implementation") << " / Host : " << config.at("host") << ENDM;
+  return std::move(db);
+}
+
+void initCCDB(const std::string& url)
+{
+  auto& mgr = o2::ccdb::BasicCCDBManager::instance();
+  mgr.setURL(url);
+  mgr.setFatalWhenNull(false);
+}
+
+ccdb::CCDBManagerInstance& getCCDB()
+{
+  return o2::ccdb::BasicCCDBManager::instance();
+}
+
+
 }
 
 }