[FOLD]

Author: vinniefalco

include/boost/ws_io/client.hpp

@@ -20,6 +20,13 @@
 namespace boost {
 namespace ws_io {
 
+struct null_decorator
+{
+    void operator()(...) const noexcept
+    {
+    }
+};
+
 /** A websocket client session
 */
 template<
@@ -41,16 +48,19 @@ class client
         BOOST_ASIO_COMPLETION_TOKEN_FOR(void(
             ::boost::beast::error_code,
             ::boost::http_proto::response_view)) HandshakeHandler =
-            asio::default_completion_token_t<executor_type>
+            asio::default_completion_token_t<executor_type>,
+        class Decorator = null_decorator
     >
     BOOST_ASIO_INITFN_AUTO_RESULT_TYPE(HandshakeHandler, void(
         ::boost::beast::error_code,
         ::boost::http_proto::response_view))
     async_handshake(
         core::string_view host,
         core::string_view target,
+        Decorator decorator = {},
         HandshakeHandler&& handler =
-            asio::default_completion_token_t<executor_type>{});
+            asio::default_completion_token_t<executor_type>{}
+        );
 
     AsyncStream&
     next_layer() noexcept

include/boost/ws_io/impl/client.hpp

@@ -94,14 +94,16 @@ template<class AsyncStream>
 template<
     BOOST_ASIO_COMPLETION_TOKEN_FOR(void(
         ::boost::beast::error_code,
-        ::boost::http_proto::response_view)) HandshakeHandler>
+        ::boost::http_proto::response_view)) HandshakeHandler,
+    class Decorator>
 BOOST_ASIO_INITFN_AUTO_RESULT_TYPE(HandshakeHandler, void(
     ::boost::beast::error_code,
     ::boost::http_proto::response_view))
 client<AsyncStream>::
 async_handshake(
     core::string_view host,
     core::string_view target,
+    Decorator decorator,
     HandshakeHandler&& handler)
 {
     return asio::async_initiate<

test/unit/client.cpp

@@ -31,8 +31,11 @@ struct client_test
         cs.async_handshake(
             "localhost",
             "/",
+            [](http_proto::request&)
+            {
+            },
             test::success_handler());
-        srv.run();
+        //srv.run();
     }
 };