Use CANCELED status in case of compilation timeout if cancel after has been set. KIKIMR-18698

1 files changed, 4 insertions, 0 deletions

diff --git a/ydb/core/kqp/session_actor/kqp_session_actor.cpp b/ydb/core/kqp/session_actor/kqp_session_actor.cpp
index 7e5fedd18c..d6b74dcede 100644
--- a/ydb/core/kqp/session_actor/kqp_session_actor.cpp
+++ b/ydb/core/kqp/session_actor/kqp_session_actor.cpp
@@ -1507,6 +1507,10 @@ public:
 
             const auto& phyQuery = QueryState->PreparedQuery->GetPhysicalQuery();
             FillColumnsMeta(phyQuery, response);
+        } else if (compileResult->Status == Ydb::StatusIds::TIMEOUT && QueryState->QueryDeadlines.CancelAt) {
+            // The compile timeout cause cancelation execution of request.
+            // So in case of cancel after we can reply with canceled status
+            ev.SetYdbStatus(Ydb::StatusIds::CANCELLED);
         }
     }