Add the async transaction related storage driver operations

Author: AlexUstinov

Orm/Xtensive.Orm/Orm/Providers/StorageDriver.Operations.cs

@@ -195,9 +195,7 @@ public void BeginTransaction(Session session, SqlConnection connection, Isolatio
           isolationLevel);
       }
 
-      if (isolationLevel == null) {
-        isolationLevel = IsolationLevelConverter.Convert(GetConfiguration(session).DefaultIsolationLevel);
-      }
+      isolationLevel ??= IsolationLevelConverter.Convert(GetConfiguration(session).DefaultIsolationLevel);
 
       try {
         connection.BeginTransaction(isolationLevel.Value);
@@ -207,6 +205,24 @@ public void BeginTransaction(Session session, SqlConnection connection, Isolatio
       }
     }
 
+    public async ValueTask BeginTransactionAsync(
+      Session session, SqlConnection connection, IsolationLevel? isolationLevel, CancellationToken token = default)
+    {
+      if (isLoggingEnabled) {
+        SqlLog.Info(Strings.LogSessionXBeginningTransactionWithYIsolationLevel, session.ToStringSafely(),
+          isolationLevel);
+      }
+
+      isolationLevel ??= IsolationLevelConverter.Convert(GetConfiguration(session).DefaultIsolationLevel);
+
+      try {
+        await connection.BeginTransactionAsync(isolationLevel.Value, token).ConfigureAwait(false);
+      }
+      catch (Exception exception) {
+        throw ExceptionBuilder.BuildException(exception);
+      }
+    }
+
     public void CommitTransaction(Session session, SqlConnection connection)
     {
       if (isLoggingEnabled) {
@@ -221,6 +237,21 @@ public void CommitTransaction(Session session, SqlConnection connection)
       }
     }
 
+    public async ValueTask CommitTransactionAsync(
+      Session session, SqlConnection connection, CancellationToken token = default)
+    {
+      if (isLoggingEnabled) {
+        SqlLog.Info(Strings.LogSessionXCommitTransaction, session.ToStringSafely());
+      }
+
+      try {
+        await connection.CommitAsync(token).ConfigureAwait(false);
+      }
+      catch (Exception exception) {
+        throw ExceptionBuilder.BuildException(exception);
+      }
+    }
+
     public void RollbackTransaction(Session session, SqlConnection connection)
     {
       if (isLoggingEnabled) {
@@ -235,14 +266,29 @@ public void RollbackTransaction(Session session, SqlConnection connection)
       }
     }
 
+    public async ValueTask RollbackTransactionAsync(
+      Session session, SqlConnection connection, CancellationToken token = default)
+    {
+      if (isLoggingEnabled) {
+        SqlLog.Info(Strings.LogSessionXRollbackTransaction, session.ToStringSafely());
+      }
+
+      try {
+        await connection.RollbackAsync(token).ConfigureAwait(false);
+      }
+      catch (Exception exception) {
+        throw ExceptionBuilder.BuildException(exception);
+      }
+    }
+
     public void MakeSavepoint(Session session, SqlConnection connection, string name)
     {
       if (isLoggingEnabled) {
         SqlLog.Info(Strings.LogSessionXMakeSavepointY, session.ToStringSafely(), name);
       }
 
       if (!hasSavepoints) {
-        return; // Driver does not support savepoints, so let's fail later (on rollback)
+        return; // Driver does not support save points, so let's fail later (on rollback)
       }
 
       try {
@@ -253,6 +299,25 @@ public void MakeSavepoint(Session session, SqlConnection connection, string name
       }
     }
 
+    public async ValueTask MakeSavepointAsync(
+      Session session, SqlConnection connection, string name, CancellationToken token = default)
+    {
+      if (isLoggingEnabled) {
+        SqlLog.Info(Strings.LogSessionXMakeSavepointY, session.ToStringSafely(), name);
+      }
+
+      if (!hasSavepoints) {
+        return; // Driver does not support save points, so let's fail later (on rollback)
+      }
+
+      try {
+        await connection.MakeSavepointAsync(name, token).ConfigureAwait(false);
+      }
+      catch (Exception exception) {
+        throw ExceptionBuilder.BuildException(exception);
+      }
+    }
+
     public void RollbackToSavepoint(Session session, SqlConnection connection, string name)
     {
       if (isLoggingEnabled) {
@@ -270,7 +335,26 @@ public void RollbackToSavepoint(Session session, SqlConnection connection, strin
         throw ExceptionBuilder.BuildException(exception);
       }
     }
- 
+
+    public async ValueTask RollbackToSavepointAsync(
+      Session session, SqlConnection connection, string name, CancellationToken token = default)
+    {
+      if (isLoggingEnabled) {
+        SqlLog.Info(Strings.LogSessionXRollbackToSavepointY, session.ToStringSafely(), name);
+      }
+
+      if (!hasSavepoints) {
+        throw new NotSupportedException(Strings.ExCurrentStorageProviderDoesNotSupportSavepoints);
+      }
+
+      try {
+        await connection.RollbackToSavepointAsync(name, token).ConfigureAwait(false);
+      }
+      catch (Exception exception) {
+        throw ExceptionBuilder.BuildException(exception);
+      }
+    }
+
     public void ReleaseSavepoint(Session session, SqlConnection connection, string name)
     {
       if (isLoggingEnabled) {
@@ -285,6 +369,21 @@ public void ReleaseSavepoint(Session session, SqlConnection connection, string n
       }
     }
 
+    public async ValueTask ReleaseSavepointAsync(
+      Session session, SqlConnection connection, string name, CancellationToken token = default)
+    {
+      if (isLoggingEnabled) {
+        SqlLog.Info(Strings.LogSessionXReleaseSavepointY, session.ToStringSafely(), name);
+      }
+
+      try {
+        await connection.ReleaseSavepointAsync(name, token).ConfigureAwait(false);
+      }
+      catch (Exception exception) {
+        throw ExceptionBuilder.BuildException(exception);
+      }
+    }
+
     #region Sync Execute methods
 
     public int ExecuteNonQuery(Session session, DbCommand command) =>