Jellyfin.Database/Jellyfin.Database.Providers.PgSql/PgSqlDatabaseProvider.cs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75

using System;
using Jellyfin.Server.Implementations;
using Jellyfin.Server.Implementations.DatabaseConfiguration;
using MediaBrowser.Common.Configuration;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Logging;
using Npgsql;

namespace Jellyfin.Database.Providers.PgSql;

/// <summary>
/// Configures jellyfin to use an SqLite database.
/// </summary>
[JellyfinDatabaseProviderKey("Jellyfin-PgSql")]
public sealed class PgSqlDatabaseProvider : IJellyfinDatabaseProvider
{
    private readonly IConfigurationManager _configurationManager;
    private readonly ILogger<PgSqlDatabaseProvider> _logger;

    /// <summary>
    /// Initializes a new instance of the <see cref="PgSqlDatabaseProvider"/> class.
    /// </summary>
    /// <param name="configurationManager">Configuration manager to get PgSQL connection data.</param>
    /// <param name="logger">A logger.</param>
    public PgSqlDatabaseProvider(IConfigurationManager configurationManager, ILogger<PgSqlDatabaseProvider> logger)
    {
        _configurationManager = configurationManager;
        _logger = logger;
    }

    /// <inheritdoc/>
    public IDbContextFactory<JellyfinDbContext>? DbContextFactory { get; set; }

    /// <inheritdoc/>
    public void Initialise(DbContextOptionsBuilder options)
    {
        var dbSettings = _configurationManager.GetConfiguration<DatabaseConfigurationOptions>("database");

        if (dbSettings.PostgreSql is null)
        {
            throw new InvalidOperationException("Selected PgSQL as database provider but did not provide required configuration. Please see docs.");
        }

        var connectionBuilder = new NpgsqlConnectionStringBuilder();
        connectionBuilder.ApplicationName = "jellyfin";
        connectionBuilder.CommandTimeout = dbSettings.PostgreSql.Timeout;
        connectionBuilder.Database = dbSettings.PostgreSql.DatabaseName;
        connectionBuilder.Username = dbSettings.PostgreSql.Username;
        connectionBuilder.Password = dbSettings.PostgreSql.Password;
        connectionBuilder.Host = dbSettings.PostgreSql.ServerName;
        connectionBuilder.Port = dbSettings.PostgreSql.Port;

        var connectionString = connectionBuilder.ToString();

        options
            .UseNpgsql(connectionString, pgSqlOptions => pgSqlOptions.MigrationsAssembly(GetType().Assembly));
    }

    /// <inheritdoc/>
    public Task RunScheduledOptimisation(CancellationToken cancellationToken)
    {
        return Task.CompletedTask;
    }

    /// <inheritdoc/>
    public void OnModelCreating(ModelBuilder modelBuilder)
    {
    }

    /// <inheritdoc/>
    public ValueTask DisposeAsync()
    {
        return ValueTask.CompletedTask;
    }
}