From 1c318f6e3c953469fc30fee22d93679c5fc12e24 Mon Sep 17 00:00:00 2001
From: Tian jianyong <11429339@qq.com>
Date: Fri, 20 Dec 2024 16:46:19 +0800
Subject: [PATCH] =?UTF-8?q?=E8=A7=A3=E5=86=B3=E7=BC=96=E8=AF=91=E9=94=99?=
 =?UTF-8?q?=E8=AF=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/network/WebSocketServer.cpp | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/network/WebSocketServer.cpp b/src/network/WebSocketServer.cpp
index aafe70e..dd18d23 100644
--- a/src/network/WebSocketServer.cpp
+++ b/src/network/WebSocketServer.cpp
@@ -1,3 +1,6 @@
+#include <boost/beast/core.hpp>
+#include <boost/beast/websocket.hpp>
+#include <boost/asio/ip/tcp.hpp>
 #include "network/WebSocketServer.h"
 #include "utils/Logger.h"
 #include <nlohmann/json.hpp>
@@ -91,12 +94,12 @@ void WebSocketServer::handleAccept() {
 
 void WebSocketServer::doRead(std::shared_ptr<boost::beast::websocket::stream<boost::asio::ip::tcp::socket>> ws) {
     // 为每个会话创建一个专用的缓冲区
-    auto buffer = std::make_shared<boost::beast::flat_buffer>();
+    auto buffer = std::make_shared<boost::beast::multi_buffer>();
     
     // 异步读取消息
     ws->async_read(
         *buffer,
-        [this, ws, buffer](boost::system::error_code ec, std::size_t bytes_transferred) {
+        [this, ws, buffer = buffer](boost::system::error_code ec, std::size_t bytes_transferred) {
             if (!ec) {
                 // 成功读取消息，继续读取下一条
                 buffer->consume(buffer->size());  // 清空缓冲区