Sha256: 6e327f17e74e5c43ca25bc25b0f28f8108a1fa4af444e183c94385f11ab4f72b
Contents?: true
Size: 1.64 KB
Versions: 12
Compression:
Stored size: 1.64 KB
Contents
require 'active_record/schema_dumper'
module Ridgepole
module Ext
module SchemaDumper
def self.prepended(klass)
klass.extend ClassMethods
end
module ClassMethods
attr_reader :__with_default_fk_name
def with_default_fk_name(value)
@__with_default_fk_name = value
yield
ensure
remove_instance_variable(:@__with_default_fk_name)
end
end
def foreign_keys(table, stream)
return super unless self.class.__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
end
end
end
module ActiveRecord
class SchemaDumper
prepend Ridgepole::Ext::SchemaDumper
end
end
Version data entries
12 entries across 12 versions & 1 rubygems