Ejemplo n.º 1
0
        IEnumerable <string> RunAutoMigrations(int schemaVersion)
        {
            var schema = schemaVersion == -1 // fresh database
                ? new Dictionary <string, List <string> >()
                : store.Database.QuerySchema();

            var commands = differ.CalculateSchemaChanges(schema, store.Configuration);

            if (!commands.Any())
            {
                return(Enumerable.Empty <string>());
            }

            logger.LogInformation("Found {0} differences between current schema and configuration. Migrates schema to get up to date.", commands.Count);

            return(commands.SelectMany(command => ExecuteCommand(command, false), (_, tablename) => tablename));
        }
Ejemplo n.º 2
0
        public void Run()
        {
            if (!store.Configuration.RunSchemaMigrationsOnStartup)
            {
                return;
            }

            var requiresReprojection = new List <string>();

            var database      = ((DocumentStore)store).Database;
            var configuration = store.Configuration;

            using (var tx = new TransactionScope(TransactionScopeOption.RequiresNew, new TransactionOptions {
                IsolationLevel = IsolationLevel.Serializable
            }))
            {
                // lock other migrators out while executing
                using (var connection = database.Connect())
                {
                    var parameters = new DynamicParameters();
                    parameters.Add("@Resource", "HybridDb");
                    parameters.Add("@DbPrincipal", "public");
                    parameters.Add("@LockMode", "Exclusive");
                    parameters.Add("@LockOwner", "Transaction");
                    parameters.Add("@LockTimeout", TimeSpan.FromMinutes(1).TotalMilliseconds);
                    parameters.Add("@Result", dbType: DbType.Int32, direction: ParameterDirection.ReturnValue);

                    connection.Connection.Execute(@"sp_getapplock", parameters, commandType: CommandType.StoredProcedure);

                    var result = parameters.Get <int>("@Result");

                    if (result < 0)
                    {
                        throw new InvalidOperationException($"sp_getapplock failed with code {result}");
                    }

                    connection.Complete();
                }

                // create metadata table if it does not exist
                var metadata = new Table("HybridDb", new Column("SchemaVersion", typeof(int)));
                configuration.Tables.TryAdd(metadata.Name, metadata);
                new CreateTable(metadata).Execute(database);

                // get schema version
                var currentSchemaVersion = database.RawQuery <int>(
                    $"select top 1 SchemaVersion from {database.FormatTableNameAndEscape("HybridDb")}")
                                           .SingleOrDefault();

                if (currentSchemaVersion > store.Configuration.ConfiguredVersion)
                {
                    throw new InvalidOperationException(
                              $"Database schema is ahead of configuration. Schema is version {currentSchemaVersion}, " +
                              $"but configuration is version {store.Configuration.ConfiguredVersion}.");
                }

                // run provided migrations only if we are using real tables
                if (database is SqlServerUsingRealTables)
                {
                    if (currentSchemaVersion < configuration.ConfiguredVersion)
                    {
                        var migrationsToRun = migrations.OrderBy(x => x.Version).Where(x => x.Version > currentSchemaVersion).ToList();
                        logger.Information("Migrates schema from version {0} to {1}.", currentSchemaVersion, configuration.ConfiguredVersion);

                        foreach (var migration in migrationsToRun)
                        {
                            var migrationCommands = migration.MigrateSchema();
                            foreach (var command in migrationCommands)
                            {
                                requiresReprojection.AddRange(ExecuteCommand(database, command));
                            }

                            currentSchemaVersion++;
                        }
                    }
                }
                else
                {
                    logger.Information("Skips provided migrations when not using real tables.");
                }

                // get the diff and run commands to get to configured schema
                var schema   = database.QuerySchema().Values.ToList(); // demeter go home!
                var commands = differ.CalculateSchemaChanges(schema, configuration);

                if (commands.Any())
                {
                    logger.Information("Found {0} differences between current schema and configuration. Migrates schema to get up to date.", commands.Count);

                    foreach (var command in commands)
                    {
                        requiresReprojection.AddRange(ExecuteCommand(database, command));
                    }
                }

                // flag each document of tables that need to run a re-projection
                foreach (var tablename in requiresReprojection)
                {
                    // TODO: Only set RequireReprojection on command if it is documenttable - can it be done?
                    var design = configuration.DocumentDesigns.FirstOrDefault(x => x.Table.Name == tablename);
                    if (design == null)
                    {
                        continue;
                    }

                    database.RawExecute(
                        $"update {database.FormatTableNameAndEscape(tablename)} set AwaitsReprojection=@AwaitsReprojection",
                        new { AwaitsReprojection = true });
                }

                // update the schema version
                database.RawExecute(string.Format(@"
if not exists (select * from {0}) 
    insert into {0} (SchemaVersion) values (@version); 
else
    update {0} set SchemaVersion=@version",
                                                  database.FormatTableNameAndEscape("HybridDb")),
                                    new { version = currentSchemaVersion });

                tx.Complete();
            }
        }