From d3c16baa2a1abf80a384f215dee5be1849d65306 Mon Sep 17 00:00:00 2001 From: Mike Dilger Date: Sat, 17 Feb 2024 13:31:46 +1300 Subject: [PATCH] specify WebSocketConfig --- src/main.rs | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/src/main.rs b/src/main.rs index aa147fe..9f005a5 100644 --- a/src/main.rs +++ b/src/main.rs @@ -32,6 +32,7 @@ use std::net::SocketAddr; use std::pin::Pin; use std::task::{Context, Poll}; use tokio::net::{TcpListener, TcpStream}; +use tungstenite::protocol::WebSocketConfig; use tungstenite::Message; #[tokio::main] @@ -154,7 +155,14 @@ async fn handle_http_request( mut request: Request, ) -> Result, Error> { if hyper_tungstenite::is_upgrade_request(&request) { - let (response, websocket) = hyper_tungstenite::upgrade(&mut request, None)?; + let web_socket_config = WebSocketConfig { + max_write_buffer_size: 1024 * 1024, // 1 MB + max_message_size: Some(1024 * 1024), // 1 MB + max_frame_size: Some(1024 * 1024), // 1 MB + ..Default::default() + }; + let (response, websocket) = + hyper_tungstenite::upgrade(&mut request, Some(web_socket_config))?; tokio::spawn(async move { // Await the websocket upgrade process match websocket.await {