From aca22613d7be81bc11f4b73a08dc201ecb3dd0eb Mon Sep 17 00:00:00 2001
From: JT <Hawxy@users.noreply.github.com>
Date: Tue, 19 Nov 2024 18:23:34 +0800
Subject: [PATCH] Permit database implementations to override the base's
 connection source

---
 src/Weasel.Core/Migrations/DatabaseBase.cs  | 7 +------
 src/Weasel.Postgresql/PostgresqlDatabase.cs | 5 +++++
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/src/Weasel.Core/Migrations/DatabaseBase.cs b/src/Weasel.Core/Migrations/DatabaseBase.cs
index 13fd8b5e..e8259c20 100644
--- 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
index 70c1cb1d..053356ab 100644
--- 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();