feat: add PostgresPeriodicTuner for periodic database maintenance and optimize NoWarn settings

2026-07-07 08:29:35 -04:00
parent 4a408dc625
commit b475a8264a
7 changed files with 150 additions and 58 deletions
@@ -0,0 +1,138 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Threading;
+using System.Threading.Tasks;
+using Jellyfin.Data.Queries;
+using Jellyfin.Extensions.Dapper;
+using MediaBrowser.Controller.Configuration;
+using MediaBrowser.Controller.Plugins;
+using MediaBrowser.Controller.Session;
+using MediaBrowser.Model.Tasks;
+using Microsoft.Extensions.Logging;
+
+namespace Jellyfin.Server.Implementations.ScheduledTasks
+{
+    /// <summary>
+    /// Class PostgresPeriodicTuner.
+    /// </summary>
+    public class PostgresPeriodicTuner : IScheduledTask, IConfigurableScheduledTask
+    {
+        private readonly ILogger<PostgresPeriodicTuner> _logger;
+        private readonly IDbContext _dbContext;
+
+        /// <summary>
+        /// Initializes a new instance of the <see cref="PostgresPeriodicTuner"/> class.
+        /// </summary>
+        /// <param name="logger">The logger.</param>
+        /// <param name="dbContext">The database context.</param>
+        public PostgresPeriodicTuner(ILogger<PostgresPeriodicTuner> logger, IDbContext dbContext)
+        {
+            _logger = logger;
+            _dbContext = dbContext;
+        }
+
+        /// <inheritdoc />
+        public string Name => "PostgreSQL Periodic Tuner";
+
+        /// <inheritdoc />
+        public string Key => "PostgresPeriodicTuner";
+
+        /// <inheritdoc />
+        public string Description => "Periodically checks for dead tuples and runs VACUUM ANALYZE on tables that need it.";
+
+        /// <inheritdoc />
+        public string Category => "Database";
+
+        /// <inheritdoc />
+        public bool IsHidden => false;
+
+        /// <inheritdoc />
+        public bool IsEnabled => true;
+
+        /// <inheritdoc />
+        public bool IsLogged => true;
+
+        /// <summary>
+        /// Executes the task.
+        /// </summary>
+        /// <param name="cancellationToken">The cancellation token.</param>
+        /// <param name="progress">The progress.</param>
+        /// <returns>A <see cref="Task"/> representing the asynchronous operation.</returns>
+        public async Task ExecuteAsync(CancellationToken cancellationToken, IProgress<double> progress)
+        {
+            _logger.LogInformation("PostgreSQL Periodic Tuner task started.");
+
+            try
+            {
+                const string DeadTupleQuery = @"
+                    SELECT
+                        relname AS TableName,
+                        n_dead_tup AS DeadTuples
+                    FROM
+                        pg_stat_user_tables
+                    WHERE
+                        n_dead_tup > 1000 -- Example threshold: more than 1000 dead tuples
+                    ORDER BY
+                        n_dead_tup DESC;";
+
+                var tablesToVacuum = (await _dbContext.QueryAsync<(string TableName, long DeadTuples)>(DeadTupleQuery)).ToList();
+
+                if (tablesToVacuum.Count == 0)
+                {
+                    _logger.LogInformation("No tables require vacuuming at this time.");
+                    return;
+                }
+
+                progress.Report(10);
+
+                double progressStep = 90.0 / tablesToVacuum.Count;
+                double currentProgress = 10.0;
+
+                foreach (var table in tablesToVacuum)
+                {
+                    cancellationToken.ThrowIfCancellationRequested();
+                    _logger.LogInformation("Found {DeadTuples} dead tuples in table {TableName}. Running VACUUM ANALYZE.", table.DeadTuples, table.TableName);
+
+                    try
+                    {
+                        await _dbContext.ExecuteAsync($"VACUUM ANALYZE \"{table.TableName}\"");
+                        _logger.LogInformation("Successfully vacuumed and analyzed table {TableName}.", table.TableName);
+                    }
+                    catch (Exception ex)
+                    {
+                        _logger.LogError(ex, "Error running VACUUM ANALYZE on table {TableName}.", table.TableName);
+                    }
+
+                    currentProgress += progressStep;
+                    progress.Report(currentProgress);
+                }
+            }
+            catch (Exception ex)
+            {
+                _logger.LogError(ex, "Error during PostgreSQL Periodic Tuner task.");
+            }
+            finally
+            {
+                _logger.LogInformation("PostgreSQL Periodic Tuner task finished.");
+                progress.Report(100);
+            }
+        }
+
+        /// <summary>
+        /// Gets the default triggers.
+        /// </summary>
+        /// <returns>IEnumerable&lt;TaskTriggerInfo&gt;.</returns>
+        public IEnumerable<TaskTriggerInfo> GetDefaultTriggers()
+        {
+            return new[]
+            {
+                new TaskTriggerInfo
+                {
+                    Type = TaskTriggerInfo.TriggerInterval,
+                    IntervalTicks = TimeSpan.FromHours(24).Ticks
+                }
+            };
+        }
+    }
+}