ZmqRequestSets.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 "../../core/ActionMessage.hpp"
#include "ZmqContextManager.h"
#include "cppzmq/zmq.hpp"
 
#include <deque>
#include <map>
#include <memory>
#include <optional>
#include <string>
#include <utility>
#include <vector>
 
namespace helics {
namespace zeromq {
    class WaitingResponse {
      public:
        int route = 0;  
        std::uint16_t loops = 0;  
        bool waiting = false;  
        ActionMessage txmsg;  
    };
 
    class ZmqRequestSets {
      public:
        ZmqRequestSets();
        void addRoutes(int routeNumber, const std::string& routeInfo);
        bool transmit(int routeNumber, const ActionMessage& command);
        bool waiting() const;
        int checkForMessages();
        int checkForMessages(std::chrono::milliseconds timeout);
        bool hasMessages() const;
        std::optional<ActionMessage> getMessage();
        void close();
 
      private:
        void SendDelayedMessages();
 
      private:
        std::map<int, std::unique_ptr<zmq::socket_t>> routes;  
        std::map<int, bool> routes_waiting;  
        std::vector<zmq::pollitem_t> active_routes;  
        std::vector<WaitingResponse> active_messages;  
        std::vector<std::pair<int, ActionMessage>>
            waiting_messages;  
        std::deque<ActionMessage> Responses;  
        std::shared_ptr<ZmqContextManager> ctx;  
    };
}  // namespace zeromq
}  // namespace helics