From ccf2b92fe4495ce8e0471c394b6ae5d161ea4149 Mon Sep 17 00:00:00 2001
From: Enes Hoxha <eneshoxha@outlook.com>
Date: Thu, 29 Jan 2026 15:59:55 +0100
Subject: [PATCH] Remove Unit of Work infrastructure and registration

Deleted IUnitOfWork and related interfaces and classes, and removed UnitOfWork registration from DI setup. Transaction management via Unit of Work is no longer supported in the codebase.
---
 .../ServiceCollectionExtensions.cs            |  9 ---
 .../Infrastructure/IUnitOfWork.cs             | 32 ----------
 .../Infrastructure/UnitOfWork.cs              | 61 -------------------
 3 files changed, 102 deletions(-)
 delete mode 100644 src/Cortex.Mediator/Infrastructure/IUnitOfWork.cs
 delete mode 100644 src/Cortex.Mediator/Infrastructure/UnitOfWork.cs

diff --git a/src/Cortex.Mediator/DependencyInjection/ServiceCollectionExtensions.cs b/src/Cortex.Mediator/DependencyInjection/ServiceCollectionExtensions.cs
index 70c888d..fcd40c2 100644
--- a/src/Cortex.Mediator/DependencyInjection/ServiceCollectionExtensions.cs
+++ b/src/Cortex.Mediator/DependencyInjection/ServiceCollectionExtensions.cs
@@ -1,5 +1,4 @@
 using Cortex.Mediator.Commands;
-using Cortex.Mediator.Infrastructure;
 using Cortex.Mediator.Notifications;
 using Cortex.Mediator.Processors;
 using Cortex.Mediator.Queries;
@@ -28,8 +27,6 @@ public static IServiceCollection AddCortexMediator(
             // Validation has been removed for issue #118
             //services.AddValidatorsFromAssemblies(handlerAssemblyMarkerTypes.Select(t => t.Assembly));
 
-            services.AddUnitOfWork();
-
             RegisterHandlers(services, handlerAssemblyMarkerTypes, options);
             RegisterProcessors(services, handlerAssemblyMarkerTypes, options);
             RegisterPipelineBehaviors(services, options);
@@ -161,11 +158,5 @@ private static void RegisterPipelineBehaviors(IServiceCollection services, Media
                 services.AddTransient(typeof(IStreamQueryPipelineBehavior<,>), behaviorType);
             }
         }
-
-        private static void AddUnitOfWork(this IServiceCollection services)
-        {
-            services.AddScoped<IUnitOfWork>(provider =>
-                new UnitOfWork(provider.GetRequiredService<IDbConnection>()));
-        }
     }
 }
diff --git a/src/Cortex.Mediator/Infrastructure/IUnitOfWork.cs b/src/Cortex.Mediator/Infrastructure/IUnitOfWork.cs
deleted file mode 100644
index 60c6acd..0000000
--- a/src/Cortex.Mediator/Infrastructure/IUnitOfWork.cs
+++ /dev/null
@@ -1,32 +0,0 @@
-using System;
-using System.Threading.Tasks;
-
-namespace Cortex.Mediator.Infrastructure
-{
-    /// <summary>
-    /// Represents a unit of work for transaction management.
-    /// </summary>
-    public interface IUnitOfWork
-    {
-        /// <summary>
-        /// Begins a new transaction.
-        /// </summary>
-        Task<IUnitOfWorkTransaction> BeginTransactionAsync();
-    }
-
-    /// <summary>
-    /// Represents a transaction within a unit of work.
-    /// </summary>
-    public interface IUnitOfWorkTransaction : IAsyncDisposable
-    {
-        /// <summary>
-        /// Commits the transaction.
-        /// </summary>
-        Task CommitAsync();
-
-        /// <summary>
-        /// Rolls back the transaction.
-        /// </summary>
-        Task RollbackAsync();
-    }
-}
diff --git a/src/Cortex.Mediator/Infrastructure/UnitOfWork.cs b/src/Cortex.Mediator/Infrastructure/UnitOfWork.cs
deleted file mode 100644
index 301bf78..0000000
--- a/src/Cortex.Mediator/Infrastructure/UnitOfWork.cs
+++ /dev/null
@@ -1,61 +0,0 @@
-using System.Data;
-using System.Threading.Tasks;
-
-namespace Cortex.Mediator.Infrastructure
-{
-    /// <summary>
-    /// Default implementation of IUnitOfWork using System.Data.
-    /// </summary>
-    public class UnitOfWork : IUnitOfWork
-    {
-        private readonly IDbConnection _connection;
-
-        public UnitOfWork(IDbConnection connection)
-        {
-            _connection = connection;
-        }
-
-        public async Task<IUnitOfWorkTransaction> BeginTransactionAsync()
-        {
-            if (_connection.State != ConnectionState.Open)
-            {
-                _connection.Open();
-            }
-
-            var transaction = _connection.BeginTransaction();
-            return new UnitOfWorkTransaction(transaction);
-        }
-
-        private class UnitOfWorkTransaction : IUnitOfWorkTransaction
-        {
-            private readonly IDbTransaction _transaction;
-            private bool _disposed;
-
-            public UnitOfWorkTransaction(IDbTransaction transaction)
-            {
-                _transaction = transaction;
-            }
-
-            public Task CommitAsync()
-            {
-                _transaction.Commit();
-                return Task.CompletedTask;
-            }
-
-            public Task RollbackAsync()
-            {
-                _transaction.Rollback();
-                return Task.CompletedTask;
-            }
-
-            public async ValueTask DisposeAsync()
-            {
-                if (_disposed) return;
-
-                _transaction.Dispose();
-                _disposed = true;
-                await Task.CompletedTask;
-            }
-        }
-    }
-}