Module: Tapsoob::Schema

Extended by:
Schema
Included in:
Schema
Defined in:
lib/tapsoob/schema.rb

Instance Method Summary collapse

Instance Method Details

#dump(database_url) ⇒ Object 



12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
 
# File 'lib/tapsoob/schema.rb', line 12

def dump(database_url)
  db = Sequel.connect(database_url)
  db.extension :schema_dumper
  template = ERB.new "Class.new(Sequel::Migration) do\n  def up\n  <% db.tables.each do |table| %>\n<%= db.dump_table_schema(table, indexes: false) %>\n  <% end %>\n  end\n\n  def down\n  <% db.tables.each do |table| %>\ndrop_table(\"<%= table %>\", if_exists: true)\n  <% end %>\n  end\nend\n"

  template.result(binding)
end

#dump_table(database_url, table) ⇒ Object 



34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
 
# File 'lib/tapsoob/schema.rb', line 34

def dump_table(database_url, table)
  table = table.to_sym
  Sequel.connect(database_url) do |db|
    db.extension :schema_dumper
    "Class.new(Sequel::Migration) do\n  def up\n\#{db.dump_table_schema(table, indexes: false)}\n  end\n\n  def down\ndrop_table(\"\#{table}\", if_exists: true)\n  end\nend\n"
  end
end

#indexes(database_url) ⇒ Object 



52
53
54
55
56
 
# File 'lib/tapsoob/schema.rb', line 52

def indexes(database_url)
  db = Sequel.connect(database_url)
  db.extension :schema_dumper
  db.dump_indexes_migration
end

#indexes_individual(database_url) ⇒ Object 



58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
 
# File 'lib/tapsoob/schema.rb', line 58

def indexes_individual(database_url)
  idxs = {}
  Sequel.connect(database_url) do |db|
    db.extension :schema_dumper

    tables = db.tables
    tables.each do |table|
      idxs[table] = db.send(:dump_table_indexes, table, :add_index, {}).split("\n")
    end
  end

  idxs.each do |table, indexes|
    idxs[table] = indexes.map do |idx|
      "Class.new(Sequel::Migration) do\n  def up\n\#{idx}\n  end\nend\n"
    end
  end
  JSON.generate(idxs)
end

#load(database_url, schema, options = { drop: false }) ⇒ Object 



83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
 
# File 'lib/tapsoob/schema.rb', line 83

def load(database_url, schema, options = { drop: false })
  Sequel.connect(database_url) do |db|
    db.extension :schema_dumper
    klass = eval(schema)
    if options[:drop]
      # Start special hack for MySQL
      db.run("SET foreign_key_checks = 0") if [:mysql, :mysql2].include?(db.adapter_scheme)

      # Run down migration
      klass.apply(db, :down)

      # End special hack for MySQL
      db.run("SET foreign_key_checks = 1") if [:mysql, :mysql2].include?(db.adapter_scheme)
    end
    klass.apply(db, :up)
  end
end

#load_indexes(database_url, indexes) ⇒ Object 



101
102
103
104
105
106
 
# File 'lib/tapsoob/schema.rb', line 101

def load_indexes(database_url, indexes)
  Sequel.connect(database_url) do |db|
    db.extension :schema_dumper
    eval(indexes).apply(db, :up)
  end
end

#reset_db_sequences(database_url) ⇒ Object 



108
109
110
111
112
113
114
115
 
# File 'lib/tapsoob/schema.rb', line 108

def reset_db_sequences(database_url)
  db = Sequel.connect(database_url)
  db.extension :schema_dumper
  return unless db.respond_to?(:reset_primary_key_sequence)
  db.tables.each do |table|
    db.reset_primary_key_sequence(table)
  end
end