WsClient.hpp

Go to the documentation of this file.

// @formatter:off
//
// Balau core C++ library
//
// Copyright (C) 2017 Bora Software (contact@borasoftware.com)
//
// Licensed under the Boost Software License - Version 1.0 - August 17th, 2003.
// See the LICENSE file for the full license text.
//


#ifndef COM_BORA_SOFTWARE__BALAU_NETWORK_HTTP_CLIENT__WS_CLIENT
#define COM_BORA_SOFTWARE__BALAU_NETWORK_HTTP_CLIENT__WS_CLIENT

#include <Balau/Type/ToString.hpp>

#include <Balau/Exception/NetworkExceptions.hpp>
#include <Balau/Network/Http/Server/NetworkTypes.hpp>
#include <Balau/Resource/Url.hpp>
#include <Balau/Type/StdTypes.hpp>

#include <boost/beast/core.hpp>
#include <boost/beast/websocket.hpp>
#include <boost/asio/connect.hpp>
#include <boost/asio/ip/tcp.hpp>
#include <cstdlib>

namespace Balau::Network::Http {

class WsClient {
    public: explicit WsClient(std::string host_, unsigned short port_ = 80)
        : host(std::move(host_))
        , port(port_) {}

    public: WsClient(const WsClient &) = default;

    public: WsClient(WsClient &&) = default;

    public: WsClient & operator = (const WsClient &) = default;

    public: WsClient & operator = (WsClient &&) = default;

    public: template <typename ResponseBufferT, typename RequestBufferT>
    ResponseBufferT send(const std::string_view & path, const RequestBufferT & requestData) {

        boost::asio::io_context ioc;
        TCP::resolver resolver { ioc };
        TCP::socket socket { ioc };

        auto resolverResults = resolver.resolve(host.c_str(), ::toString(port).c_str());
        boost::asio::connect(socket, resolverResults.begin(), resolverResults.end());

        WS::stream<TCP::socket> ws { ioc };

        ws.handshake(host, path.empty() ? "/" : std::string(path));
        ws.write(requestData);

        ResponseBufferT buffer {};

        ws.read(buffer);
        ws.close(WS::close_code::normal);


        boost::system::error_code errorCode {};
        socket.shutdown(TCP::socket::shutdown_both, errorCode);

        // TODO clarify the following.
        if (errorCode && errorCode != boost::system::errc::not_connected) {
            throw boost::system::system_error { errorCode }; // TODO
        }


        return buffer;
    }


    private: std::string host;
    private: unsigned short port;
};

} // namespace Balau::Network::Http

#endif // COM_BORA_SOFTWARE__BALAU_NETWORK_HTTP_CLIENT__HTTPS_CLIENT