forked from boostorg/beast
f22895224b
fix #374, fix #446 * auto_fragment option struct is removed Actions Required: * Change call sites which use auto_fragment with set_option to call stream::auto_fragment instead.
76 lines
1.9 KiB
C++
76 lines
1.9 KiB
C++
//
|
|
// Copyright (c) 2013-2017 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)
|
|
//
|
|
|
|
#include "websocket_async_echo_server.hpp"
|
|
#include "websocket_sync_echo_server.hpp"
|
|
#include <boost/asio/io_service.hpp>
|
|
#include <boost/asio/signal_set.hpp>
|
|
#include <iostream>
|
|
|
|
/// Block until SIGINT or SIGTERM is received.
|
|
void
|
|
sig_wait()
|
|
{
|
|
boost::asio::io_service ios;
|
|
boost::asio::signal_set signals(
|
|
ios, SIGINT, SIGTERM);
|
|
signals.async_wait(
|
|
[&](boost::system::error_code const&, int)
|
|
{
|
|
});
|
|
ios.run();
|
|
}
|
|
|
|
class set_stream_options
|
|
{
|
|
beast::websocket::permessage_deflate pmd_;
|
|
|
|
public:
|
|
set_stream_options(set_stream_options const&) = default;
|
|
|
|
set_stream_options(
|
|
beast::websocket::permessage_deflate const& pmd)
|
|
: pmd_(pmd)
|
|
{
|
|
}
|
|
|
|
template<class NextLayer>
|
|
void
|
|
operator()(beast::websocket::stream<NextLayer>& ws) const
|
|
{
|
|
ws.auto_fragment(false);
|
|
ws.set_option(pmd_);
|
|
ws.set_option(beast::websocket::read_message_max{64 * 1024 * 1024});
|
|
}
|
|
};
|
|
|
|
int main()
|
|
{
|
|
using namespace beast::websocket;
|
|
using endpoint_type = boost::asio::ip::tcp::endpoint;
|
|
using address_type = boost::asio::ip::address;
|
|
|
|
beast::error_code ec;
|
|
|
|
permessage_deflate pmd;
|
|
pmd.client_enable = true;
|
|
pmd.server_enable = true;
|
|
pmd.compLevel = 3;
|
|
|
|
websocket::async_echo_server s1{&std::cout, 1};
|
|
s1.on_new_stream(set_stream_options{pmd});
|
|
s1.open(endpoint_type{
|
|
address_type::from_string("127.0.0.1"), 6000 }, ec);
|
|
|
|
websocket::sync_echo_server s2{&std::cout};
|
|
s2.on_new_stream(set_stream_options{pmd});
|
|
s2.open(endpoint_type{
|
|
address_type::from_string("127.0.0.1"), 6001 }, ec);
|
|
|
|
sig_wait();
|
|
}
|