Please work lol

OpenShock · Nov 19, 2024 · 9e31bd5 · 9e31bd5
1 parent d13ad43
commit 9e31bd5
Showing 1 changed file with 8 additions and 13 deletions.
diff --git a/Common/OpenShockServiceHelper.cs b/Common/OpenShockServiceHelper.cs
@@ -5,7 +5,6 @@
 using Microsoft.AspNetCore.Mvc;
 using Microsoft.EntityFrameworkCore;
 using Microsoft.Extensions.DependencyInjection.Extensions;
-using Npgsql;
 using OpenShock.Common.Authentication;
 using OpenShock.Common.Authentication.Handlers;
 using OpenShock.Common.Authentication.Services;
@@ -19,7 +18,6 @@
 using OpenShock.Common.Services.RedisPubSub;
 using OpenShock.Common.Services.Session;
 using OpenTelemetry.Metrics;
-using OpenTelemetry.Trace;
 using Redis.OM;
 using Redis.OM.Contracts;
 using StackExchange.Redis;
@@ -155,19 +153,16 @@ public static ServicesResult AddOpenShockServices(this IServiceCollection servic
 
         // <---- Postgres EF Core ---->
 
-        // How do I do this now with EFCore?!
-#pragma warning disable CS0618
-        NpgsqlConnection.GlobalTypeMapper.MapEnum<ControlType>();
-        NpgsqlConnection.GlobalTypeMapper.MapEnum<PermissionType>();
-        NpgsqlConnection.GlobalTypeMapper.MapEnum<ShockerModelType>();
-        NpgsqlConnection.GlobalTypeMapper.MapEnum<RankType>();
-        NpgsqlConnection.GlobalTypeMapper.MapEnum<OtaUpdateStatus>();
-#pragma warning restore CS0618
-
-
         services.AddDbContextPool<OpenShockContext>(builder =>
         {
-            builder.UseNpgsql(config.Db.Conn);
+            builder.UseNpgsql(config.Db.Conn, optionsBuilder =>
+            {
+                optionsBuilder.MapEnum<RankType>();
+                optionsBuilder.MapEnum<ControlType>();
+                optionsBuilder.MapEnum<PermissionType>();
+                optionsBuilder.MapEnum<ShockerModelType>();
+                optionsBuilder.MapEnum<OtaUpdateStatus>();
+            });
             if (config.Db.Debug)
             {
                 builder.EnableSensitiveDataLogging();