From 0374e054020c45ca0bb7239ef3564255b4f41717 Mon Sep 17 00:00:00 2001
From: Nguyen Xuan Nhan <nguyenxuannhan407@gmail.com>
Date: Mon, 26 Aug 2024 00:18:01 +0700
Subject: [PATCH] refactor: Migration memory outbox to mongodb outbox for
 Rating service

---
 Directory.Packages.props                      |  2 +-
 src/BookWorm.Basket/Extensions/Extensions.cs  |  5 ++-
 .../Extensions/Extensions.cs                  |  4 ++-
 src/BookWorm.Rating/BookWorm.Rating.csproj    |  2 +-
 src/BookWorm.Rating/Extensions/Extensions.cs  | 32 ++++++++++---------
 5 files changed, 26 insertions(+), 19 deletions(-)

diff --git a/Directory.Packages.props b/Directory.Packages.props
index 3f545e9..5d6f4d8 100644
--- a/Directory.Packages.props
+++ b/Directory.Packages.props
@@ -24,7 +24,6 @@
     <PackageVersion Include="Aspire.Hosting.Redis" Version="$(AspireVersion)" />
     <PackageVersion Include="Aspire.Npgsql" Version="$(AspireVersion)" />
     <PackageVersion Include="Aspire.Npgsql.EntityFrameworkCore.PostgreSQL" Version="$(AspireVersion)" />
-    <PackageVersion Include="Aspire.MongoDB.Driver" Version="$(AspireVersion)" />
     <PackageVersion Include="Aspire.Azure.Storage.Blobs" Version="$(AspireVersion)" />
     <PackageVersion Include="Aspire.Azure.AI.OpenAI" Version="$(AspireUnstablePackagesVersion)" />
     <PackageVersion Include="Aspire.StackExchange.Redis" Version="$(AspireVersion)" />
@@ -94,6 +93,7 @@
     <!-- MassTransit -->
     <PackageVersion Include="MassTransit" Version="$(MassTransitVersion)" />
     <PackageVersion Include="MassTransit.RabbitMQ" Version="$(MassTransitVersion)" />
+    <PackageVersion Include="MassTransit.MongoDb" Version="$(MassTransitVersion)" />
     <PackageVersion Include="MassTransit.EntityFrameworkCore" Version="$(MassTransitVersion)" />
     <!-- Miscellaneous -->
     <PackageVersion Include="Scrutor" Version="4.2.2" />
diff --git a/src/BookWorm.Basket/Extensions/Extensions.cs b/src/BookWorm.Basket/Extensions/Extensions.cs
index 90adfb8..c802e4e 100644
--- a/src/BookWorm.Basket/Extensions/Extensions.cs
+++ b/src/BookWorm.Basket/Extensions/Extensions.cs
@@ -21,7 +21,10 @@ public static void AddApplicationServices(this IHostApplicationBuilder builder)
 
         builder.Services.AddValidatorsFromAssemblyContaining<global::Program>(includeInternalTypes: true);
 
-        builder.AddRabbitMqEventBus(typeof(global::Program), cfg => cfg.AddInMemoryInboxOutbox());
+        builder.AddRabbitMqEventBus(typeof(global::Program), cfg =>
+        {
+            cfg.AddInMemoryInboxOutbox();
+        });
 
         builder.Services.AddSingleton<IActivityScope, ActivityScope>();
         builder.Services.AddSingleton<CommandHandlerMetrics>();
diff --git a/src/BookWorm.Ordering/Extensions/Extensions.cs b/src/BookWorm.Ordering/Extensions/Extensions.cs
index 6008004..4631fa7 100644
--- a/src/BookWorm.Ordering/Extensions/Extensions.cs
+++ b/src/BookWorm.Ordering/Extensions/Extensions.cs
@@ -29,6 +29,7 @@ public static void AddApplicationServices(this IHostApplicationBuilder builder)
         builder.Services.AddValidatorsFromAssemblyContaining<global::Program>(includeInternalTypes: true);
 
         builder.AddRabbitMqEventBus(typeof(global::Program), cfg =>
+        {
             cfg.AddEntityFrameworkOutbox<OrderingContext>(o =>
             {
                 o.QueryDelay = TimeSpan.FromSeconds(1);
@@ -36,7 +37,8 @@ public static void AddApplicationServices(this IHostApplicationBuilder builder)
                 o.UsePostgres();
 
                 o.UseBusOutbox();
-            }));
+            });
+        });
 
         builder.Services.AddMarten(_ =>
             {
diff --git a/src/BookWorm.Rating/BookWorm.Rating.csproj b/src/BookWorm.Rating/BookWorm.Rating.csproj
index c40a08c..8487eb4 100644
--- a/src/BookWorm.Rating/BookWorm.Rating.csproj
+++ b/src/BookWorm.Rating/BookWorm.Rating.csproj
@@ -3,7 +3,7 @@
   <ItemGroup>
     <PackageReference Include="Microsoft.AspNetCore.OpenApi" />
     <PackageReference Include="Swashbuckle.AspNetCore" />
-    <PackageReference Include="Aspire.MongoDB.Driver" />
+    <PackageReference Include="MassTransit.MongoDb" />
   </ItemGroup>    
 
   <ItemGroup>
diff --git a/src/BookWorm.Rating/Extensions/Extensions.cs b/src/BookWorm.Rating/Extensions/Extensions.cs
index 7de583c..2c0f879 100644
--- a/src/BookWorm.Rating/Extensions/Extensions.cs
+++ b/src/BookWorm.Rating/Extensions/Extensions.cs
@@ -18,7 +18,6 @@ public static void AddApplicationServices(this IHostApplicationBuilder builder)
         builder.Services.AddProblemDetails();
 
         builder.AddDefaultAuthentication();
-        builder.AddMongoDBClient(ServiceName.Database.Rating);
         builder.Services.AddMediatR(cfg =>
         {
             cfg.RegisterServicesFromAssemblyContaining<global::Program>();
@@ -27,9 +26,25 @@ public static void AddApplicationServices(this IHostApplicationBuilder builder)
             cfg.AddOpenBehavior(typeof(MetricsBehavior<,>));
         });
 
+        var conn = builder.Configuration.GetConnectionString(ServiceName.Database.Rating);
+
+        builder.Services.AddSingleton<IMongoClient>(_ => new MongoClient(conn));
+
+        builder.Services.AddSingleton(provider =>
+            provider.GetService<IMongoClient>()!.GetDatabase(MongoUrl.Create(conn).DatabaseName));
+
+        builder.Services.AddScoped(provider =>
+            provider.GetService<IMongoDatabase>()!.GetCollection<Feedback>(nameof(Feedback)));
+
         builder.AddRabbitMqEventBus(typeof(global::Program), cfg =>
         {
-            cfg.AddInMemoryInboxOutbox();
+            cfg.AddMongoDbOutbox(o =>
+            {
+                o.DisableInboxCleanupService();
+                o.ClientFactory(provider => provider.GetRequiredService<IMongoClient>());
+                o.DatabaseFactory(provider => provider.GetRequiredService<IMongoDatabase>());
+                o.UseBusOutbox(bo => bo.DisableDeliveryService());
+            });
         });
 
         builder.Services.AddValidatorsFromAssemblyContaining<global::Program>(includeInternalTypes: true);
@@ -38,19 +53,6 @@ public static void AddApplicationServices(this IHostApplicationBuilder builder)
         builder.Services.AddSingleton<CommandHandlerMetrics>();
         builder.Services.AddSingleton<QueryHandlerMetrics>();
 
-        builder.Services.AddSingleton(serviceProvider =>
-        {
-            var url = builder.Configuration.GetConnectionString(ServiceName.Database.Rating);
-            var client = serviceProvider.GetService<IMongoClient>();
-            return client!.GetDatabase(MongoUrl.Create(url).DatabaseName);
-        });
-
-        builder.Services.AddScoped(serviceProvider =>
-        {
-            var database = serviceProvider.GetService<IMongoDatabase>();
-            return database!.GetCollection<Feedback>(nameof(Feedback));
-        });
-
         builder.Services.AddScoped<IRatingRepository, RatingRepository>();
 
         builder.AddVersioning();