使用Boost.asio与Boost.beast基于协程连接ws
目录[*]目录
[*]前言
[*]准备工作
[*]实现
[*]初始化io_context并监听信号
[*]启动连接ws的线程并启动io_context
[*]建立tcp链接（以下步骤皆位于ws函数中）
[*]ws握手
[*]传输数据
[*]效果
[*]总结
前言
本文主要介绍一个使用Boost.asio和Boost.beast基于协程连接Websocket(ws)的方法。其中C++版本为20，Boost版本为1.82。
准备工作
首先需要构造一个最基本的ws服务器用于测试。
本文使用nodejs构造了一个简单的ws服务器，基于ws库。
const WebSocket = require('ws');const wss = new WebSocket.Server({ port: 8080 });wss.on('connection', function connection(ws) {console.log('New client connected')ws.on('message', function incoming(message) { console.log('received: %s', message); ws.send(message);});});console.log('WebSocket server is running on port 8080');实现
初始化io_context并监听信号
boost::asio::io_context io_context;boost::asio::signal_set signals(io_context, SIGINT, SIGTERM);signals.async_wait([&](auto, auto){ io_context.stop(); });启动连接ws的线程并启动io_context
boost::asio::co_spawn(io_context, ws, boost::asio::detached);io_context.run();其中ws的签名为boost::asio::awaitable<void> ws()
建立tcp链接（以下步骤皆位于ws函数中）
这一步可以分为两个步骤，解析dns以及建立tcp链接。
auto executor = co_await boost::asio::this_coro::executor;boost::asio::ip::tcp::socket socket(executor);boost::asio::ip::tcp::resolver resolver(executor);// 如果不使用dns解析，也可以直接使用以下直接代替// boost::asio::ip::tcp::endpoint(boost::asio::ip::address::from_string("127.0.0.1"), 8080)auto point = co_await resolver.async_resolve("localhost", "8080", boost::asio::use_awaitable);co_await socket.async_connect(point->endpoint(), boost::asio::use_awaitable);ws握手
先使用boost::beast::websocket::stream<boost::asio::ip::tcp::socket&>包装，然后进行握手。
boost::beast::websocket::stream<boost::asio::ip::tcp::socket&> ws(socket);co_await ws.async_handshake("127.0.0.1", "/", boost::asio::use_awaitable);握手过程中发送的信息类似于
GET / HTTP/1.1Host: www.example.comUpgrade: websocketConnection: upgradeSec-WebSocket-Key: 2pGeTR0DsE4dfZs2pH+8MA==Sec-WebSocket-Version: 13User-Agent: Boost.Beast/216传输数据
boost::asio::steady_timer timer(executor);for (;;) {co_await ws.async_write(boost::asio::buffer("hello"), boost::asio::use_awaitable);std::cout << "send: hello" << std::endl; boost::beast::flat_buffer buffer;co_await ws.async_read(buffer, boost::asio::use_awaitable);std::cout << boost::format("recv: %s") % std::string((char *)buffer.data().data(), buffer.data().size()) << std::endl; timer.expires_after(std::chrono::seconds(1));co_await timer.async_wait(boost::asio::use_awaitable);}效果
总结
有了协程之后，boost感觉好用多了
页:
[1]