TimeoutMonitor.h

/*
Copyright (c) 2017-2024,
Battelle Memorial Institute; Lawrence Livermore National Security, LLC; Alliance for Sustainable
Energy, LLC.  See the top-level NOTICE for additional details. All rights reserved.
SPDX-License-Identifier: BSD-3-Clause
*/
#pragma once
#include "GlobalFederateId.hpp"
 
#include <chrono>
#include <vector>
 
namespace helics {
class CommonCore;
class CoreBroker;
class ActionMessage;
 
struct linkConnection {
    bool waitingForPingReply{false};  
    bool activeConnection{false};  
    bool disablePing{false};  
    GlobalFederateId connection{0};  
    decltype(std::chrono::steady_clock::now()) lastPing;
};
class TimeoutMonitor {
  public:
    void tick(CommonCore* core);
    void tick(CoreBroker* brk);
    void pingReply(const ActionMessage& cmd, CoreBroker* brk = nullptr);
    void setTimeout(std::chrono::milliseconds to) { timeout = to; }
    void reset();
    void pingSub(CoreBroker* brk);
    void setParentId(GlobalBrokerId parent_id) { parentConnection.connection = parent_id; }
    void disableParentPing(bool value = true)
    {
        parentConnection.disablePing = value;
        parentConnection.waitingForPingReply = false;
    }
 
  private:
    std::chrono::milliseconds timeout{100'000'000};  
    bool waitingForConnection{false};  
    decltype(std::chrono::steady_clock::now()) startWaiting;  
    linkConnection parentConnection;  
    std::vector<linkConnection> connections;  
 
    // int tickCounter;
};
 
}  // namespace helics