fix

Author: KirillKurdyukov

src/Ydb.Sdk/src/Ado/Session/PoolingSession.cs

@@ -99,7 +99,9 @@ StatusCode.BadSession or
             StatusCode.SessionBusy or
             StatusCode.SessionExpired or
             StatusCode.ClientTransportTimeout or
-            StatusCode.ClientTransportUnavailable)
+            StatusCode.ClientTransportUnavailable or
+            StatusCode.ClientTransportResourceExhausted or
+            StatusCode.ClientTransportUnknown)
         {
             _logger.LogWarning("Session[{SessionId}] is deactivated. Reason Status: {Status}", SessionId, statusCode);
 

src/Ydb.Sdk/src/Ado/Session/RetryableSession.cs

@@ -87,22 +87,30 @@ public async Task<bool> MoveNextAsync(CancellationToken cancellationToken = defa
         {
             using var session = await _sessionSource.OpenSession(ct);
 
-            var responses = new List<ExecuteQueryResponsePart>();
-            var serverStream = await session.ExecuteQuery(_query, _parameters, _settings, null);
-
-            while (await serverStream.MoveNextAsync(ct))
+            try
             {
-                var current = serverStream.Current;
+                var responses = new List<ExecuteQueryResponsePart>();
+                var serverStream = await session.ExecuteQuery(_query, _parameters, _settings, null);
 
-                if (current.Status.IsNotSuccess())
+                while (await serverStream.MoveNextAsync(ct))
                 {
-                    throw YdbException.FromServer(current.Status, current.Issues);
+                    var current = serverStream.Current;
+
+                    if (current.Status.IsNotSuccess())
+                    {
+                        throw YdbException.FromServer(current.Status, current.Issues);
+                    }
+
+                    responses.Add(serverStream.Current);
                 }
 
-                responses.Add(serverStream.Current);
+                return responses;
+            }
+            catch (YdbException e)
+            {
+                session.OnNotSuccessStatusCode(e.Code);
+                throw;
             }
-
-            return responses;
         }, cancellationToken);
 
         return _responses.Count > 0;