feat: Adding postgres support to ombi (beta) (#5050)

Ombi-app · Jul 11, 2024 · f8c6102 · f8c6102
1 parent 8ff7d99
commit f8c6102
Show file tree

Hide file tree

Showing 22 changed files with 5,660 additions and 12 deletions.
diff --git a/src/Ombi.Core.Tests/Engine/MovieRequestLimitsTests.cs b/src/Ombi.Core.Tests/Engine/MovieRequestLimitsTests.cs
@@ -300,13 +300,13 @@ public async Task UserPassedIn_MovieLimit_Set_Limit_Daily_AllRequestsToday()
                 {
                     UserId = "id1",
                     RequestType = RequestType.Movie,
-                    RequestDate = today.AddHours(-1),
+                    RequestDate = today.AddMinutes(-1),
                 },
                 new RequestLog
                 {
                     UserId = "id1",
                     RequestType = RequestType.Movie,
-                    RequestDate = today.AddHours(-2),
+                    RequestDate = today.AddMinutes(-2),
                 },
             };
             var repoMock = _mocker.GetMock<IRepository<RequestLog>>();

diff --git a/src/Ombi.Core.Tests/Engine/MusicRequestLimitTests.cs b/src/Ombi.Core.Tests/Engine/MusicRequestLimitTests.cs
@@ -298,13 +298,13 @@ public async Task UserPassedIn_MusicLimit_Set_Limit_Daily_AllRequestsToday()
                 {
                     UserId = "id1",
                     RequestType = RequestType.Album,
-                    RequestDate = today.AddHours(-1),
+                    RequestDate = today.AddMinutes(-1),
                 },
                 new RequestLog
                 {
                     UserId = "id1",
                     RequestType = RequestType.Album,
-                    RequestDate = today.AddHours(-2),
+                    RequestDate = today.AddMinutes(-2),
                 },
             };
             var repoMock = _mocker.GetMock<IRepository<RequestLog>>();

diff --git a/src/Ombi.Core.Tests/Engine/TvRequestLimitsTests.cs b/src/Ombi.Core.Tests/Engine/TvRequestLimitsTests.cs
@@ -304,15 +304,15 @@ public async Task UserPassedIn_TvLimit_Set_Limit_Daily_AllRequestsToday()
                 {
                     UserId = "id1",
                     RequestType = RequestType.TvShow,
-                    RequestDate = today.AddHours(-1),
+                    RequestDate = today.AddMinutes(-1),
                     EpisodeCount = 1,
                 },
                 new RequestLog
                 {
                     UserId = "id1",
                     RequestType = RequestType.TvShow,
                     EpisodeCount = 1,
-                    RequestDate = today.AddHours(-2),
+                    RequestDate = today.AddMinutes(-2),
                 },
             };
             var repoMock = _mocker.GetMock<IRepository<RequestLog>>();
@@ -345,15 +345,15 @@ public async Task UserPassedIn_TvLimit_Set_Limit_Daily_MultipleEpisodeRequests()
                 {
                     UserId = "id1",
                     RequestType = RequestType.TvShow,
-                    RequestDate = today.AddHours(-1),
+                    RequestDate = today.AddMinutes(-1),
                     EpisodeCount = 5,
                 },
                 new RequestLog
                 {
                     UserId = "id1",
                     RequestType = RequestType.TvShow,
                     EpisodeCount = 4,
-                    RequestDate = today.AddHours(-2),
+                    RequestDate = today.AddMinutes(-2),
                 },
             };
             var repoMock = _mocker.GetMock<IRepository<RequestLog>>();

diff --git a/src/Ombi.Schedule.Tests/IssuesPurgeTests.cs b/src/Ombi.Schedule.Tests/IssuesPurgeTests.cs
@@ -91,12 +91,12 @@ public async Task DoesNot_Delete_AnyIssues()
                 new Issues
                 {
                     Status = IssueStatus.Resolved,
-                    ResovledDate = DateTime.Now.AddDays(-2)
+                    ResovledDate = DateTime.UtcNow.AddDays(-2)
                 },
                 new Issues
                 {
                     Status = IssueStatus.Resolved,
-                    ResovledDate = DateTime.Now.AddDays(-4)
+                    ResovledDate = DateTime.UtcNow.AddDays(-4)
                 }
             };
 

diff --git a/src/Ombi.Store/Context/Postgres/ExternalPostgresContext.cs b/src/Ombi.Store/Context/Postgres/ExternalPostgresContext.cs
@@ -0,0 +1,16 @@
+using Microsoft.EntityFrameworkCore;
+
+namespace Ombi.Store.Context.Postgres
+{
+    public sealed class ExternalPostgresContext : ExternalContext
+    {
+        private static bool _created;
+        public ExternalPostgresContext(DbContextOptions<ExternalPostgresContext> options) : base(options)
+        {
+            if (_created) return;
+
+            _created = true;
+            Database.Migrate();
+        }
+    }
+}
diff --git a/src/Ombi.Store/Context/Postgres/OmbiPostgresContext.cs b/src/Ombi.Store/Context/Postgres/OmbiPostgresContext.cs
@@ -0,0 +1,22 @@
+using Microsoft.EntityFrameworkCore;
+
+namespace Ombi.Store.Context.Postgres
+{
+    public sealed class OmbiPostgresContext : OmbiContext
+    {
+        private static bool _created;
+
+        public OmbiPostgresContext(DbContextOptions<OmbiPostgresContext> options) : base(options)
+        {
+            if (_created) return;
+            _created = true;
+
+            Database.Migrate();
+        }
+
+        public override void Dispose()
+        {
+            base.Dispose();
+        }
+    }
+}
diff --git a/src/Ombi.Store/Context/Postgres/PostgresModuleInitializer.cs b/src/Ombi.Store/Context/Postgres/PostgresModuleInitializer.cs
@@ -0,0 +1,17 @@
+using System;
+using System.Runtime.CompilerServices;
+
+namespace Ombi.Store.Context.Postgres;
+
+public static class PostgresModuleInitializer
+{
+#pragma warning disable CA2255
+    // This is required to ensure that Npgsql uses a timestamp behavior that does not require a timezone
+    // Reference: https://stackoverflow.com/a/73586129
+    [ModuleInitializer]
+#pragma warning restore CA2255
+    public static void Initialize()
+    {
+        AppContext.SetSwitch("Npgsql.EnableLegacyTimestampBehavior", true);
+    }
+}
diff --git a/src/Ombi.Store/Context/Postgres/SettingsPostgresContext.cs b/src/Ombi.Store/Context/Postgres/SettingsPostgresContext.cs
@@ -0,0 +1,16 @@
+using Microsoft.EntityFrameworkCore;
+
+namespace Ombi.Store.Context.Postgres
+{
+    public sealed class SettingsPostgresContext : SettingsContext
+    {
+        private static bool _created;
+        public SettingsPostgresContext(DbContextOptions<SettingsPostgresContext> options) : base(options)
+        {
+            if (_created) return;
+
+            _created = true;
+            Database.Migrate();
+        }
+    }
+}
diff --git a/src/Ombi.Store/MigrationHelper.cs b/src/Ombi.Store/MigrationHelper.cs
@@ -20,5 +20,13 @@ INSERT INTO AspNetRoles(Id, ConcurrencyStamp, Name, NormalizedName)
 SELECT '{Guid.NewGuid()}','{Guid.NewGuid()}','{role}', '{role.ToUpper()}' 
 WHERE NOT EXISTS(SELECT 1 FROM AspNetRoles WHERE Name = '{role}');");
         }
+
+        public static void InsertRolePostgres(this MigrationBuilder mb, string role)
+        {
+            mb.Sql($@"
+INSERT INTO public.""AspNetRoles""(""Id"", ""ConcurrencyStamp"", ""Name"", ""NormalizedName"")
+SELECT '{Guid.NewGuid()}','{Guid.NewGuid()}','{role}', '{role.ToUpper()}'
+WHERE NOT EXISTS(SELECT 1 FROM public.""AspNetRoles"" WHERE ""Name"" = '{role}');");
+        }
     }
 }