Add subcomponent to conn str

SapiensAnatis · May 10, 2024 · afa2205 · afa2205
1 parent 4aac749
commit afa2205
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 5 deletions.
diff --git a/DragaliaAPI/DragaliaAPI.Database/ApiContextFactory.cs b/DragaliaAPI/DragaliaAPI.Database/ApiContextFactory.cs
@@ -21,7 +21,7 @@ public ApiContext CreateDbContext(string[] args)
             ?.Get<PostgresOptions>();
 
         DbContextOptions<ApiContext> contextOptions = new DbContextOptionsBuilder<ApiContext>()
-            .UseNpgsql(postgresOptions?.GetConnectionString())
+            .UseNpgsql(postgresOptions?.GetConnectionString("IDesignTimeDbContextFactory"))
             .Options;
 
         return new ApiContext(contextOptions, new StubPlayerIdentityService());

diff --git a/DragaliaAPI/DragaliaAPI.Database/DatabaseConfiguration.cs b/DragaliaAPI/DragaliaAPI.Database/DatabaseConfiguration.cs
@@ -24,7 +24,9 @@ PostgresOptions postgresOptions
     {
         services = services
             .AddDbContext<ApiContext>(options =>
-                options.UseNpgsql(postgresOptions.GetConnectionString()).EnableDetailedErrors()
+                options
+                    .UseNpgsql(postgresOptions.GetConnectionString("ApiContext"))
+                    .EnableDetailedErrors()
             )
 #pragma warning disable CS0618 // Type or member is obsolete
             .AddScoped<IDeviceAccountRepository, DeviceAccountRepository>()

diff --git a/DragaliaAPI/DragaliaAPI.Database/PostgresOptions.cs b/DragaliaAPI/DragaliaAPI.Database/PostgresOptions.cs
@@ -16,7 +16,12 @@ public class PostgresOptions
 
     public bool DisableAutoMigration { get; set; }
 
-    public string GetConnectionString()
+    /// <summary>
+    /// Gets a connection string for the current instance.
+    /// </summary>
+    /// <param name="subComponent">An optional suffix to add to the application name of the connection string.</param>
+    /// <returns>The connection string.</returns>
+    public string GetConnectionString(string subComponent)
     {
         NpgsqlConnectionStringBuilder connectionStringBuilder =
             new()
@@ -27,7 +32,7 @@ public string GetConnectionString()
                 Password = this.Password,
                 Database = this.Database,
                 IncludeErrorDetail = true,
-                ApplicationName = "DragaliaAPI",
+                ApplicationName = $"DragaliaAPI_{subComponent}",
             };
 
         return connectionStringBuilder.ConnectionString;

diff --git a/DragaliaAPI/DragaliaAPI/ServiceConfiguration.cs b/DragaliaAPI/DragaliaAPI/ServiceConfiguration.cs
@@ -281,7 +281,7 @@ PostgresOptions postgresOptions
                 .UseSimpleAssemblyNameTypeSerializer()
                 .UseRecommendedSerializerSettings()
                 .UsePostgreSqlStorage(pgCfg =>
-                    pgCfg.UseNpgsqlConnection(postgresOptions.GetConnectionString())
+                    pgCfg.UseNpgsqlConnection(postgresOptions.GetConnectionString("Hangfire"))
                 )
         );