MultiBroker.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 "../core/CoreBroker.hpp"
#include "../network/NetworkBroker.hpp"
 
#include <memory>
#include <string>
#include <utility>
#include <vector>
 
namespace helics {
class ActionMessage;
class CommsInterface;
 
// small function to call to force symbol linkages
bool allowMultiBroker();
 
class MultiBroker: public CoreBroker {
  protected:
    std::atomic<int> disconnectionStage{0};  
    std::vector<std::unique_ptr<CommsInterface>> comms;  
    std::unique_ptr<CommsInterface> masterComm;
    NetworkBrokerData netInfo{gmlc::networking::InterfaceTypes::TCP};
    std::string configFile;  
    std::atomic<bool> brokerInitialized{false};  
    CoreType type{CoreType::MULTI};  
    std::vector<std::pair<route_id, int>> routingTable;  // index of the routes
  public:
    MultiBroker() noexcept;
 
    explicit MultiBroker(std::string_view brokerName);
 
    ~MultiBroker();
 
  private:
    virtual bool brokerConnect() override;
    virtual void brokerDisconnect() override;
    virtual bool tryReconnect() override;
    virtual std::shared_ptr<helicsCLI11App> generateCLI() override;
 
  protected:
    virtual std::string generateLocalAddressString() const override;
 
  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;
};
}  // namespace helics