forked from boostorg/mqtt5
Korina Šimičević
afc270f10e
Summary: related to T15996 folder structure include/async_mqtt5 -> include/boost/mqtt5 namespace async_mqtt5 -> namespace boost::mqtt5 all tabs replaced with 4 spaces (because tabs are banned) boost-like order of includes TODO: fix all docs Reviewers: ivica Reviewed By: ivica Subscribers: iljazovic, miljen Differential Revision: https://repo.mireo.local/D33152
55 lines
1.6 KiB
C++
55 lines
1.6 KiB
C++
//
|
|
// Copyright (c) 2023-2024 Ivica Siladic, Bruno Iljazovic, Korina Simicevic
|
|
//
|
|
// Distributed under the Boost Software License, Version 1.0.
|
|
// (See accompanying file LICENSE or copy at http://www.boost.org/LICENSE_1_0.txt)
|
|
//
|
|
|
|
#ifndef BOOST_MQTT5_WEBSOCKET_HPP
|
|
#define BOOST_MQTT5_WEBSOCKET_HPP
|
|
|
|
#include <boost/mqtt5/types.hpp>
|
|
|
|
#include <boost/beast/http/field.hpp>
|
|
#include <boost/beast/websocket/rfc6455.hpp>
|
|
#include <boost/beast/websocket/stream.hpp>
|
|
|
|
namespace boost::mqtt5 {
|
|
|
|
// Trait definition for Beast
|
|
template <typename Stream>
|
|
struct ws_handshake_traits<boost::beast::websocket::stream<Stream>> {
|
|
|
|
template <typename CompletionToken>
|
|
static decltype(auto) async_handshake(
|
|
boost::beast::websocket::stream<Stream>& stream,
|
|
authority_path ap, CompletionToken&& token
|
|
) {
|
|
using namespace boost::beast;
|
|
|
|
// Set suggested timeout settings for the websocket
|
|
stream.set_option(
|
|
websocket::stream_base::timeout::suggested(role_type::client)
|
|
);
|
|
|
|
stream.binary(true);
|
|
|
|
// Set a decorator to change the User-Agent of the handshake
|
|
stream.set_option(websocket::stream_base::decorator(
|
|
[](websocket::request_type& req) {
|
|
req.set(http::field::sec_websocket_protocol, "mqtt");
|
|
req.set(http::field::user_agent, "boost.mqtt");
|
|
})
|
|
);
|
|
|
|
stream.async_handshake(
|
|
ap.host + ':' + ap.port, ap.path,
|
|
std::forward<CompletionToken>(token)
|
|
);
|
|
}
|
|
};
|
|
|
|
} // end namespace boost::mqtt5
|
|
|
|
#endif // !BOOST_MQTT5_WEBSOCKET_HPP
|