PoolSvc.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef POOLSVC_H
#define POOLSVC_H

 
#include "PoolSvc/IPoolSvc.h"
#include "PoolSvc/FileCatalogUtils.h"
#include "GaudiKernel/IIoComponent.h"
#include "GaudiKernel/SmartIF.h"
#include "AthenaBaseComps/AthService.h"
#include "CxxUtils/checker_macros.h"
#include "PersistentDataModel/Guid.h"
 
#include <string>
#include <vector>
#include <map>
#include <mutex>
#include <memory>
 
// Forward declarations
namespace pool {
   class IContainer;
   class IDatabase;
   class ISession;
}
class SessionTool;

class PoolSvc : public extends<AthService, IPoolSvc, IIoComponent> {
 
public: // Non-static members
   virtual StatusCode initialize() override;
   virtual StatusCode io_reinit() override;
   virtual StatusCode start() override;
   virtual StatusCode stop() override;
   virtual StatusCode finalize() override;
   virtual StatusCode io_finalize() override;

   virtual
   Token* registerForWrite(const Placement* placement,
                           const void* obj,
                           const RootType& classDesc) override;

   virtual
   void setObjPtr(void*& obj, const Token* token) override;

   virtual
   unsigned int getOutputContext(const std::string& label) override;

   virtual
   unsigned int getInputContext(const std::string& label,
                                unsigned int maxFile = 0) override;

   virtual
   std::map<std::string, unsigned int> getInputContextMap() const override;

   virtual
   unsigned int getInputContextMapSize() const override;

   virtual
   pool::FileDescriptor getInputFileDescriptor(unsigned int contextId,
                                               const std::string& connection) override;

   virtual
   void setShareMode(bool shareCat) override;

   virtual
   void startCatalog() override;

   virtual
   void commitCatalog() override;

   virtual
   void lookupBestPfn(const std::string& dbID, std::string& pfn, std::string& type) const override;

   virtual
   StatusCode connectCollection(const std::string& connection,
           const std::string& collectionName,
           unsigned int contextId = IPoolSvc::kInputStream) override;

   virtual
   StatusCode checkCollection(const std::string& connection,
           unsigned int contextId,
           bool noContainer) override;

   virtual
   Token* getToken(const std::string& connection,
       const std::string& collection,
       const unsigned long ientry) override;

   virtual
   StatusCode connect(Io::IoFlag type,
       unsigned int contextId = IPoolSvc::kInputStream) override;

   virtual
   StatusCode commit(unsigned int contextId = IPoolSvc::kInputStream) override;

   virtual
   StatusCode commitAndHold(unsigned int contextId = IPoolSvc::kInputStream) override;

   virtual
   StatusCode disconnect(unsigned int contextId = IPoolSvc::kInputStream) override;

   virtual
   StatusCode disconnectDb(const std::string& connection,
       unsigned int contextId = IPoolSvc::kInputStream) override;

   virtual
   StatusCode getAttribute(const std::string& optName,
       std::string& data,
       long tech,
       unsigned int contextId = IPoolSvc::kInputStream) override;

   virtual
   StatusCode getAttribute(const std::string& optName,
       std::string& data,
       long tech,
       const std::string& dbName,
       const std::string& contName = "",
       unsigned int contextId = IPoolSvc::kInputStream) override;

   virtual
   StatusCode setAttribute(const std::string& optName,
       const std::string& data,
       long tech,
       unsigned int contextId = IPoolSvc::kOutputStream) override;

   virtual
   StatusCode setAttribute(const std::string& optName,
       const std::string& data,
       long tech,
       const std::string& dbName,
       const std::string& contName = "",
       unsigned int contextId = IPoolSvc::kOutputStream) override;

   StatusCode createSession( const std::string& sessionName, int ageLimit = -1 );
   
   using base_class::base_class;

   virtual ~PoolSvc();
 
private: // data
   using CallMutex = std::recursive_mutex;
   // Lock Guard class to safely lock a mutex for a given contextId
   class ContextLock {
      std::unique_lock< CallMutex > m_lock;
   public:
      ContextLock(int contextId, CallMutex &glob_mtx, const std::vector<CallMutex*> &ctx_mutexes) {
         // lock the global mutex to gain exclusive access to the context mutexes vector
         std::lock_guard<CallMutex>    temp_lock( glob_mtx );
         // lock the mutex for the given context ID for the lifetime of this object
         m_lock = std::unique_lock< CallMutex >{ *ctx_mutexes[contextId] };
      }
      ~ContextLock() { m_lock.unlock(); }
   };
 
   mutable CallMutex                                 m_pool_mut;
 
   bool                                              m_shareCat{false};
   SmartIF<Gaudi::IFileCatalogMgr>                   m_catalogMgr;
   SmartIF<Gaudi::IFileCatalog>                      m_catalog;
   std::vector<ToolHandle<SessionTool>>              m_dbSessionVec;
 
   std::vector<CallMutex*>                           m_pers_mut;
   std::map<std::string, unsigned int>               m_inputContextLabel;
   std::map<std::string, unsigned int>               m_outputContextLabel;
   std::string                                       m_mainOutputLabel{};
   std::map<unsigned int, unsigned int>              m_contextMaxFile;
   // Cache for open file guids for each m_dbsessionVec member, protected by m_pers_mut
   mutable std::map<unsigned int, std::list<Guid> >  m_guidLists ATLAS_THREAD_SAFE;
 
private: // properties
   Gaudi::Property<int> m_dbAgeLimit{this,"MaxFilesOpen",0};
   Gaudi::Property<std::string> m_writeCatalog{this,"WriteCatalog","xmlcatalog_file:PoolFileCatalog.xml"};
   Gaudi::Property<std::vector<std::string>> m_readCatalog{this,"ReadCatalog",{},"List of catalog files to read from","OrderedSet<std::string>"};
   Gaudi::Property<bool> m_useROOTIMT{this,"UseROOTImplicitMT",true};
   Gaudi::Property<bool> m_useROOTMaxTree{this,"UseROOTIncreaseVMaxTree",false};

   Gaudi::Property<bool> m_attemptCatalogPatch{this,"AttemptCatalogPatch",true};
 
 
private: // internal helper functions
   // delete all Persistency Services, Catalog, Mutexes and Indexes
   void clearState();
 
   // Set up m_catalogMgr/m_catalog. Returns false on failure.
   bool createCatalog();
   void patchCatalog(const std::string& pfn, pool::IDatabase& dbH) const;
 
   // setup persistency
   StatusCode setupPersistencySvc();

   std::unique_ptr<pool::IDatabase> getDbHandle(unsigned int contextId, const std::string& dbName);
   std::unique_ptr<pool::IContainer> getContainerHandle(pool::IDatabase* dbH, const std::string& contName);

   std::string poolCondPath(const std::string& leaf);
};
#endif