Permit database implementations to override the base's connection source

JasperFx · Nov 19, 2024 · aca2261 · aca2261
1 parent a836da3
commit aca2261
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/src/Weasel.Core/Migrations/DatabaseBase.cs b/src/Weasel.Core/Migrations/DatabaseBase.cs
@@ -58,11 +58,6 @@ private static TConnection CreateConnection(string connectionString)
     public string Identifier { get; }
 
 
-    TConnection IConnectionSource<TConnection>.CreateConnection()
-    {
-        return _connectionSource();
-    }
-
     /// <summary>
     ///     All referenced schema names by the known objects in this database
     /// </summary>
@@ -235,7 +230,7 @@ public async Task AssertDatabaseMatchesConfigurationAsync(CancellationToken ct =
         }
     }
 
-    public TConnection CreateConnection()
+    public virtual TConnection CreateConnection()
     {
         return _connectionSource();
     }

diff --git a/src/Weasel.Postgresql/PostgresqlDatabase.cs b/src/Weasel.Postgresql/PostgresqlDatabase.cs
@@ -60,6 +60,11 @@ public NpgsqlConnection CreateConnection(TargetSessionAttributes targetSessionAt
         return base.CreateConnection();
     }
 
+    public override NpgsqlConnection CreateConnection()
+    {
+        return CreateConnection();
+    }
+
     public ValueTask DisposeAsync()
     {
         return DataSource.DisposeAsync();