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 <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> RecordsetPtrMap;
 
// Pointers to recordset maps by connection name
typedef std::map<std::string, RecordsetPtrMap> RecordsetPtrsByConn;
 
// Session map
typedef std::map<std::string, coral::ISessionProxy*> SessionMap;
 
// Lookup table for global tag contents quick access
typedef std::pair<std::string, std::string> TagNameId;
typedef std::map<std::string, TagNameId> TagNameIdByNode;
typedef std::map<std::string, TagNameIdByNode*> GlobalTagLookupMap; // Key - <Global_Tag_Name>::<Connection>
 
class RDBAccessSvc final : public AthService, virtual public IRDBAccessSvc 
{
 public:
  RDBAccessSvc(const std::string& name, ISvcLocator* svc);
 
  StatusCode initialize() override;
  StatusCode finalize() override;
  StatusCode queryInterface( const InterfaceID& riid, void** ppvInterface ) override;
 
  friend class SvcFactory<RDBAccessSvc>;
 
  static const InterfaceID& interfaceID() { return IID_IRDBAccessSvc; }
 
  bool connect(const std::string& connName) override;
 
  bool disconnect(const std::string& connName) override;
 
  bool shutdown(const std::string& connName) override;
 
  IRDBRecordset_ptr getRecordsetPtr(const std::string& node
                    , const std::string& tag
                    , const std::string& tag2node=""
                    , const std::string& 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> m_openConnections;
 
  RecordsetPtrsByConn m_recordsetptrs;  
  GlobalTagLookupMap m_globalTagLookup;
 
  std::mutex m_recordsetMutex;
  std::mutex m_sessionMutex;
 
  bool shutdown_connection(const std::string& connName);
};
 
#endif