From 01b7b0715659d6a97bdd3ff6f78de9b25888259e Mon Sep 17 00:00:00 2001
From: Alexandr Gorshenin <alexandr268@ydb.tech>
Date: Wed, 25 Jun 2025 15:37:17 +0100
Subject: [PATCH] Fixed session status updating

---
 query/src/main/java/tech/ydb/query/impl/SessionImpl.java | 1 +
 1 file changed, 1 insertion(+)

diff --git a/query/src/main/java/tech/ydb/query/impl/SessionImpl.java b/query/src/main/java/tech/ydb/query/impl/SessionImpl.java
index b6a03388..b533b2fd 100644
--- a/query/src/main/java/tech/ydb/query/impl/SessionImpl.java
+++ b/query/src/main/java/tech/ydb/query/impl/SessionImpl.java
@@ -315,6 +315,7 @@ public CompletableFuture<Result<QueryInfo>> execute(PartsHandler handler) {
                     result.completeExceptionally(th);
                 }
                 if (status != null) {
+                    updateSessionState(status);
                     if (status.isSuccess()) {
                         result.complete(Result.success(new QueryInfo(stats.get()), status));
                     } else {