TimeoutAlg.h

Go to the documentation of this file.

/*
 * Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration.
 */

 
#ifndef ATHENASERVICES_TIMEOUTALG_H
#define ATHENASERVICES_TIMEOUTALG_H
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "AthenaKernel/SlotSpecificObj.h"
#include "AthenaKernel/Timeout.h"
#include "CxxUtils/checker_macros.h"
#include "GaudiKernel/IIncidentListener.h"
 
#include <chrono>
#include <future>
#include <thread>
 

class TimeoutAlg : public extends<AthReentrantAlgorithm, IIncidentListener>,
                   public Athena::TimeoutMaster
{
public:
  using base_class::base_class;
 
  virtual StatusCode initialize() override;
  virtual StatusCode execute (const EventContext& ctx) const override;
  virtual StatusCode stop() override;
  virtual void handle(const Incident& inc) override;
 
private:
  using clock_t = std::chrono::steady_clock;

  void timeoutThread();

  void handleTimeout(EventContext::ContextID_t slot);

  Gaudi::Property<unsigned long long> m_timeoutProp{
    this, "Timeout", 0, "Timeout in nanoseconds (0 means disabled)"
  };
  Gaudi::Property<unsigned long long> m_checkInterval{
    this, "MaxCheckInterval", 10*1e9, "Maximum time (ns) between timeout checks"
  };
  Gaudi::Property<bool> m_dumpState{
    this, "DumpSchedulerState", false, "Print scheduler state on timeout"
  };
  Gaudi::Property<bool> m_abort{
    this, "AbortJob", false, "Abort job on timeout"
  };


  std::chrono::nanoseconds m_timeout;

  mutable SG::SlotSpecificObj<clock_t::time_point> m_eventStartTime ATLAS_THREAD_SAFE;

  mutable std::thread m_thread ATLAS_THREAD_SAFE;

  std::promise<void> m_stop_thread;

  std::atomic<bool> m_stopped{false};

  std::mutex m_handleMutex;
};
 
#endif