Add ability to set read_buffer_size for websockets

tokio-rs · Jan 10, 2025 · 22fa1fb · 22fa1fb
1 parent d07d129
commit 22fa1fb
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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 128 KiB.
+    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.
     ///