Added removing session from the pool, if client cancelled the stream

Author: alex268

query/src/main/java/tech/ydb/query/impl/SessionImpl.java

@@ -58,6 +58,11 @@ abstract class SessionImpl implements QuerySession {
             YdbQuery.DeleteSessionResponse::getStatus, YdbQuery.DeleteSessionResponse::getIssuesList
     );
 
+    private static final Status CANCELLED = Status.of(
+            StatusCode.CLIENT_CANCELLED,
+            Issue.of("Stream was cancelled by client, session will be removed", Issue.Severity.WARNING)
+    );
+
     private final QueryServiceRpc rpc;
     private final String sessionId;
     private final long nodeID;
@@ -313,6 +318,7 @@ public CompletableFuture<Result<QueryInfo>> execute(PartsHandler handler) {
 
         @Override
         public void cancel() {
+            updateSessionState(CANCELLED);
             grpcStream.cancel();
         }
     }

query/src/main/java/tech/ydb/query/impl/SessionPool.java

@@ -175,6 +175,7 @@ public void updateSessionState(Status status) {
                     status.getCode() == StatusCode.INTERNAL_ERROR ||
                     status.getCode() == StatusCode.CLIENT_DEADLINE_EXCEEDED ||
                     status.getCode() == StatusCode.CLIENT_DEADLINE_EXPIRED ||
+                    status.getCode() == StatusCode.CLIENT_CANCELLED ||
                     status.getCode() == StatusCode.TRANSPORT_UNAVAILABLE;
             if (isStatusBroken) {
                 logger.warn("QuerySession[{}] broken with status {}", getId(), status);