From 2c9f3be19551a075654830441d196a20c7e5a3fd Mon Sep 17 00:00:00 2001
From: Evgeniy Ivanov <eivanov89@ydb.tech>
Date: Thu, 21 Nov 2024 09:12:24 +0100
Subject: [PATCH] Use tail sends when replying from QP (#11813)

---
 ydb/core/kqp/proxy_service/kqp_proxy_service.cpp | 2 +-
 ydb/core/kqp/session_actor/kqp_session_actor.cpp | 2 +-
 ydb/core/kqp/session_actor/kqp_worker_actor.cpp  | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/ydb/core/kqp/proxy_service/kqp_proxy_service.cpp b/ydb/core/kqp/proxy_service/kqp_proxy_service.cpp
index c46e8bed14f7..340cdb4e2c37 100644
--- a/ydb/core/kqp/proxy_service/kqp_proxy_service.cpp
+++ b/ydb/core/kqp/proxy_service/kqp_proxy_service.cpp
@@ -919,7 +919,7 @@ class TKqpProxyService : public TActorBootstrapped<TKqpProxyService> {
             LocalSessions->StartIdleCheck(info, GetSessionIdleDuration());
         }
 
-        Send(proxyRequest->Sender, ev->Release().Release(), 0, proxyRequest->SenderCookie);
+        Send<ESendingType::Tail>(proxyRequest->Sender, ev->Release().Release(), 0, proxyRequest->SenderCookie);
 
         if (info && proxyRequest->EventType == TKqpEvents::EvQueryRequest) {
             LocalSessions->DetachQueryText(info);
diff --git a/ydb/core/kqp/session_actor/kqp_session_actor.cpp b/ydb/core/kqp/session_actor/kqp_session_actor.cpp
index 718e9ffbbb05..71600ee6db41 100644
--- a/ydb/core/kqp/session_actor/kqp_session_actor.cpp
+++ b/ydb/core/kqp/session_actor/kqp_session_actor.cpp
@@ -2025,7 +2025,7 @@ class TKqpSessionActor : public TActorBootstrapped<TKqpSessionActor> {
             TlsActivationContext->AsActorContext()
         );
 
-        Send(QueryState->Sender, QueryResponse.release(), 0, QueryState->ProxyRequestId);
+        Send<ESendingType::Tail>(QueryState->Sender, QueryResponse.release(), 0, QueryState->ProxyRequestId);
         LOG_D("Sent query response back to proxy, proxyRequestId: " << QueryState->ProxyRequestId
             << ", proxyId: " << QueryState->Sender.ToString());
 
diff --git a/ydb/core/kqp/session_actor/kqp_worker_actor.cpp b/ydb/core/kqp/session_actor/kqp_worker_actor.cpp
index a3d8face05c8..8f90454c5ff7 100644
--- a/ydb/core/kqp/session_actor/kqp_worker_actor.cpp
+++ b/ydb/core/kqp/session_actor/kqp_worker_actor.cpp
@@ -753,7 +753,7 @@ class TKqpWorkerActor : public TActorBootstrapped<TKqpWorkerActor> {
             response.SetSessionId(SessionId);
         }
 
-        ctx.Send(QueryState->Sender, responseEv.Release(), 0, QueryState->ProxyRequestId);
+        ctx.Send<ESendingType::Tail>(QueryState->Sender, responseEv.Release(), 0, QueryState->ProxyRequestId);
         LOG_D("Sent query response back to proxy, proxyRequestId: " << QueryState->ProxyRequestId
             << ", proxyId: " << QueryState->Sender.ToString());