init commit

Author: KirillKurdyukov

src/Ydb.Sdk/src/Ado/YdbConnection.cs

@@ -102,30 +102,37 @@ public override async Task OpenAsync(CancellationToken cancellationToken)
 
     public override async Task CloseAsync()
     {
-        if (State == ConnectionState.Closed)
+        // ReSharper disable once SwitchStatementHandlesSomeKnownEnumValuesWithDefault
+        switch (State)
         {
-            return;
-        }
-
-        try
-        {
-            if (LastReader is { IsClosed: false })
-            {
-                await LastReader.CloseAsync();
-            }
-
-            if (CurrentTransaction is { Completed: false })
-            {
-                await CurrentTransaction.RollbackAsync();
-            }
-
-            OnStateChange(OpenToClosedEventArgs);
-
-            ConnectionState = ConnectionState.Closed;
-        }
-        finally
-        {
-            _session.Close();
+            case ConnectionState.Closed:
+                return;
+            case ConnectionState.Broken:
+                ConnectionState = ConnectionState.Closed;
+                return;
+            default:
+                try
+                {
+                    if (LastReader is { IsClosed: false })
+                    {
+                        await LastReader.CloseAsync();
+                    }
+
+                    if (CurrentTransaction is { Completed: false })
+                    {
+                        await CurrentTransaction.RollbackAsync();
+                    }
+
+                    OnStateChange(OpenToClosedEventArgs);
+
+                    ConnectionState = ConnectionState.Closed;
+                }
+                finally
+                {
+                    _session.Close();
+                }
+
+                break;
         }
     }
 
@@ -145,19 +152,14 @@ public override string ConnectionString
 
     public override string Database => _connectionStringBuilder?.Database ?? string.Empty;
 
-    public override ConnectionState State => ConnectionState;
+    public override ConnectionState State =>
+        ConnectionState != ConnectionState.Closed && _session.IsBroken // maybe is updated asynchronously
+            ? ConnectionState.Broken
+            : ConnectionState;
 
     private ConnectionState ConnectionState { get; set; } = ConnectionState.Closed; // Invoke AsyncOpen()
 
-    internal void OnNotSuccessStatusCode(StatusCode code)
-    {
-        _session.OnNotSuccessStatusCode(code);
-
-        if (_session.IsBroken)
-        {
-            ConnectionState = ConnectionState.Broken;
-        }
-    }
+    internal void OnNotSuccessStatusCode(StatusCode code) => _session.OnNotSuccessStatusCode(code);
 
     internal YdbDataReader? LastReader { get; set; }
     internal string LastCommand { get; set; } = string.Empty;
@@ -203,15 +205,15 @@ public override Task<DataTable> GetSchemaAsync(
 
     internal void ThrowIfConnectionClosed()
     {
-        if (ConnectionState is ConnectionState.Closed or ConnectionState.Broken)
+        if (State is ConnectionState.Closed or ConnectionState.Broken)
         {
             throw new InvalidOperationException("Connection is closed");
         }
     }
 
     private void ThrowIfConnectionOpen()
     {
-        if (ConnectionState == ConnectionState.Open)
+        if (State == ConnectionState.Open)
         {
             throw new InvalidOperationException("Connection already open");
         }

src/Ydb.Sdk/src/IDriver.cs

@@ -210,7 +210,7 @@ protected async ValueTask<CallOptions> GetCallOptions(GrpcRequestSettings settin
     }
 
     public ILoggerFactory LoggerFactory { get; }
-    public void RegisterOwner() => _ownerCount++;
+    public void RegisterOwner() => Interlocked.Increment(ref _ownerCount);
     public bool IsDisposed => Disposed == 1;
 
     public void Dispose() => DisposeAsync().AsTask().GetAwaiter().GetResult();