1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
# frozen_string_literal: true
require "data_migrate/config"
module DataMigrate
##
# This class extends DatabaseTasks to add a schema_file method.
class DatabaseTasks
extend ActiveRecord::Tasks::DatabaseTasks
class << self
def schema_file_type(_format = nil)
"data_schema.rb"
end
def dump_filename(namespace, format = ActiveRecord::Base.schema_format)
filename = if namespace == "primary"
schema_file_type(format)
else
"#{namespace}_#{schema_file_type(format)}"
end
ENV["DATA_SCHEMA"] || File.join(schema_location, filename)
end
def schema_location
db_dir
end
def check_schema_file(filename)
unless File.exist?(filename)
message = +%{#{filename} doesn't exist yet. Run `rake data:migrate` to create it, then try again.}
Kernel.abort message
end
end
end
# This overrides ActiveRecord::Tasks::DatabaseTasks
def self.schema_file(_format = nil)
File.join(db_dir, "data_schema.rb")
end
def self.forward(step = 1)
DataMigrate::DataMigrator.assure_data_schema_table
migrations = pending_migrations.reverse.pop(step).reverse
migrations.each do | pending_migration |
if pending_migration[:kind] == :data
ActiveRecord::Migration.write("== %s %s" % ["Data", "=" * 71])
DataMigrate::DataMigrator.run(:up, data_migrations_path, pending_migration[:version])
elsif pending_migration[:kind] == :schema
ActiveRecord::Migration.write("== %s %s" % ["Schema", "=" * 69])
DataMigrate::SchemaMigration.run(:up, schema_migrations_path, pending_migration[:version])
end
end
end
def self.data_migrations_path
DataMigrate.config.data_migrations_path
end
def self.schema_migrations_path
"db/migrate/"
end
def self.pending_migrations
sort_migrations(pending_schema_migrations, pending_data_migrations)
end
def self.pending_data_migrations
data_migrations = DataMigrate::DataMigrator.migrations(data_migrations_path)
sort_migrations(DataMigrate::DataMigrator.new(:up, data_migrations ).
pending_migrations.map {|m| { version: m.version, name: m.name, kind: :data }})
end
def self.pending_schema_migrations
::DataMigrate::SchemaMigration.pending_schema_migrations
end
def self.sort_migrations(set1, set2 = nil)
migrations = set1 + (set2 || [])
migrations.sort {|a, b| sort_string(a) <=> sort_string(b)}
end
def self.sort_string(migration)
"#{migration[:version]}_#{migration[:kind] == :data ? 1 : 0}"
end
def self.past_migrations(sort = nil)
sort = sort.downcase if sort
db_list_data =
if DataMigrate::DataSchemaMigration.table_exists?
DataMigrate::DataSchemaMigration.normalized_versions.sort
else
[]
end
db_list_schema = ActiveRecord::SchemaMigration.normalized_versions.sort.sort
migrations = db_list_data.map do |d|
{
version: d.to_i, kind: :data
}
end +
db_list_schema.map do |d|
{
version: d.to_i, kind: :schema
}
end
sort == "asc" ? sort_migrations(migrations) : sort_migrations(migrations).reverse
end
end
end
|
