RDBAccessSvc.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/

 
#ifndef RDBACCESSSVC_RDBACCESSSVC_H
#define RDBACCESSSVC_RDBACCESSSVC_H
 
#include "RDBAccessSvc/IRDBAccessSvc.h"
#include "RDBRecordset.h"
 
#include "AthenaBaseComps/AthService.h"
 
#include <string>
#include <string_view>
#include <map>
 
class ISvcLocator;
class RDBRecordset;
 
namespace coral 
{
  class ISessionProxy;
}
 
template <class TYPE> class SvcFactory;
 
 
// Recordsets of single connection
typedef std::map<std::string, IRDBRecordset_ptr, std::less<>> RecordsetPtrMap;
 
// Pointers to recordset maps by connection name
typedef std::map<std::string, RecordsetPtrMap, std::less<>> RecordsetPtrsByConn;
 
// Session map
typedef std::map<std::string, coral::ISessionProxy*, std::less<>> SessionMap;
 
// Lookup table for global tag contents quick access
typedef std::pair<std::string, std::string> TagNameId;
typedef std::map<std::string, TagNameId, std::less<>> TagNameIdByNode;
typedef std::map<std::string, TagNameIdByNode*, std::less<>> GlobalTagLookupMap; // Key - <Global_Tag_Name>::<Connection>

 
class RDBAccessSvc final : public extends<AthService, IRDBAccessSvc>
{
 public:
  RDBAccessSvc(const std::string& name, ISvcLocator* svc);
 
  StatusCode initialize() override;
  StatusCode finalize() override;

  bool connect(std::string_view connName) override;

  bool disconnect(std::string_view connName) override;

  bool shutdown(std::string_view connName) override;

  IRDBRecordset_ptr getRecordsetPtr(std::string_view node
                    , std::string_view tag
                    , std::string_view tag2node=""
                    , std::string_view connName = "ATLASDD") override;

  std::string getChildTag(const std::string& childNode
              , const std::string& parentTag
              , const std::string& parentNode
              , const std::string& connName) override;
 
  std::string getChildTag(const std::string& childNode
              , const std::string& parentTag
              , const std::string& parentNode
              , const std::string& connName
              , bool force);
 
  std::unique_ptr<IRDBQuery> getQuery(const std::string& node
                      , const std::string& tag
                      , const std::string& tag2node
                      , const std::string& connName) override;
 
  void getTagDetails(RDBTagDetails& tagDetails
             , const std::string& tag
             , const std::string& connName = "ATLASDD") override;
 
  void getAllLeafNodes(std::vector<std::string>& list
               , const std::string& connName = "ATLASDD");
 
  std::vector<std::string> getLockedSupportedTags(const std::string& supportedFlag
                          , const std::string& connName = "ATLASDD");
 
  coral::ISessionProxy* getSession(const std::string& connName = "ATLASDD");
 
private:
  SessionMap m_sessions;
  std::map<std::string, unsigned int, std::less<>> m_openConnections;
 
  RecordsetPtrsByConn m_recordsetptrs;  
  GlobalTagLookupMap m_globalTagLookup;
 
  std::mutex m_recordsetMutex;
  std::mutex m_sessionMutex;
 
  bool shutdown_connection(std::string_view connName);
};
 
#endif