TcpComms.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 "gmlc/containers/BlockingQueue.hpp"
 
#include <atomic>
#include <memory>
#include <set>
#include <string>
 
namespace gmlc::networking {
class AsioContextManager;
class TcpConnection;
}  // namespace gmlc::networking
 
namespace helics::tcp {
 
class TcpComms final: public NetworkCommsInterface {
  public:
    TcpComms() noexcept;
    ~TcpComms();
    virtual void loadNetworkInfo(const NetworkBrokerData& netInfo) override;
 
    virtual void setFlag(std::string_view flag, bool val) override;
 
  private:
    bool reuse_address{false};
    std::string encryption_config;
    virtual int getDefaultBrokerPort() const override;
    virtual void queue_rx_function() override;  
    virtual void queue_tx_function() override;  
 
    virtual void closeReceiver() override;  
 
    bool establishBrokerConnection(
        std::shared_ptr<gmlc::networking::AsioContextManager>& ioctx,
        std::shared_ptr<gmlc::networking::TcpConnection>& brokerConnection);
    int processIncomingMessage(ActionMessage&& cmd);
    // promise and future for communicating port number from tx_thread to rx_thread
    gmlc::containers::BlockingQueue<ActionMessage> rxMessageQueue;
 
    void txReceive(const char* data, size_t bytes_received, const std::string& errorMessage);
 
    size_t dataReceive(gmlc::networking::TcpConnection* connection,
                       const char* data,
                       size_t bytes_received);
 
    //  bool errorHandle()
};
 
}  // namespace helics::tcp