From 17218dc6c88848938cebc560deafbf1c2184fa56 Mon Sep 17 00:00:00 2001
From: moh-eulith <101080211+moh-eulith@users.noreply.github.com>
Date: Sun, 7 May 2023 11:13:10 -0400
Subject: [PATCH] minor optimization: reserve buffer once length is known (ws)
 (#2950)

---
 actix-http/src/ws/frame.rs | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/actix-http/src/ws/frame.rs b/actix-http/src/ws/frame.rs
index c7e0427ea..dddb03d18 100644
--- a/actix-http/src/ws/frame.rs
+++ b/actix-http/src/ws/frame.rs
@@ -1,3 +1,4 @@
+use std::cmp::min;
 use std::convert::TryFrom;
 
 use bytes::{Buf, BufMut, BytesMut};
@@ -96,6 +97,10 @@ impl Parser {
 
         // not enough data
         if src.len() < idx + length {
+            let min_length = min(length, max_size);
+            if src.capacity() < idx + min_length {
+                src.reserve(idx + min_length - src.capacity());
+            }
             return Ok(None);
         }