From 8720ed3826ddf7ea50dc881fe7bf324895d33e2c Mon Sep 17 00:00:00 2001
From: zainkabani <77307340+zainkabani@users.noreply.github.com>
Date: Wed, 21 Dec 2022 09:57:53 -0500
Subject: [PATCH] Buffer copy data messages (#265)

* Buffer copy data messages

* Update comment
---
 src/server.rs | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/server.rs b/src/server.rs
index 65fb8d9..05a3b77 100644
--- a/src/server.rs
+++ b/src/server.rs
@@ -510,9 +510,13 @@ impl Server {
                     break;
                 }
 
-                // CopyData: we are not buffering this one because there will be many more
-                // and we don't know how big this packet could be, best not to take a risk.
-                'd' => break,
+                // CopyData
+                'd' => {
+                    // Don't flush yet, buffer until we reach limit
+                    if self.buffer.len() >= 8196 {
+                        break;
+                    }
+                }
 
                 // CopyDone
                 // Buffer until ReadyForQuery shows up, so don't exit the loop yet.