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;
    }
  }

}  // namespace l1t