style

ydb-platform · Jan 15, 2025 · 2c0380e · 2c0380e
1 parent 4514e46
commit 2c0380e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/internal/query/session.go b/internal/query/session.go
@@ -181,8 +181,9 @@ func (s *Session) Query(
 
 func (s *Session) setStatusFromError(err error) {
 	switch {
-	case xerrors.IsTransportError(err) ||
-		xerrors.IsOperationError(err, Ydb.StatusIds_SESSION_BUSY, Ydb.StatusIds_BAD_SESSION):
+	case xerrors.IsTransportError(err):
+		s.SetStatus(session.StatusError)
+	case xerrors.IsOperationError(err, Ydb.StatusIds_SESSION_BUSY, Ydb.StatusIds_BAD_SESSION):
 		s.SetStatus(session.StatusError)
 	case xerrors.IsOperationError(err, Ydb.StatusIds_BAD_SESSION):
 		s.SetStatus(session.StatusClosed)