PostgresSqlDatabaseContext.cs

Go to the documentation of this file.

using System;
 
using Microsoft.EntityFrameworkCore;
 
using Tgstation.Server.Host.Configuration;
 
namespace Tgstation.Server.Host.Database
{
    sealed class PostgresSqlDatabaseContext : DatabaseContext
    {
        protected override DeleteBehavior RevInfoCompileJobDeleteBehavior => DeleteBehavior.Cascade;
 
        public PostgresSqlDatabaseContext(
            DbContextOptions<PostgresSqlDatabaseContext> dbContextOptions)
            : base(dbContextOptions)
        {
        }
 
        public static void ConfigureWith(DbContextOptionsBuilder options, DatabaseConfiguration databaseConfiguration)
        {
            ArgumentNullException.ThrowIfNull(options);
            ArgumentNullException.ThrowIfNull(databaseConfiguration);
 
            if (databaseConfiguration.DatabaseType != DatabaseType.PostgresSql)
                throw new InvalidOperationException($"Invalid DatabaseType for {nameof(PostgresSqlDatabaseContext)}!");
 
            // Why the fuck is this an AppContext switch
            AppContext.SetSwitch("Npgsql.EnableLegacyTimestampBehavior", true);
            options.UseNpgsql(databaseConfiguration.ConnectionString, options =>
            {
                options.EnableRetryOnFailure();
                options.UseQuerySplittingBehavior(QuerySplittingBehavior.SingleQuery);
 
                if (!String.IsNullOrEmpty(databaseConfiguration.ServerVersion))
                    options.SetPostgresVersion(
                        Version.Parse(databaseConfiguration.ServerVersion));
            });
        }
    }
}