fix PR

Signed-off-by: adi_holden <[email protected]>

Author: adiholden

src/server/io_utils.cc

@@ -12,6 +12,18 @@ using namespace std;
 namespace dfly {
 
 io::Result<size_t> BufferedStreamerBase::WriteSome(const iovec* vec, uint32_t len) {
+  // Shrink producer_buf_ only if it is empty, its capacity reached 10 times more than max buffer
+  // memory and the write len is less than max buffer memory.
+  if (producer_buf_.InputLen() == 0 && producer_buf_.Capacity() > max_buffered_mem_ * 10) {
+    uint32_t write_len = 0;
+    for (uint32_t i = 0; i < len; ++i) {
+      write_len += vec->iov_len;
+    }
+    if (write_len < max_buffered_mem_) {
+      consumer_buf_ = base::IoBuf{max_buffered_mem_};
+    }
+  }
+
   return io::BufSink{&producer_buf_}.WriteSome(vec, len);
 }
 
@@ -58,9 +70,6 @@ error_code BufferedStreamerBase::ConsumeIntoSink(io::Sink* dest) {
     }
 
     consumer_buf_.Clear();
-    if (consumer_buf_.Capacity() > max_buffered_mem_ * 10) {
-      consumer_buf_ = base::IoBuf{};
-    }
   }
   return std::error_code{};
 }

src/server/io_utils.h

@@ -20,7 +20,7 @@ class BufferedStreamerBase : public io::Sink {
  protected:
   // Initialize with global cancellation and optional stall conditions.
   BufferedStreamerBase(const Cancellation* cll, unsigned max_buffered_cnt = 5,
-                       unsigned max_buffered_mem = 512)
+                       unsigned max_buffered_mem = 8192)
       : cll_{cll}, max_buffered_cnt_{max_buffered_cnt}, max_buffered_mem_{max_buffered_mem} {
   }