feat: Add add distributed transaction to EfTxBehavior

deployments/docker-compose/docker-compose.yaml

@@ -1,24 +1,27 @@
 version: "3.3"
 services:
 
-    ######################################################
-    #  Postgres
-    ######################################################
+   #######################################################
+   #  Postgres
+   ######################################################
   postgres:
-    image: postgres:latest
-    container_name: postgres
-    restart: on-failure
-    ports:
-      - '5432:5432'
-    environment:
-      - POSTGRES_USER=postgres
-      - POSTGRES_PASSWORD=postgres
-    command:
-      - "postgres"
-      - "-c"
-      - "wal_level=logical"
-    networks:
-      - booking
+   image: postgres:latest
+   container_name: postgres
+   restart: on-failure
+   ports:
+     - '5432:5432'
+   environment:
+     - POSTGRES_USER=postgres
+     - POSTGRES_PASSWORD=postgres
+   command:
+     - "postgres"
+     - "-c"
+     - "wal_level=logical"
+     - "-c"
+     - "max_prepared_transactions=10"
+   networks:
+     - booking
+
 
 
     #######################################################

deployments/docker-compose/infrastracture.yaml

@@ -31,6 +31,8 @@ services:
      - "postgres"
      - "-c"
      - "wal_level=logical"
+     - "-c"
+     - "max_prepared_transactions=10"
    networks:
      - booking
 

deployments/kubernetes/booking-microservices.yml

@@ -390,7 +390,13 @@ metadata:
   name: postgres
 spec:
   containers:
-    - env:
+    - args:
+        - postgres
+        - -c
+        - wal_level=logical
+        - -c
+        - max_prepared_transactions=10  
+      env:
         - name: POSTGRES_PASSWORD
           value: postgres
         - name: POSTGRES_USER

src/BuildingBlocks/EFCore/AppDbContextBase.cs

@@ -1,16 +1,18 @@
 namespace BuildingBlocks.EFCore;
 
 using System.Collections.Immutable;
-using BuildingBlocks.Core.Event;
-using BuildingBlocks.Core.Model;
+using Core.Event;
+using Core.Model;
 using Microsoft.EntityFrameworkCore;
-using System.Data;
+using Microsoft.EntityFrameworkCore.Storage;
 using Web;
 using Exception = System.Exception;
+using IsolationLevel = System.Data.IsolationLevel;
 
 public abstract class AppDbContextBase : DbContext, IDbContext
 {
     private readonly ICurrentUserProvider _currentUserProvider;
+    private IDbContextTransaction? _currentTransaction;
 
     protected AppDbContextBase(DbContextOptions options, ICurrentUserProvider currentUserProvider) :
         base(options)
@@ -22,13 +24,53 @@ public abstract class AppDbContextBase : DbContext, IDbContext
     {
     }
 
+    public async Task BeginTransactionalAsync(CancellationToken cancellationToken = default)
+    {
+        _currentTransaction ??= await Database.BeginTransactionAsync(IsolationLevel.ReadCommitted, cancellationToken);
+    }
+
+    //ref: https://learn.microsoft.com/en-us/ef/core/miscellaneous/connection-resiliency#execution-strategies-and-transactions
+    public async Task CommitTransactionalAsync(CancellationToken cancellationToken = default)
+    {
+        try
+        {
+            await SaveChangesAsync(cancellationToken);
+            await _currentTransaction?.CommitAsync(cancellationToken)!;
+        }
+        catch
+        {
+            await _currentTransaction?.RollbackAsync(cancellationToken)!;
+            throw;
+        }
+        finally
+        {
+            _currentTransaction?.Dispose();
+            _currentTransaction = null;
+        }
+    }
+
+
+    public async Task RollbackTransactionAsync(CancellationToken cancellationToken = default)
+    {
+        try
+        {
+            await _currentTransaction?.RollbackAsync(cancellationToken)!;
+        }
+        finally
+        {
+            _currentTransaction?.Dispose();
+            _currentTransaction = null;
+        }
+    }
+
     //ref: https://learn.microsoft.com/en-us/ef/core/miscellaneous/connection-resiliency#execution-strategies-and-transactions
     public Task ExecuteTransactionalAsync(CancellationToken cancellationToken = default)
     {
         var strategy = Database.CreateExecutionStrategy();
         return strategy.ExecuteAsync(async () =>
         {
-            await using var transaction = await Database.BeginTransactionAsync(IsolationLevel.ReadCommitted, cancellationToken);
+            await using var transaction =
+                await Database.BeginTransactionAsync(IsolationLevel.ReadCommitted, cancellationToken);
             try
             {
                 await SaveChangesAsync(cancellationToken);

src/BuildingBlocks/EFCore/EfTxBehavior.cs

@@ -5,21 +5,27 @@ using Microsoft.Extensions.Logging;
 
 namespace BuildingBlocks.EFCore;
 
+using System.Transactions;
+using PersistMessageProcessor;
+
 public class EfTxBehavior<TRequest, TResponse> : IPipelineBehavior<TRequest, TResponse>
     where TRequest : notnull, IRequest<TResponse>
     where TResponse : notnull
 {
     private readonly ILogger<EfTxBehavior<TRequest, TResponse>> _logger;
     private readonly IDbContext _dbContextBase;
+    private readonly IPersistMessageDbContext _persistMessageDbContext;
     private readonly IEventDispatcher _eventDispatcher;
 
     public EfTxBehavior(
         ILogger<EfTxBehavior<TRequest, TResponse>> logger,
         IDbContext dbContextBase,
+        IPersistMessageDbContext persistMessageDbContext,
         IEventDispatcher eventDispatcher)
     {
         _logger = logger;
         _dbContextBase = dbContextBase;
+        _persistMessageDbContext = persistMessageDbContext;
         _eventDispatcher = eventDispatcher;
     }
 
@@ -42,6 +48,10 @@ public class EfTxBehavior<TRequest, TResponse> : IPipelineBehavior<TRequest, TRe
             nameof(EfTxBehavior<TRequest, TResponse>),
             typeof(TRequest).FullName);
 
+        // ref: https://learn.microsoft.com/en-us/ef/core/saving/transactions#using-systemtransactions
+        using var scope = new TransactionScope(TransactionScopeOption.Required,
+            new TransactionOptions { IsolationLevel = IsolationLevel.ReadCommitted },
+            TransactionScopeAsyncFlowOption.Enabled);
 
         var response = await next();
 
@@ -55,11 +65,16 @@ public class EfTxBehavior<TRequest, TResponse> : IPipelineBehavior<TRequest, TRe
             var domainEvents = _dbContextBase.GetDomainEvents();
 
             if (domainEvents is null || !domainEvents.Any())
+            {
                 return response;
-
-            await _dbContextBase.ExecuteTransactionalAsync(cancellationToken);
+            }
 
             await _eventDispatcher.SendAsync(domainEvents.ToArray(), typeof(TRequest), cancellationToken);
+
+            await _dbContextBase.SaveChangesAsync(cancellationToken);
+            await _persistMessageDbContext.SaveChangesAsync(cancellationToken);
+
+            scope.Complete();
         }
     }
 }

src/BuildingBlocks/EFCore/Extensions.cs

@@ -36,7 +36,7 @@ public static class Extensions
                     {
                         dbOptions.MigrationsAssembly(typeof(TContext).Assembly.GetName().Name);
                         //ref: https://learn.microsoft.com/en-us/ef/core/miscellaneous/connection-resiliency
-                        dbOptions.EnableRetryOnFailure(3, TimeSpan.FromSeconds(1), null);
+                        // dbOptions.EnableRetryOnFailure(3, TimeSpan.FromSeconds(1), null);
                     })
                 // https://github.com/efcore/EFCore.NamingConventions
                 .UseSnakeCaseNamingConvention();

src/BuildingBlocks/EFCore/IDbContext.cs

@@ -7,6 +7,9 @@ public interface IDbContext
 {
     DbSet<TEntity> Set<TEntity>() where TEntity : class;
     IReadOnlyList<IDomainEvent> GetDomainEvents();
+    public Task BeginTransactionalAsync(CancellationToken cancellationToken = default);
+    public Task CommitTransactionalAsync(CancellationToken cancellationToken = default);
+    public Task RollbackTransactionAsync(CancellationToken cancellationToken = default);
     Task ExecuteTransactionalAsync(CancellationToken cancellationToken = default);
     Task<int> SaveChangesAsync(CancellationToken cancellationToken = default);
 }

src/BuildingBlocks/PersistMessageProcessor/Data/PersistMessageDbContext.cs

@@ -8,7 +8,6 @@ using Configurations;
 using Core.Model;
 using global::Polly;
 using Microsoft.Extensions.Logging;
-using Newtonsoft.Json;
 using Exception = System.Exception;
 
 public class PersistMessageDbContext : DbContext, IPersistMessageDbContext

src/BuildingBlocks/TestBase/TestContainers.cs

@@ -1,4 +1,5 @@
 namespace BuildingBlocks.TestBase;
+
 using Testcontainers.EventStoreDb;
 using Testcontainers.MongoDb;
 using Testcontainers.PostgreSql;
@@ -7,21 +8,25 @@ using Web;
 
 public static class TestContainers
 {
-    public static RabbitMqContainerOptions RabbitMqContainerConfiguration { get;}
-    public static PostgresContainerOptions PostgresContainerConfiguration { get;}
-    public static PostgresPersistContainerOptions PostgresPersistContainerConfiguration { get;}
-    public static MongoContainerOptions MongoContainerConfiguration { get;}
-    public static EventStoreContainerOptions EventStoreContainerConfiguration { get;}
+    public static RabbitMqContainerOptions RabbitMqContainerConfiguration { get; }
+    public static PostgresContainerOptions PostgresContainerConfiguration { get; }
+    public static PostgresPersistContainerOptions PostgresPersistContainerConfiguration { get; }
+    public static MongoContainerOptions MongoContainerConfiguration { get; }
+    public static EventStoreContainerOptions EventStoreContainerConfiguration { get; }
 
     static TestContainers()
     {
         var configuration = ConfigurationHelper.GetConfiguration();
 
-        RabbitMqContainerConfiguration = configuration.GetOptions<RabbitMqContainerOptions>(nameof(RabbitMqContainerOptions));
-        PostgresContainerConfiguration = configuration.GetOptions<PostgresContainerOptions>(nameof(PostgresContainerOptions));
-        PostgresPersistContainerConfiguration = configuration.GetOptions<PostgresPersistContainerOptions>(nameof(PostgresPersistContainerOptions));
+        RabbitMqContainerConfiguration =
+            configuration.GetOptions<RabbitMqContainerOptions>(nameof(RabbitMqContainerOptions));
+        PostgresContainerConfiguration =
+            configuration.GetOptions<PostgresContainerOptions>(nameof(PostgresContainerOptions));
+        PostgresPersistContainerConfiguration =
+            configuration.GetOptions<PostgresPersistContainerOptions>(nameof(PostgresPersistContainerOptions));
         MongoContainerConfiguration = configuration.GetOptions<MongoContainerOptions>(nameof(MongoContainerOptions));
-        EventStoreContainerConfiguration = configuration.GetOptions<EventStoreContainerOptions>(nameof(EventStoreContainerOptions));
+        EventStoreContainerConfiguration =
+            configuration.GetOptions<EventStoreContainerOptions>(nameof(EventStoreContainerOptions));
     }
 
     public static PostgreSqlContainer PostgresTestContainer()
@@ -34,6 +39,7 @@ public static class TestContainers
         var builder = baseBuilder
             .WithImage(PostgresContainerConfiguration.ImageName)
             .WithName(PostgresContainerConfiguration.Name)
+            .WithCommand(new string[2] { "-c", "max_prepared_transactions=10" })
             .WithPortBinding(PostgresContainerConfiguration.Port, true)
             .Build();
 
@@ -50,6 +56,7 @@ public static class TestContainers
         var builder = baseBuilder
             .WithImage(PostgresPersistContainerConfiguration.ImageName)
             .WithName(PostgresPersistContainerConfiguration.Name)
+            .WithCommand(new string[2] { "-c", "max_prepared_transactions=10" })
             .WithPortBinding(PostgresPersistContainerConfiguration.Port, true)
             .Build();
 

src/Services/Flight/src/Flight/FlightCreatedConsumerHandler.cs

@@ -5,7 +5,7 @@ using MassTransit;
 
 namespace Flight;
 
-public class CreateFlightConsumerHandler : IConsumer<FlightCreated>
+public class FlightCreatedConsumerHandler : IConsumer<FlightCreated>
 {
     public Task Consume(ConsumeContext<FlightCreated> context)
     {

src/Services/Flight/src/Flight/Flights/Features/CreatingFlight/V1/CreateFlight.cs

@@ -54,7 +54,7 @@ public class CreateFlightEndpoint : IMinimalEndpoint
 
                 return Results.CreatedAtRoute("GetFlightById", new { id = result.Id }, response);
             })
-            // .RequireAuthorization()
+            .RequireAuthorization()
             .WithName("CreateFlight")
             .WithApiVersionSet(builder.NewApiVersionSet("Flight").Build())
             .Produces<CreateFlightResponseDto>(StatusCodes.Status201Created)

src/Services/Identity/src/Identity/Data/IdentityContext.cs

@@ -9,18 +9,19 @@ using BuildingBlocks.Core.Event;
 using BuildingBlocks.Core.Model;
 using BuildingBlocks.EFCore;
 using Identity.Identity.Models;
-using Microsoft.AspNetCore.Http;
 using Microsoft.AspNetCore.Identity.EntityFrameworkCore;
 using Microsoft.EntityFrameworkCore;
-using Microsoft.EntityFrameworkCore.Storage;
 
 namespace Identity.Data;
 
 using System;
+using Microsoft.EntityFrameworkCore.Storage;
 
 public sealed class IdentityContext : IdentityDbContext<User, Role, Guid,
     UserClaim, UserRole, UserLogin, RoleClaim, UserToken>, IDbContext
 {
+    private IDbContextTransaction? _currentTransaction;
+
     public IdentityContext(DbContextOptions<IdentityContext> options) : base(options)
     {
     }
@@ -33,6 +34,45 @@ public sealed class IdentityContext : IdentityDbContext<User, Role, Guid,
         builder.ToSnakeCaseTables();
     }
 
+    public async Task BeginTransactionalAsync(CancellationToken cancellationToken = default)
+    {
+        _currentTransaction ??= await Database.BeginTransactionAsync(IsolationLevel.ReadCommitted, cancellationToken);
+    }
+
+    //ref: https://learn.microsoft.com/en-us/ef/core/miscellaneous/connection-resiliency#execution-strategies-and-transactions
+    public async Task CommitTransactionalAsync(CancellationToken cancellationToken = default)
+    {
+        try
+        {
+            await SaveChangesAsync(cancellationToken);
+            await _currentTransaction?.CommitAsync(cancellationToken)!;
+        }
+        catch
+        {
+            await _currentTransaction?.RollbackAsync(cancellationToken)!;
+            throw;
+        }
+        finally
+        {
+            _currentTransaction?.Dispose();
+            _currentTransaction = null;
+        }
+    }
+
+
+    public async Task RollbackTransactionAsync(CancellationToken cancellationToken = default)
+    {
+        try
+        {
+            await _currentTransaction?.RollbackAsync(cancellationToken)!;
+        }
+        finally
+        {
+            _currentTransaction?.Dispose();
+            _currentTransaction = null;
+        }
+    }
+
     //ref: https://learn.microsoft.com/en-us/ef/core/miscellaneous/connection-resiliency#execution-strategies-and-transactions
     public Task ExecuteTransactionalAsync(CancellationToken cancellationToken = default)
     {

src/Services/Identity/src/Identity/Identity/Features/RegisteringNewUser/V1/RegisterNewUser.cs

@@ -46,6 +46,7 @@ public class RegisterNewUserEndpoint : IMinimalEndpoint
 
                 return Results.Ok(response);
             })
+            .RequireAuthorization()
             .WithName("RegisterUser")
             .WithApiVersionSet(builder.NewApiVersionSet("Identity").Build())
             .Produces<RegisterNewUserResponseDto>()

src/Services/Identity/src/Identity/UserCreatedConsumerHandler.cs

@@ -0,0 +1,14 @@
+namespace Identity;
+
+using System;
+using System.Threading.Tasks;
+using BuildingBlocks.Contracts.EventBus.Messages;
+using MassTransit;
+
+public class UserCreatedConsumerHandler : IConsumer<UserCreated>
+{
+    public Task Consume(ConsumeContext<UserCreated> context)
+    {
+        Console.WriteLine("It's for test"); return Task.CompletedTask;
+    }
+}