Add project files.

1 files changed, 69 insertions, 0 deletions

diff --git a/VNLib.Plugins.Extensions.Data/TransactionalDbContext.cs b/VNLib.Plugins.Extensions.Data/TransactionalDbContext.cs
new file mode 100644
index 0000000..8573c8f
--- /dev/null
+++ b/VNLib.Plugins.Extensions.Data/TransactionalDbContext.cs
@@ -0,0 +1,69 @@
+using System;
+using System.Threading;
+using System.Threading.Tasks;
+using Microsoft.EntityFrameworkCore;
+using Microsoft.EntityFrameworkCore.Storage;
+
+namespace VNLib.Plugins.Extensions.Data
+{
+    public abstract class TransactionalDbContext : DbContext, IAsyncDisposable, IDisposable
+    {
+        /// <summary>
+        /// <inheritdoc/>
+        /// </summary>
+        protected TransactionalDbContext()
+        {}
+        /// <summary>
+        /// <inheritdoc/>
+        /// </summary>
+        protected TransactionalDbContext(DbContextOptions options) : base(options)
+        {}
+
+        /// <summary>
+        /// The transaction that was opened on the current context
+        /// </summary>
+        public IDbContextTransaction? Transaction { get; set; }
+        ///<inheritdoc/>
+        public override void Dispose()
+        {
+            //dispose the transaction
+            this.Transaction?.Dispose();
+            base.Dispose();
+        }
+
+        /// <summary>
+        /// Opens a single transaction on the current context. If a transaction is already open, 
+        /// it is disposed and a new transaction is begun.
+        /// </summary>
+        public async Task OpenTransactionAsync(CancellationToken cancellationToken = default)
+        {
+            //open a new transaction on the current database
+            this.Transaction = await base.Database.BeginTransactionAsync(cancellationToken);
+        }
+        /// <summary>
+        /// Invokes the <see cref="IDbContextTransaction.Commit"/> on the current context
+        /// </summary>
+        public Task CommitTransactionAsync(CancellationToken token = default)
+        {
+            return Transaction != null ? Transaction.CommitAsync(token) : Task.CompletedTask;
+        }
+        /// <summary>
+        /// Invokes the <see cref="IDbContextTransaction.Rollback"/> on the current context
+        /// </summary>
+        public Task RollbackTransctionAsync(CancellationToken token = default)
+        {
+            return Transaction != null ? Transaction.RollbackAsync(token) : Task.CompletedTask;
+        }
+        ///<inheritdoc/>
+        public override async ValueTask DisposeAsync()
+        {
+            //If transaction has been created, dispose the transaction
+            if(this.Transaction != null)
+            {
+                await this.Transaction.DisposeAsync();
+            }
+            await base.DisposeAsync();
+            GC.SuppressFinalize(this);
+        }
+    }
+}
\ No newline at end of file