Ejemplo n.º 1
0
        public async Task Migrate_AllScriptOk_NoRollback()
        {
            var config           = ConfigProvider.GetConfig();
            var connectionString = String.Format(config.ConnectionStringMask, "test_ok");


            var builder = new MigratorBuilder();

            builder.UseCodeMigrations().FromAssembly(Assembly.GetExecutingAssembly());
            builder.UseScriptMigrations().FromDirectory(Path.Combine(Directory.GetCurrentDirectory(), "ScriptMigrations"));
            builder.UsePostgreSQL(connectionString);

            builder.UseUpgradeMigrationPolicy(MigrationPolicy.All);
            builder.UseDowngradeMigrationPolicy(MigrationPolicy.All);
            builder.SetUpTargetVersion(new DbVersion(3, 0));

            var migrator = builder.Build();

            await migrator.MigrateAsync();

            var migrationProvider = new PostgreDbProvider(new PostgreDbProviderOptions(connectionString));
            await migrationProvider.OpenConnectionAsync();

            var currentDbVersion = await migrationProvider.GetDbVersionSafeAsync();

            await migrationProvider.CloseConnectionAsync();

            Assert.Equal(new DbVersion(3, 0), currentDbVersion);
        }
        public async Task Migrate_AllScriptOk_Rollback()
        {
            var config           = ConfigProvider.GetConfig();
            var connectionString = String.Format(config.ConnectionStringMask, "test_rollback");


            var builder = new MigratorBuilder();

            builder.UseCodeMigrations().FromAssembly(Assembly.GetExecutingAssembly());
            builder.UseScriptMigrations().FromDirectory(Path.Combine(Directory.GetCurrentDirectory(), "ScriptMigrations"));
            builder.UsePostgreSQL(connectionString);

            builder.UseUpgradeMigrationPolicy(MigrationPolicy.Allowed);
            builder.UseDowngradeMigrationPolicy(MigrationPolicy.Allowed);
            builder.SetUpTargetVersion(new DbVersion(6, 0));

            var migrator = builder.Build();

            try
            {
                await migrator.MigrateAsync();

                // last migration is incorrect, can not go here
                Assert.False(true);
            }
            catch
            {
                // ignored
            }

            var migrationProvider = new PostgreDbProvider(new PostgreDbProviderOptions(connectionString));
            await migrationProvider.OpenConnectionAsync();

            var actualAppliedMigrations = await migrationProvider.GetAppliedMigrationVersionAsync();

            await migrationProvider.CloseConnectionAsync();

            var expectedAppliedMigrations = new HashSet <DbVersion>
            {
                new DbVersion(1, 0),
                new DbVersion(2, 0),
                new DbVersion(3, 0),
                new DbVersion(4, 0),
                new DbVersion(5, 0)
            };

            Assert.Equal(expectedAppliedMigrations, actualAppliedMigrations);
        }
        public async Task Migrate_AllScriptOk_SwitchedOffTransaction()
        {
            var config           = ConfigProvider.GetConfig();
            var connectionString = String.Format(config.ConnectionStringMask, "test_without_transactions");


            var builder = new MigratorBuilder();

            builder.UseCodeMigrations().FromAssembly(Assembly.GetExecutingAssembly());
            builder.UseScriptMigrations().FromDirectory(Path.Combine(Directory.GetCurrentDirectory(), "ScriptMigrations"));
            builder.UsePostgreSQL(connectionString);

            builder.UseUpgradeMigrationPolicy(MigrationPolicy.Allowed);
            builder.UseDowngradeMigrationPolicy(MigrationPolicy.Allowed);
            builder.SetUpTargetVersion(new DbVersion(5, 0));

            var migrator = builder.Build();

            await migrator.MigrateAsync();

            var migrationProvider = new PostgreDbProvider(new PostgreDbProviderOptions(connectionString));
            await migrationProvider.OpenConnectionAsync();

            var actualAppliedMigrations = await migrationProvider.GetAppliedMigrationVersionAsync();

            await migrationProvider.CloseConnectionAsync();

            var expectedAppliedMigrations = new HashSet <DbVersion>
            {
                new DbVersion(1, 0),
                new DbVersion(2, 0),
                new DbVersion(3, 0),
                new DbVersion(4, 0),
                new DbVersion(5, 0)
            };

            Assert.Equal(expectedAppliedMigrations, actualAppliedMigrations);
        }
Ejemplo n.º 4
0
        public async Task StartAsync(CancellationToken cToken)
        {
            _logger.LogInformation("Мигратор запущен");

            var builder = new MigratorBuilder(_services);

            builder.UsePostgreSQL(
                _configuration.MigrationOptions.ConnectionString,
                lcCollate: _configuration.MigrationOptions.LC_Collate,
                lcCtype: _configuration.MigrationOptions.LC_Type,
                databaseEncoding: _configuration.MigrationOptions.DataBaseEncoding,
                migrationTableHistoryName: "migration_history");
            builder.UseScriptMigrations()
            .FromAssembly(typeof(MigrationService).Assembly, "Migrations.Scripts.");
            builder.UseCodeMigrations()
            .FromAssembly <IMigration>(typeof(MigrationService).Assembly);
            builder.UseUpgradeMigrationPolicy(MigrationPolicy.Allowed);
            builder.UserLogger(_logger);

            if (_configuration.MigrationOptions.LogSql)
            {
                builder.UseLoggerForSql(_logger);
            }

            if (!String.IsNullOrWhiteSpace(_configuration.MigrationOptions.GrantUser))
            {
                builder.UseVariable(DefaultVariables.User, _configuration.MigrationOptions.GrantUser);
            }

            var migrator = builder.Build();
            var result   = await migrator.MigrateSafeAsync(cToken);

            _logger.LogInformation(result.IsSuccessfully
                ? "Миграции успешно выполнены"
                : $"Ошибка миграции: {result.ErrorMessage}");
        }