FWIO/RNTupleTempOutput/src/TimeoutRNTupleTempOutputModule.cc - master - githubmirror/cmssw

FWIO/RNTupleTempOutput/src/TimeoutRNTupleTempOutputModule.cc

55 строк2 KB

Christopher Jones

All RNTupleTemp related files

03 дек 2025, 00:03

03 дек 2025, 00:03fd9ee5f

#include "FWIO/RNTupleTempOutput/interface/TimeoutRNTupleTempOutputModule.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"

namespace edm::rntuple_temp {

  void TimeoutRNTupleTempOutputModule::write(EventForOutput const& e) {
    eventsWrittenInCurrentFile++;
    RNTupleTempOutputModule::write(e);
  }

  TimeoutRNTupleTempOutputModule::TimeoutRNTupleTempOutputModule(ParameterSet const& ps)
      : edm::one::OutputModuleBase::OutputModuleBase(ps),
        RNTupleTempOutputModule(ps),
        m_lastEvent(time(nullptr)),
        eventsWrittenInCurrentFile(0),
        m_timeout(-1)  // we want the first event right away
  {}

  void TimeoutRNTupleTempOutputModule::fillDescriptions(ConfigurationDescriptions& descriptions) {
    ParameterSetDescription desc;
    RNTupleTempOutputModule::fillDescription(desc);
    descriptions.add("TimeoutRNTupleTempOutputModule", desc);
  }

  bool TimeoutRNTupleTempOutputModule::shouldWeCloseFile() const {
    time_t now(time(nullptr));
    if (RNTupleTempOutputModule::shouldWeCloseFile()) {
      edm::LogVerbatim("TimeoutRNTupleTempOutputModule")
          << " Closing file " << currentFileName() << " with " << eventsWrittenInCurrentFile << " events.";
      eventsWrittenInCurrentFile = 0;
      m_lastEvent = now;
      return true;
    }
    //    std::cout <<" Events "<< eventsWrittenInCurrentFile<<" time "<< now - m_lastEvent<<std::endl;
    if (eventsWrittenInCurrentFile == 0)
      return false;
    if (now - m_lastEvent < m_timeout)
      return false;
    // next files are needed in 15, 30 and 60 sec
    m_lastEvent = now;
    if (m_timeout == 30)
      m_timeout = 60;
    if (m_timeout == 15)
      m_timeout = 30;
    if (m_timeout == -1)
      m_timeout = 15;

    edm::LogVerbatim("TimeoutRNTupleTempOutputModule")
        << " Closing file " << currentFileName() << " with " << eventsWrittenInCurrentFile << " events.";
    eventsWrittenInCurrentFile = 0;
    return true;
  }
}  // namespace edm::rntuple_temp