Open a connection to the db if necessary.

leaderboardsgg · Jun 24, 2024 · 784709f · 784709f
1 parent d20276d
commit 784709f
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/LeaderboardBackend/Models/Entities/ApplicationContext.cs b/LeaderboardBackend/Models/Entities/ApplicationContext.cs
@@ -1,3 +1,4 @@
+using System.Data;
 using System.Reflection;
 using Microsoft.EntityFrameworkCore;
 using Npgsql;
@@ -31,9 +32,22 @@ public ApplicationContext(DbContextOptions<ApplicationContext> options)
     public void MigrateDatabase()
     {
         Database.Migrate();
+        bool tempConnection = false;
+        NpgsqlConnection connection = (NpgsqlConnection)Database.GetDbConnection();
+
+        if (connection.State is ConnectionState.Closed)
+        {
+            tempConnection = true;
+            Database.OpenConnection();
+        }
 
         // when new extensions have been enabled by migrations, Npgsql's type cache must be refreshed
-        ((NpgsqlConnection)Database.GetDbConnection()).ReloadTypes();
+        connection.ReloadTypes();
+
+        if (tempConnection)
+        {
+            Database.CloseConnection();
+        }
     }
 
     protected override void OnModelCreating(ModelBuilder modelBuilder)