1 files changed, 153 insertions, 0 deletions
diff --git a/src/boost/libs/beast/example/websocket/client/coro/websocket_client_coro.cpp b/src/boost/libs/beast/example/websocket/client/coro/websocket_client_coro.cpp
new file mode 100644
index 000000000..7a4b0616d
--- /dev/null
+++ b/src/boost/libs/beast/example/websocket/client/coro/websocket_client_coro.cpp
@@ -0,0 +1,153 @@
+//
+// Copyright (c) 2016-2019 Vinnie Falco (vinnie dot falco at gmail dot com)
+//
+// Distributed under the Boost Software License, Version 1.0. (See accompanying
+// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
+//
+// Official repository: https://github.com/boostorg/beast
+//
+
+//------------------------------------------------------------------------------
+//
+// Example: WebSocket client, coroutine
+//
+//------------------------------------------------------------------------------
+
+#include <boost/beast/core.hpp>
+#include <boost/beast/websocket.hpp>
+#include <boost/asio/spawn.hpp>
+#include <cstdlib>
+#include <functional>
+#include <iostream>
+#include <string>
+
+namespace beast = boost::beast;         // from <boost/beast.hpp>
+namespace http = beast::http;           // from <boost/beast/http.hpp>
+namespace websocket = beast::websocket; // from <boost/beast/websocket.hpp>
+namespace net = boost::asio;            // from <boost/asio.hpp>
+using tcp = boost::asio::ip::tcp;       // from <boost/asio/ip/tcp.hpp>
+
+//------------------------------------------------------------------------------
+
+// Report a failure
+void
+fail(beast::error_code ec, char const* what)
+{
+    std::cerr << what << ": " << ec.message() << "\n";
+}
+
+// Sends a WebSocket message and prints the response
+void
+do_session(
+    std::string host,
+    std::string const& port,
+    std::string const& text,
+    net::io_context& ioc,
+    net::yield_context yield)
+{
+    beast::error_code ec;
+
+    // These objects perform our I/O
+    tcp::resolver resolver(ioc);
+    websocket::stream<beast::tcp_stream> ws(ioc);
+
+    // Look up the domain name
+    auto const results = resolver.async_resolve(host, port, yield[ec]);
+    if(ec)
+        return fail(ec, "resolve");
+
+    // Set a timeout on the operation
+    beast::get_lowest_layer(ws).expires_after(std::chrono::seconds(30));
+
+    // Make the connection on the IP address we get from a lookup
+    auto ep = beast::get_lowest_layer(ws).async_connect(results, yield[ec]);
+    if(ec)
+        return fail(ec, "connect");
+
+    // Update the host_ string. This will provide the value of the
+    // Host HTTP header during the WebSocket handshake.
+    // See https://tools.ietf.org/html/rfc7230#section-5.4
+    host += ':' + std::to_string(ep.port());
+
+    // Turn off the timeout on the tcp_stream, because
+    // the websocket stream has its own timeout system.
+    beast::get_lowest_layer(ws).expires_never();
+
+    // Set suggested timeout settings for the websocket
+    ws.set_option(
+        websocket::stream_base::timeout::suggested(
+            beast::role_type::client));
+
+    // Set a decorator to change the User-Agent of the handshake
+    ws.set_option(websocket::stream_base::decorator(
+        [](websocket::request_type& req)
+        {
+            req.set(http::field::user_agent,
+                std::string(BOOST_BEAST_VERSION_STRING) +
+                    " websocket-client-coro");
+        }));
+
+    // Perform the websocket handshake
+    ws.async_handshake(host, "/", yield[ec]);
+    if(ec)
+        return fail(ec, "handshake");
+
+    // Send the message
+    ws.async_write(net::buffer(std::string(text)), yield[ec]);
+    if(ec)
+        return fail(ec, "write");
+
+    // This buffer will hold the incoming message
+    beast::flat_buffer buffer;
+
+    // Read a message into our buffer
+    ws.async_read(buffer, yield[ec]);
+    if(ec)
+        return fail(ec, "read");
+
+    // Close the WebSocket connection
+    ws.async_close(websocket::close_code::normal, yield[ec]);
+    if(ec)
+        return fail(ec, "close");
+
+    // If we get here then the connection is closed gracefully
+
+    // The make_printable() function helps print a ConstBufferSequence
+    std::cout << beast::make_printable(buffer.data()) << std::endl;
+}
+
+//------------------------------------------------------------------------------
+
+int main(int argc, char** argv)
+{
+    // Check command line arguments.
+    if(argc != 4)
+    {
+        std::cerr <<
+            "Usage: websocket-client-coro <host> <port> <text>\n" <<
+            "Example:\n" <<
+            "    websocket-client-coro echo.websocket.org 80 \"Hello, world!\"\n";
+        return EXIT_FAILURE;
+    }
+    auto const host = argv[1];
+    auto const port = argv[2];
+    auto const text = argv[3];
+
+    // The io_context is required for all I/O
+    net::io_context ioc;
+
+    // Launch the asynchronous operation
+    boost::asio::spawn(ioc, std::bind(
+        &do_session,
+        std::string(host),
+        std::string(port),
+        std::string(text),
+        std::ref(ioc),
+        std::placeholders::_1));
+
+    // Run the I/O service. The call will return when
+    // the socket is closed.
+    ioc.run();
+
+    return EXIT_SUCCESS;
+}