From a9c47c2bf4a2b4ed7f98627912447219d2034795 Mon Sep 17 00:00:00 2001
From: Anton Ivashkin <ianton@live.com>
Date: Wed, 17 Sep 2025 17:09:41 +0200
Subject: [PATCH] Fix for restart on connection lost

---
 src/QueryPipeline/RemoteQueryExecutorReadContext.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/QueryPipeline/RemoteQueryExecutorReadContext.cpp b/src/QueryPipeline/RemoteQueryExecutorReadContext.cpp
index 0e3fb4952eb4..53d292cdef06 100644
--- a/src/QueryPipeline/RemoteQueryExecutorReadContext.cpp
+++ b/src/QueryPipeline/RemoteQueryExecutorReadContext.cpp
@@ -81,6 +81,7 @@ void RemoteQueryExecutorReadContext::Task::run(AsyncCallback async_callback, Sus
             throw;
         read_context.packet.type = Protocol::Server::ConnectionLost;
         read_context.packet.exception = std::make_unique<Exception>(getCurrentExceptionMessageAndPattern(true), getCurrentExceptionCode());
+        read_context.has_read_packet_part = PacketPart::Body;
         suspend_callback();
     }
 }