nik/chorus
1
0
Fork 0
mirror of https://github.com/mikedilger/chorus.git synced 2026-05-03 06:51:42 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

specify WebSocketConfig

Browse Source
This commit is contained in:
Mike Dilger 2024-02-17 13:31:46 +13:00
parent a5ac657a65
commit d3c16baa2a
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/main.rs
View File

@ -32,6 +32,7 @@ use std::net::SocketAddr;
use std::pin::Pin; use std::pin::Pin;
use std::task::{Context, Poll}; use std::task::{Context, Poll};
use tokio::net::{TcpListener, TcpStream}; use tokio::net::{TcpListener, TcpStream};
use tungstenite::protocol::WebSocketConfig;
use tungstenite::Message; use tungstenite::Message;
#[tokio::main] #[tokio::main]
@ -154,7 +155,14 @@ async fn handle_http_request(
mut request: Request<Body>, mut request: Request<Body>,
) -> Result<Response<Body>, Error> { ) -> Result<Response<Body>, Error> {
if hyper_tungstenite::is_upgrade_request(&request) { 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 { tokio::spawn(async move {
// Await the websocket upgrade process // Await the websocket upgrade process
match websocket.await { match websocket.await {