SocketCommunication.hpp

Go to the documentation of this file.

#pragma once
 
#include <boost/asio.hpp>
#include <map>
#include <memory>
#include <set>
#include <stddef.h>
#include <string>
#include <thread>
 
#include "com/Communication.hpp"
#include "com/SharedPointer.hpp"
#include "com/SocketSendQueue.hpp"
#include "logging/Logger.hpp"
#include "precice/impl/Types.hpp"
#include "utils/networking.hpp"
 
namespace precice::com {
class SocketCommunication : public Communication {
public:
  SocketCommunication(unsigned short portNumber       = 0,
                      bool           reuseAddress     = false,
                      std::string    networkName      = utils::networking::loopbackInterfaceName(),
                      std::string    addressDirectory = ".");
 
  explicit SocketCommunication(std::string const &addressDirectory);
 
  ~SocketCommunication() override;
 
  size_t getRemoteCommunicatorSize() override;
 
  void acceptConnection(std::string const &acceptorName,
                        std::string const &requesterName,
                        std::string const &tag,
                        int                acceptorRank,
                        int                rankOffset = 0) override;
 
  void acceptConnectionAsServer(std::string const &acceptorName,
                                std::string const &requesterName,
                                std::string const &tag,
                                int                acceptorRank,
                                int                requesterCommunicatorSize) override;
 
  void requestConnection(std::string const &acceptorName,
                         std::string const &requesterName,
                         std::string const &tag,
                         int                requesterRank,
                         int                requesterCommunicatorSize) override;
 
  void requestConnectionAsClient(std::string const   &acceptorName,
                                 std::string const   &requesterName,
                                 std::string const   &tag,
                                 std::set<int> const &acceptorRanks,
                                 int                  requesterRank) override;
 
  void closeConnection() override;
 
  void send(std::string const &itemToSend, Rank rankReceiver) override;
 
  void send(precice::span<const int> itemsToSend, Rank rankReceiver) override;
 
  PtrRequest aSend(precice::span<const int> itemsToSend, Rank rankReceiver) override;
 
  void send(precice::span<const double> itemsToSend, Rank rankReceiver) override;
 
  PtrRequest aSend(precice::span<const double> itemsToSend, Rank rankReceiver) override;
 
  void send(double itemToSend, Rank rankReceiver) override;
 
  PtrRequest aSend(const double &itemToSend, Rank rankReceiver) override;
 
  void send(int itemToSend, Rank rankReceiver) override;
 
  PtrRequest aSend(const int &itemToSend, Rank rankReceiver) override;
 
  void send(bool itemToSend, Rank rankReceiver) override;
 
  PtrRequest aSend(const bool &itemToSend, Rank rankReceiver) override;
 
  void receive(std::string &itemToReceive, Rank rankSender) override;
 
  void receive(precice::span<int> itemsToReceive, Rank rankSender) override;
 
  void receive(precice::span<double> itemsToReceive, Rank rankSender) override;
 
  PtrRequest aReceive(precice::span<double> itemsToReceive,
                      int                   rankSender) override;
 
  void receive(double &itemToReceive, Rank rankSender) override;
 
  PtrRequest aReceive(double &itemToReceive, Rank rankSender) override;
 
  void receive(int &itemToReceive, Rank rankSender) override;
 
  PtrRequest aReceive(int &itemToReceive, Rank rankSender) override;
 
  void receive(bool &itemToReceive, Rank rankSender) override;
 
  PtrRequest aReceive(bool &itemToReceive, Rank rankSender) override;
 
  void prepareEstablishment(std::string const &acceptorName,
                            std::string const &requesterName) override;
 
  void cleanupEstablishment(std::string const &acceptorName,
                            std::string const &requesterName) override;
 
private:
  logging::Logger _log{"com::SocketCommunication"};
 
  unsigned short _portNumber;
 
  bool _reuseAddress;
 
  std::string _networkName;
 
  std::string _addressDirectory;
 
  using IOContext = boost::asio::io_context;
  using Socket    = boost::asio::ip::tcp::socket;
  using WorkGuard = boost::asio::executor_work_guard<IOContext::executor_type>;
 
  std::shared_ptr<IOContext> _ioContext;
  std::unique_ptr<WorkGuard> _workGuard;
  std::thread                _thread;
 
  std::map<int, std::shared_ptr<Socket>> _sockets;
 
  SocketSendQueue _queue;
 
  bool isClient();
  bool isServer();
 
  std::string getIpAddress();
};
class SocketCommunication : public Communication {…};
} // namespace precice::com