DataWriterExt.cc

Go to the documentation of this file.

#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include "CondTools/L1TriggerExt/interface/DataWriterExt.h"
#include "CondTools/L1Trigger/interface/Exception.h"
#include "CondCore/CondDB/interface/Exception.h"

#include "CondCore/CondDB/interface/Serialization.h"

#include <utility>

namespace l1t
{
  DataWriterExt::DataWriterExt(){}
  DataWriterExt::~DataWriterExt(){}



std::string
DataWriterExt::writePayload( const edm::EventSetup& setup,
              const std::string& recordType )
{
  WriterFactory* factory = WriterFactory::get();
  std::unique_ptr<WriterProxy> writer(factory->create( recordType + "@Writer" )) ;
  if( writer.get() == nullptr )
    {
      throw cond::Exception( "DataWriter: could not create WriterProxy with name "
                 + recordType + "@Writer" ) ;
    }

  edm::Service<cond::service::PoolDBOutputService> poolDb;
  if (!poolDb.isAvailable())
    {
      throw cond::Exception( "DataWriter: PoolDBOutputService not available."
                 ) ;
    }

  // 2010-02-16: Move session and transaction to WriterProxy::save().  Otherwise, if another transaction is
  // started while WriterProxy::save() is called (e.g. in a ESProducer like L1ConfigOnlineProdBase), the
  // transaction here will become read-only.
//   cond::DbSession session = poolDb->session();
//   cond::DbScopedTransaction tr(session);

//   // if throw transaction will unroll
//   tr.start(false);

  // update key to have new payload registered for record-type pair.
  //  std::string payloadToken = writer->save( setup, session ) ;
  std::string payloadToken = writer->save( setup ) ;

  edm::LogVerbatim( "L1-O2O" ) << recordType << " PAYLOAD TOKEN "
                   << payloadToken ;

//   tr.commit ();

  return payloadToken ;
}

void
DataWriterExt::writeKeyList( L1TriggerKeyListExt* keyList,
              edm::RunNumber_t sinceRun,
              bool logTransactions )
{
  edm::Service<cond::service::PoolDBOutputService> poolDb;
  if( !poolDb.isAvailable() )
    {
      throw cond::Exception( "DataWriter: PoolDBOutputService not available."
                 ) ;
    }

  poolDb->forceInit();
  cond::persistency::Session session = poolDb->session();
  cond::persistency::TransactionScope tr(session.transaction());

  // Write L1TriggerKeyListExt payload and save payload token before committing
  std::shared_ptr<L1TriggerKeyListExt> pointer(keyList);
  std::string payloadToken = session.storePayload(*pointer );
            
  // Commit before calling updateIOV(), otherwise PoolDBOutputService gets
  // confused.
  tr.close ();
  
  // Set L1TriggerKeyListExt IOV
  updateIOV( "L1TriggerKeyListExtRcd",
         payloadToken,
         sinceRun,
         logTransactions ) ;
}

bool
DataWriterExt::updateIOV( const std::string& esRecordName,
               const std::string& payloadToken,
               edm::RunNumber_t sinceRun,
               bool logTransactions )
{
  edm::LogVerbatim( "L1-O2O" ) << esRecordName
                   << " PAYLOAD TOKEN " << payloadToken ;

  edm::Service<cond::service::PoolDBOutputService> poolDb;
  if (!poolDb.isAvailable())
    {
      throw cond::Exception( "DataWriter: PoolDBOutputService not available."
                 ) ;
    }

  bool iovUpdated = true ;

  if( poolDb->isNewTagRequest( esRecordName ) )
    {
      sinceRun = poolDb->beginOfTime() ;
      poolDb->createNewIOV( payloadToken,
                sinceRun,
                poolDb->endOfTime(),
                esRecordName,
                logTransactions ) ;
    }
  else
    {   
      cond::TagInfo tagInfo ;
      poolDb->tagInfo( esRecordName, tagInfo ) ;

      if( sinceRun == 0 ) // find last since and add 1
    {
      sinceRun = tagInfo.lastInterval.first ;
      ++sinceRun ;
    }

      if( tagInfo.lastPayloadToken != payloadToken )
    {
      poolDb->appendSinceTime( payloadToken,
                   sinceRun,
                   esRecordName,
                   logTransactions ) ;
    }
      else
    {
      iovUpdated = false ;
      edm::LogVerbatim( "L1-O2O" ) << "IOV already up to date." ;
    }
    }

  if( iovUpdated )
    {
      edm::LogVerbatim( "L1-O2O" ) << esRecordName << " "
                   << poolDb->tag( esRecordName )
                   << " SINCE " << sinceRun ;
    }

  return iovUpdated ;
}

std::string
DataWriterExt::payloadToken( const std::string& recordName,
              edm::RunNumber_t runNumber )
{
  edm::Service<cond::service::PoolDBOutputService> poolDb;
  if( !poolDb.isAvailable() )
    {
      throw cond::Exception( "DataWriter: PoolDBOutputService not available."
                 ) ;
    }

  // Get tag corresponding to EventSetup record name.
  std::string iovTag = poolDb->tag( recordName ) ;

  // Get IOV token for tag.
  cond::persistency::Session session = poolDb->session();
  cond::persistency::IOVProxy iov = session.readIov( iovTag );
  session.transaction().start();

  std::string payloadToken("");
  auto iP = iov.find( runNumber );
  if( iP != iov.end() ){
    payloadToken = (*iP).payloadId; 
  }
  session.transaction().commit() ;
  return payloadToken ;
}

std::string
DataWriterExt::lastPayloadToken( const std::string& recordName )
{
  edm::Service<cond::service::PoolDBOutputService> poolDb;
  if( !poolDb.isAvailable() )
    {
      throw cond::Exception( "DataWriter: PoolDBOutputService not available."
                 ) ;
    }

  cond::TagInfo tagInfo ;
  poolDb->tagInfo( recordName, tagInfo ) ;
  return tagInfo.lastPayloadToken ;
}

bool
DataWriterExt::fillLastTriggerKeyList( L1TriggerKeyListExt& output )
{
  std::string keyListToken =
    lastPayloadToken( "L1TriggerKeyListExtRcd" ) ;
  if( keyListToken.empty() )
    {
      return false ;
    }
  else
    {
      readObject( keyListToken, output ) ;
      return true ;
    }
}

} // ns