Module: Ridgepole::Ext::SchemaDumper

Included in:
ActiveRecord::SchemaDumper
Defined in:
lib/ridgepole/ext/schema_dumper.rb

Defined Under Namespace

Modules: ClassMethods

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.prepended(klass) ⇒ Object 



6
7
8
 
# File 'lib/ridgepole/ext/schema_dumper.rb', line 6

def self.prepended(klass)
  klass.extend ClassMethods
end

Instance Method Details

#foreign_keys(table, stream) ⇒ Object 



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
 
# File 'lib/ridgepole/ext/schema_dumper.rb', line 21

def foreign_keys(table, stream)
  return super unless ActiveRecord::SchemaDumper.__with_default_fk_name

  if (foreign_keys = @connection.foreign_keys(table)).any?
    add_foreign_key_statements = foreign_keys.map do |foreign_key|
      parts = [
        "add_foreign_key #{remove_prefix_and_suffix(foreign_key.from_table).inspect}",
        remove_prefix_and_suffix(foreign_key.to_table).inspect
      ]

      if foreign_key.column != @connection.foreign_key_column_for(foreign_key.to_table)
        parts << "column: #{foreign_key.column.inspect}"
      end

      if foreign_key.custom_primary_key?
        parts << "primary_key: #{foreign_key.primary_key.inspect}"
      end

      parts << "name: #{foreign_key.name.inspect}"

      parts << "on_update: #{foreign_key.on_update.inspect}" if foreign_key.on_update
      parts << "on_delete: #{foreign_key.on_delete.inspect}" if foreign_key.on_delete

      "  #{parts.join(', ')}"
    end

    stream.puts add_foreign_key_statements.sort.join("\n")
  end
end