Artemis/src/Artemis.Storage/StorageMigrationService.cs

using System;
using System.Collections.Generic;
using System.Linq;
using Artemis.Storage.Migrations;
using LiteDB;
using Serilog;

namespace Artemis.Storage;

public class StorageMigrationService
{
    private readonly ILogger _logger;
    private readonly IList<IStorageMigration> _migrations;
    private readonly LiteRepository _repository;

    public StorageMigrationService(ILogger logger, LiteRepository repository, IList<IStorageMigration> migrations)
    {
        _logger = logger;
        _repository = repository;
        _migrations = migrations;

        ApplyPendingMigrations();
    }

    public void ApplyPendingMigrations()
    {
        foreach (IStorageMigration storageMigration in _migrations.OrderBy(m => m.UserVersion))
        {
            if (_repository.Database.UserVersion >= storageMigration.UserVersion)
                continue;

            _logger.Information("Applying storage migration {storageMigration} to update DB from v{oldVersion} to v{newVersion}",
                storageMigration.GetType().Name, _repository.Database.UserVersion, storageMigration.UserVersion);

            _repository.Database.BeginTrans();
            try
            {
                storageMigration.Apply(_repository);
            }
            catch (Exception)
            {
                _repository.Database.Rollback();
                throw;
            }

            _repository.Database.Commit();

            _repository.Database.UserVersion = storageMigration.UserVersion;
        }
    }
}