Async variant of SqlConnection opening

Author: alex-kulakov

Orm/Xtensive.Orm/Sql/SqlConnection.cs

@@ -167,8 +167,9 @@ public virtual void Open()
     {
       EnsureIsNotDisposed();
       var connectionHandlers = Extensions.Get<ConnectionHandlersExtension>();
-      if (connectionHandlers == null)
+      if (connectionHandlers == null) {
         UnderlyingConnection.Open();
+      }
       else {
         var handlers = connectionHandlers.Handlers;
         SqlHelper.NotifyConnectionOpening(handlers, UnderlyingConnection);
@@ -230,11 +231,27 @@ public virtual void OpenAndInitialize(string initializationScript)
     /// to ensure that all asynchronous operations have completed.</remarks>
     /// <param name="cancellationToken">Token to control cancellation.</param>
     /// <returns>Awaitable task.</returns>
-    public virtual Task OpenAsync(CancellationToken cancellationToken)
+    public virtual async Task OpenAsync(CancellationToken cancellationToken)
     {
       cancellationToken.ThrowIfCancellationRequested();
       EnsureIsNotDisposed();
-      return UnderlyingConnection.OpenAsync(cancellationToken);
+      var connectionHandlers = Extensions.Get<ConnectionHandlersExtension>();
+
+      if (connectionHandlers == null) {
+        await UnderlyingConnection.OpenAsync(cancellationToken).ConfigureAwait(false);
+      }
+      else {
+        var handlers = connectionHandlers.Handlers;
+        await SqlHelper.NotifyConnectionOpeningAsync(handlers, UnderlyingConnection, false, cancellationToken);
+        try {
+          await UnderlyingConnection.OpenAsync(cancellationToken);
+          await SqlHelper.NotifyConnectionOpenedAsync(handlers, UnderlyingConnection, false, cancellationToken);
+        }
+        catch (Exception ex) {
+          await SqlHelper.NotifyConnectionOpeningFailedAsync(handlers, UnderlyingConnection, ex, false, cancellationToken);
+          throw;
+        }
+      }
     }
 
     /// <summary>
@@ -249,21 +266,52 @@ public virtual async Task OpenAndInitializeAsync(string initializationScript, Ca
     {
       token.ThrowIfCancellationRequested();
       EnsureIsNotDisposed();
-      await UnderlyingConnection.OpenAsync(token).ConfigureAwait(false);
-      if (string.IsNullOrEmpty(initializationScript)) {
-        return;
-      }
+      var connectionHandlers = Extensions.Get<ConnectionHandlersExtension>();
+      if (connectionHandlers == null) {
+        await UnderlyingConnection.OpenAsync(token).ConfigureAwait(false);
+        if (string.IsNullOrEmpty(initializationScript)) {
+          return;
+        }
 
-      try {
-        var command = UnderlyingConnection.CreateCommand();
-        await using (command.ConfigureAwait(false)) {
-          command.CommandText = initializationScript;
-          await command.ExecuteNonQueryAsync(token).ConfigureAwait(false);
+        try {
+          var command = UnderlyingConnection.CreateCommand();
+          await using (command.ConfigureAwait(false)) {
+            command.CommandText = initializationScript;
+            _ = await command.ExecuteNonQueryAsync(token).ConfigureAwait(false);
+          }
+        }
+        catch (OperationCanceledException) {
+          await UnderlyingConnection.CloseAsync().ConfigureAwait(false);
+          throw;
         }
       }
-      catch (OperationCanceledException) {
-        await UnderlyingConnection.CloseAsync().ConfigureAwait(false);
-        throw;
+      else {
+        var handlers = connectionHandlers.Handlers;
+        await SqlHelper.NotifyConnectionOpeningAsync(handlers, UnderlyingConnection, false, token);
+        await UnderlyingConnection.OpenAsync(token).ConfigureAwait(false);
+        if (string.IsNullOrEmpty(initializationScript)) {
+          await SqlHelper.NotifyConnectionOpenedAsync(handlers, UnderlyingConnection, false, token);
+          return;
+        }
+
+        try {
+          await SqlHelper.NotifyConnectionInitializingAsync(handlers, UnderlyingConnection, initializationScript, false, token);
+          var command = UnderlyingConnection.CreateCommand();
+          await using (command.ConfigureAwait(false)) {
+            command.CommandText = initializationScript;
+            _ = await command.ExecuteNonQueryAsync(token).ConfigureAwait(false);
+          }
+          await SqlHelper.NotifyConnectionOpenedAsync(handlers, UnderlyingConnection, false, token);
+        }
+        catch (OperationCanceledException ex) {
+          await SqlHelper.NotifyConnectionOpeningFailedAsync(handlers, UnderlyingConnection, ex, false, token);
+          await UnderlyingConnection.CloseAsync().ConfigureAwait(false);
+          throw;
+        }
+        catch (Exception ex) {
+          await SqlHelper.NotifyConnectionOpeningFailedAsync(handlers, UnderlyingConnection, ex, false, token);
+          throw;
+        }
       }
     }