From 0e6e96fb8ce59bcffa577cf99190301d0c4324c3 Mon Sep 17 00:00:00 2001
From: ohaddahan <ohaddahan@gmail.com>
Date: Wed, 29 Jan 2025 13:21:56 +0200
Subject: [PATCH] ws: add `read_buffer_size(usize)` to `WebSocketUpgrade`
 (#3178)

---
 axum/src/extract/ws.rs | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/axum/src/extract/ws.rs b/axum/src/extract/ws.rs
index 96bea471..70b132cf 100644
--- a/axum/src/extract/ws.rs
+++ b/axum/src/extract/ws.rs
@@ -152,6 +152,12 @@ impl<F> std::fmt::Debug for WebSocketUpgrade<F> {
 }
 
 impl<F> WebSocketUpgrade<F> {
+    /// Read buffer capacity. The default value is 128KiB
+    pub fn read_buffer_size(mut self, size: usize) -> Self {
+        self.config.read_buffer_size = size;
+        self
+    }
+
     /// The target minimum size of the write buffer to reach before writing the data
     /// to the underlying stream.
     ///