PoolDBOutputService.h

Go to the documentation of this file.

#ifndef CondCore_PoolDBOutputService_h
#define CondCore_PoolDBOutputService_h
#include "FWCore/ServiceRegistry/interface/ActivityRegistry.h"
#include "FWCore/Utilities/interface/TypeID.h"
#include "CondCore/DBCommon/interface/Time.h"
#include "CondCore/MetaDataService/interface/MetaData.h"
#include "CondCore/DBCommon/interface/Logger.h"
#include "CondCore/DBCommon/interface/LogDBEntry.h"
#include "CondCore/DBCommon/interface/TagInfo.h"
#include <string>
#include <map>

// many many clients do not include explicitely!
#ifndef COND_EXCEPTION_H
#include "CondCore/DBCommon/interface/Exception.h"
// #warning please include  "CondCore/DBCommon/interface/Exception.h" explicitely
// #define COND_EXP_WARNING
#endif

//
// Package:     DBOutputService
// Class  :     PoolDBOutputService
// 
//
// Author:      Zhen Xie
// Fixes and other changes: Giacomo Govi
//
namespace edm{
  class Event;
  class EventSetup;
  class ParameterSet;
}
namespace cond{
  
  inline std::string classNameForTypeId( const std::type_info& typeInfo ){
    edm::TypeID type( typeInfo );
    return type.className();
  }
  
  namespace service {

    struct GetToken {
      virtual std::string operator()(cond::DbSession&) const =0;
    };
    
    struct GetTrivialToken : public GetToken {
      
      GetTrivialToken(std::string token) :
    m_token(token){}
      virtual ~GetTrivialToken(){}
      virtual std::string operator()(cond::DbSession&) const {
    return m_token;
      }
      std::string m_token;
    };
    
    template<typename T>
    struct GetTokenFromPointer : public GetToken {
      
      static
      std::string classNameForPointer( T* pointer ){
        if(!pointer) return classNameForTypeId( typeid(T) );
        return classNameForTypeId( typeid(*pointer) );
      }

      explicit GetTokenFromPointer(T * p) :
    m_p(p){}
      
      virtual std::string operator()(cond::DbSession& pooldb) const {
    std::string className = classNameForPointer( m_p );
    boost::shared_ptr<T> sptr( m_p );
    return pooldb.storeObject(m_p,className);
      }
      T* m_p;
    };

    
    class PoolDBOutputService{
    public:
      PoolDBOutputService( const edm::ParameterSet & iConfig, 
               edm::ActivityRegistry & iAR );
      //use these to control connections
      //void  postBeginJob();
      void  postEndJob();
      //
      //use these to control transaction interval
      //
      void preEventProcessing( const edm::EventID & evtID, 
                       const edm::Timestamp & iTime );
      void preModule(const edm::ModuleDescription& desc);
      void postModule(const edm::ModuleDescription& desc);
      void preBeginLumi(const edm::LuminosityBlockID&, 
            const edm::Timestamp& );
      //
      // return the database session in use
      //
      cond::DbSession session() const;
      //
      std::string tag( const std::string& recordName );
      bool isNewTagRequest( const std::string& recordName );
      const cond::Logger& queryLog() const;
      
      // 
      template<typename T>
      void writeOne( T * payload, Time_t time, const std::string& recordName, bool withlogging=false ) {
    if (isNewTagRequest(recordName) ){
      createNewIOV<T>(payload, time, endOfTime(), recordName, withlogging);
        }else{
      appendSinceTime<T>(payload, time, recordName, withlogging);
        }   
      }

      // close the IOVSequence setting lastTill
      void closeIOV(Time_t lastTill, const std::string& recordName, 
                    bool withlogging=false);

      // 
      template<typename T>
      void createNewIOV( T* firstPayloadObj,
             cond::Time_t firstSinceTime,
             cond::Time_t firstTillTime,
             const std::string& recordName,
                         bool withlogging=false){
        createNewIOV( GetTokenFromPointer<T>(firstPayloadObj),
                      firstSinceTime,
                      firstTillTime,
                      recordName,
                      withlogging); 
      }
            
      void createNewIOV( const std::string& firstPayloadToken,
                         cond::Time_t firstSinceTime,
                         cond::Time_t firstTillTime,
                         const std::string& recordName,
                         bool withlogging=false) {
        createNewIOV( GetTrivialToken(firstPayloadToken),
                      firstSinceTime,
                      firstTillTime,
                      recordName,
                      withlogging);
      }

      
      // 
      template<typename T> void appendSinceTime( T* payloadObj,
                                                 cond::Time_t sinceTime,
                                                 const std::string& recordName,
                                                 bool withlogging=false){
        add( GetTokenFromPointer<T>(payloadObj),
         sinceTime,
         recordName,
         withlogging);
      }
      
      // Append the payload and its valid sinceTime into the database
      // Note: user looses the ownership of the pointer to the payloadObj
      // Note: the iov index appended to MUST pre-existing and the existing 
      // conditions data are retrieved from the DB
      // 
      void appendSinceTime( const std::string& payloadToken,
                            cond::Time_t sinceTime,
                            const std::string& recordName,
                            bool withlogging=false) {
        add(GetTrivialToken(payloadToken),
            sinceTime,
            recordName,
            withlogging);
      }
     
      // set last till so that the iov sequence is "closed"
      // void closeSequence(cond::Time_t tillTime,
      //                 const std::string& recordName,
      //                 bool withlogging=false);


      //
      // Service time utility method 
      // return the infinity value according to the given timetype
      //
      cond::Time_t endOfTime() const;
      //
      // Service time utility method 
      // return beginning of time value according to the given timetype
      //
      cond::Time_t beginOfTime() const;
      //
      // Service time utility method 
      // return the time value of the current edm::Event according to the 
      // given timetype
      //
      cond::Time_t currentTime() const;

      // optional. User can inject additional information into the log associated with a given record
      void setLogHeaderForRecord(const std::string& recordName,
                 const std::string& provenance,
                 const std::string& usertext);
      // 
      // Retrieve tag information of the data
      // 
      void tagInfo(const std::string& recordName,
           cond::TagInfo& result );
      
      virtual ~PoolDBOutputService();  
      
    private:

      struct Record{
    Record(): m_tag(),
          m_isNewTag(false),
          m_idName(),
          m_iovtoken(),
          m_timetype(cond::runnumber),
                  m_closeIOV(false),
          m_freeInsert(false)
    {}

    std::string timetypestr() const { return cond::timeTypeSpecs[m_timetype].name;}
    std::string m_tag;
    bool m_isNewTag;
    std::string m_idName;
    std::string m_iovtoken;
    cond::TimeType m_timetype;
        bool m_closeIOV;
    bool m_freeInsert;
    };      



      void fillRecord( edm::ParameterSet & pset);
      
      void createNewIOV( GetToken const & token, 
             cond::Time_t firstSinceTime, 
             cond::Time_t firstTillTime,
             const std::string& recordName,
             bool withlogging=false);
      
      void add( GetToken const & token,  
        cond::Time_t time,
        const std::string& recordName,
        bool withlogging=false);      
      
      void connect();    
      void disconnect();
      void initDB( bool forReading=true );
      unsigned int appendIOV(cond::DbSession&,
                             Record& record,
                             const std::string& payloadToken,
                             cond::Time_t sinceTime);
      
      unsigned int 
      insertIOV(cond::DbSession& pooldb,
        Record& record,
        const std::string& payloadToken,
        cond::Time_t tillTime);

      Record & lookUpRecord(const std::string& recordName);
      cond::UserLogInfo& lookUpUserLogInfo(const std::string& recordName);
      
    private:
      cond::TimeType m_timetype; 
      std::string m_timetypestr;
      cond::Time_t m_currentTime;

      std::string m_connectionString;
      cond::DbSession m_session;
      std::string m_logConnectionString;
      std::auto_ptr<cond::Logger> m_logdb;
      bool m_dbstarted;

      std::map<std::string, Record> m_callbacks;
      std::vector< std::pair<std::string,std::string> > m_newtags;
      bool m_closeIOV;
      bool m_freeInsert;
      std::map<std::string, cond::UserLogInfo> m_logheaders;

    };//PoolDBOutputService
  }//ns service
}//ns cond
#endif