CommsBroker.hpp

/*
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 "helics/core/ActionMessage.hpp"
 
#include <atomic>
#include <memory>
#include <string>
 
namespace helics {
class CommsInterface;
 
template<class COMMS, class BrokerT>
class CommsBroker: public BrokerT {
  protected:
    std::atomic<int> disconnectionStage{0};  
    std::unique_ptr<COMMS> comms;  
    std::atomic<bool> brokerInitialized{false};  
  public:
    CommsBroker() noexcept;
    explicit CommsBroker(bool arg) noexcept;
    explicit CommsBroker(std::string_view obj_name);
    ~CommsBroker();
 
  private:
    virtual void brokerDisconnect() override;
    virtual bool tryReconnect() override;
    void commDisconnect();
    void loadComms();
 
  public:
    virtual void transmit(route_id rid, const ActionMessage& cmd) override;
    virtual void transmit(route_id rid, ActionMessage&& cmd) override;
 
    virtual void addRoute(route_id rid, int interfaceId, std::string_view routeInfo) override;
 
    virtual void removeRoute(route_id rid) override;
    COMMS* getCommsObjectPointer();
};
}  // namespace helics