ZmqCommsSS.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 "../NetworkCommsInterface.hpp"
 
#include <atomic>
#include <map>
#include <set>
#include <string>
 
namespace zmq {
class message_t;
class socket_t;
}  // namespace zmq
 
namespace helics {
namespace zeromq {
    class ZmqCommsSS final: public NetworkCommsInterface {
      public:
        ZmqCommsSS() noexcept;
        ~ZmqCommsSS() final;
        virtual void loadNetworkInfo(const NetworkBrokerData& netInfo) override;
      private:
        virtual int getDefaultBrokerPort() const override;
        virtual void queue_rx_function() override;  
        virtual void queue_tx_function() override;  
        int processIncomingMessage(zmq::message_t& msg,
                                   std::map<std::string, std::string>& connection_info);
        bool processTxControlCmd(const ActionMessage& cmd,
                                 std::map<route_id, std::string>& routes,
                                 std::map<std::string, std::string>& connection_info);
 
        int processRxMessage(zmq::socket_t& socket,
                             std::map<std::string, std::string>& connection_info);
        int replyToIncomingMessage(zmq::message_t& msg, zmq::socket_t& sock);
 
        int initializeConnectionToBroker(zmq::socket_t& brokerConnection);
 
        int initializeBrokerConnections(zmq::socket_t& brokerSocket,
                                        zmq::socket_t& brokerConnection);
    };
 
}  // namespace zeromq
}  // namespace helics