41 lines
1 KiB
Ruby
41 lines
1 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab
|
|
module Database
|
|
module SchemaMigrations
|
|
class Context
|
|
attr_reader :connection
|
|
|
|
def initialize(connection)
|
|
@connection = connection
|
|
end
|
|
|
|
def schema_directory
|
|
@schema_directory ||=
|
|
if ActiveRecord::Base.configurations.primary?(database_name)
|
|
File.join(db_dir, 'schema_migrations')
|
|
else
|
|
File.join(db_dir, "#{database_name}_schema_migrations")
|
|
end
|
|
end
|
|
|
|
def versions_to_create
|
|
versions_from_database = @connection.schema_migration.all_versions
|
|
versions_from_migration_files = @connection.migration_context.migrations.map { |m| m.version.to_s }
|
|
|
|
versions_from_database & versions_from_migration_files
|
|
end
|
|
|
|
private
|
|
|
|
def database_name
|
|
@database_name ||= @connection.pool.db_config.name
|
|
end
|
|
|
|
def db_dir
|
|
@db_dir ||= Rails.application.config.paths["db"].first
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|