From 147ad42df09f374df971d6bed36ccf67c97d87a9 Mon Sep 17 00:00:00 2001
From: Yakov Olkhovskiy <yakov@clickhouse.com>
Date: Fri, 24 May 2024 03:36:29 +0000
Subject: [PATCH] fix notchunked mode in ReadBufferFromPocoSocketChunked

---
 src/IO/ReadBufferFromPocoSocketChunked.cpp | 7 +++++++
 src/IO/ReadBufferFromPocoSocketChunked.h   | 2 +-
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/IO/ReadBufferFromPocoSocketChunked.cpp b/src/IO/ReadBufferFromPocoSocketChunked.cpp
index 4d40d8b4f14..a67a5bb41a9 100644
--- a/src/IO/ReadBufferFromPocoSocketChunked.cpp
+++ b/src/IO/ReadBufferFromPocoSocketChunked.cpp
@@ -44,6 +44,13 @@ void ReadBufferFromPocoSocketChunked::setAsyncCallback(AsyncCallback async_callb
     buffer_socket.setAsyncCallback(async_callback_);
 }
 
+bool ReadBufferFromPocoSocketChunked::hasBufferedData() const
+{
+    if (chunked)
+        return hasPendingData() || buffer_socket.hasPendingData();
+    return hasPendingData();
+}
+
 bool ReadBufferFromPocoSocketChunked::startChunk()
 {
     if (buffer_socket.read(reinterpret_cast<char *>(&chunk_left), sizeof(chunk_left)) < sizeof(chunk_left))
diff --git a/src/IO/ReadBufferFromPocoSocketChunked.h b/src/IO/ReadBufferFromPocoSocketChunked.h
index c70363cf7d8..b0f5dd7dc5f 100644
--- a/src/IO/ReadBufferFromPocoSocketChunked.h
+++ b/src/IO/ReadBufferFromPocoSocketChunked.h
@@ -16,7 +16,7 @@ public:
     bool poll(size_t timeout_microseconds);
     void setAsyncCallback(AsyncCallback async_callback_);
 
-    bool hasBufferedData() const { return hasPendingData() || buffer_socket.hasPendingData(); }
+    bool hasBufferedData() const;
 
     Poco::Net::SocketAddress peerAddress() { return peer_address; }
     Poco::Net::SocketAddress ourAddress() { return our_address; }