ZmqComms.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 <set>
#include <string>
 
namespace zmq {
class message_t;
class socket_t;
}  // namespace zmq
namespace helics {
namespace zeromq {
    class ZmqComms final: public NetworkCommsInterface {
      public:
        ZmqComms() noexcept;
        ~ZmqComms();
        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;  
        virtual void closeReceiver() override;  
 
        int processIncomingMessage(zmq::message_t& msg);
        int replyToIncomingMessage(zmq::message_t& msg, zmq::socket_t& sock);
 
        int initializeBrokerConnections(zmq::socket_t& controlSocket);
 
      public:
        std::string getPushAddress() const;
    };
 
}  // namespace zeromq
}  // namespace helics