exposing read_buffer_size

tokio-rs · Jan 16, 2025 · 2e47564 · 2e47564
1 parent e09cc59
commit 2e47564
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 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.
     ///