class DataMigrate::DataMigrator
Public Class Methods
assure_data_schema_table()
click to toggle source
# File lib/data_migrate/data_migrator.rb, line 43 def assure_data_schema_table config = ActiveRecord::Base.configurations[Rails.env || 'development'] || ENV["DATABASE_URL"] ActiveRecord::Base.establish_connection(config) sm_table = DataMigrate::DataMigrator.schema_migrations_table_name unless table_exists?(ActiveRecord::Base.connection, sm_table) ActiveRecord::Base.connection.create_table(sm_table, :id => false) do |schema_migrations_table| schema_migrations_table.column :version, :string, :null => false end suffix = ActiveRecord::Base.table_name_suffix prefix = ActiveRecord::Base.table_name_prefix index_name = "#{prefix}unique_data_migrations#{suffix}" ActiveRecord::Base.connection.add_index sm_table, :version, :unique => true, :name => index_name end end
get_all_versions(connection = ActiveRecord::Base.connection)
click to toggle source
# File lib/data_migrate/data_migrator.rb, line 18 def get_all_versions(connection = ActiveRecord::Base.connection) if table_exists?(connection, schema_migrations_table_name) # Certain versions of the gem wrote data migration versions into # schema_migrations table. After the fix, it was corrected to write into # data_migrations. However, not to break anything we are going to # get versions from both tables. # # This may cause some problems: # Eg. rake data:versions will show version from the schema_migrations table # which may be a version of actual schema migration and not data migration DataMigrate::DataSchemaMigration.all.map { |x| x.version.to_i }.sort + ActiveRecord::SchemaMigration.all.map { |x| x.version.to_i }.sort else [] end end
migrations_path()
click to toggle source
# File lib/data_migrate/data_migrator.rb, line 39 def migrations_path 'db/data' end
schema_migrations_table_name()
click to toggle source
# File lib/data_migrate/data_migrator.rb, line 35 def schema_migrations_table_name ActiveRecord::Base.table_name_prefix + 'data_migrations' + ActiveRecord::Base.table_name_suffix end
Private Class Methods
table_exists?(connection, table_name)
click to toggle source
# File lib/data_migrate/data_migrator.rb, line 65 def table_exists?(connection, table_name) # Avoid the warning that table_exists? prints in Rails 5.0 due a change in behavior between # Rails 5.0 and Rails 5.1 of this method with respect to database views. if ActiveRecord.version >= Gem::Version.new('5.0') && ActiveRecord.version < Gem::Version.new('5.1') connection.data_source_exists?(table_name) else connection.table_exists?(schema_migrations_table_name) end end
Public Instance Methods
record_version_state_after_migrating(version)
click to toggle source
# File lib/data_migrate/data_migrator.rb, line 7 def record_version_state_after_migrating(version) if down? migrated.delete(version) DataMigrate::DataSchemaMigration.where(:version => version.to_s).delete_all else migrated << version DataMigrate::DataSchemaMigration.create!(:version => version.to_s) end end