add if for whitelabel config

EasyMicroservices · Oct 26, 2023 · b2e16ad · b2e16ad
1 parent e5be6e6
commit b2e16ad
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/CSharp/EasyMicroservices.Cores.AspEntityFrameworkCoreApi/StartUpExtensions.cs b/src/CSharp/EasyMicroservices.Cores.AspEntityFrameworkCoreApi/StartUpExtensions.cs
@@ -16,6 +16,7 @@
 using System.Linq;
 using System.Net;
 using System.Net.Mime;
+using System.Text;
 using System.Text.Json;
 using System.Threading.Tasks;
 
@@ -109,8 +110,11 @@ public static async Task Build<TContext>(this IApplicationBuilder app)
                 var dbbuilder = new DatabaseCreator();
                 using var context = scope.ServiceProvider.GetRequiredService<TContext>();
                 dbbuilder.Initialize(context);
-                using var uow = scope.ServiceProvider.GetRequiredService<IUnitOfWork>() as UnitOfWork;
-                await uow.Initialize(MicroserviceName, config.GetValue<string>(ConfigName), typeof(TContext)).ConfigureAwait(false);
+                if (WhiteLabelRoute.HasValue() || ConfigName.HasValue())
+                {
+                    using var uow = scope.ServiceProvider.GetRequiredService<IUnitOfWork>() as UnitOfWork;
+                    await uow.Initialize(MicroserviceName, config.GetValue<string>(ConfigName), typeof(TContext)).ConfigureAwait(false);
+                }
             }
             var build = app.Build();
             app.Run(build);
@@ -179,7 +183,11 @@ public static async Task<WebApplication> Build<TContext>(this WebApplicationBuil
                 using var context = scope.ServiceProvider.GetRequiredService<TContext>();
                 dbbuilder.Initialize(context);
                 using var uow = scope.ServiceProvider.GetRequiredService<IUnitOfWork>() as UnitOfWork;
-                await uow.Initialize(MicroserviceName, WhiteLabelRoute ?? config.GetValue<string>(ConfigName), typeof(TContext)).ConfigureAwait(false);
+                if (WhiteLabelRoute.HasValue() || ConfigName.HasValue())
+                {
+                    var value = WhiteLabelRoute ?? config.GetValue<string>(ConfigName);
+                    await uow.Initialize(MicroserviceName, WhiteLabelRoute ?? config.GetValue<string>(ConfigName), typeof(TContext)).ConfigureAwait(false);
+                }
             }
             return build;
         }