Ensure the primary node is used in multi-host environments

JasperFx · Nov 4, 2024 · 2cca2f4 · 2cca2f4
1 parent 37cdaa9
commit 2cca2f4
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/Persistence/Wolverine.Postgresql/PostgresqlMessageStore.cs b/src/Persistence/Wolverine.Postgresql/PostgresqlMessageStore.cs
@@ -47,10 +47,17 @@ internal class PostgresqlMessageStore : MessageDatabase<NpgsqlConnection>, IData
 
 
     public PostgresqlMessageStore(DatabaseSettings databaseSettings, DurabilitySettings settings, NpgsqlDataSource dataSource,
-        ILogger<PostgresqlMessageStore> logger) : this(databaseSettings, settings, dataSource, logger, Array.Empty<SagaTableDefinition>())
+        ILogger<PostgresqlMessageStore> logger) : this(databaseSettings, settings, GetPrimaryNpgsqlNodeIfPossible(dataSource), logger, Array.Empty<SagaTableDefinition>())
     {
     }
 
+    private static NpgsqlDataSource GetPrimaryNpgsqlNodeIfPossible(NpgsqlDataSource dataSource)
+    {
+        if (dataSource is NpgsqlMultiHostDataSource multiHost)
+            return multiHost.WithTargetSession(TargetSessionAttributes.Primary);
+        return dataSource;
+    }
+
     public PostgresqlMessageStore(DatabaseSettings databaseSettings, DurabilitySettings settings, NpgsqlDataSource dataSource,
         ILogger<PostgresqlMessageStore> logger, IEnumerable<SagaTableDefinition> sagaTypes) : base(databaseSettings, dataSource,
         settings, logger, new PostgresqlMigrator(), "public")