Sha256: e8d49bca4df9d4887e5cf0481dee3ae422e5cc65238aca18ca829a41de059282
Contents?: true
Size: 1.9 KB
Versions: 9
Compression:
Stored size: 1.9 KB
Contents
# frozen_string_literal: true
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 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
]
parts << "column: #{foreign_key.column.inspect}" if foreign_key.column != foreign_key_column_for(foreign_key)
parts << "primary_key: #{foreign_key.primary_key.inspect}" if foreign_key.custom_primary_key?
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
private
def foreign_key_column_for(foreign_key)
if ActiveRecord.gem_version < Gem::Version.new('7.1.0')
@connection.foreign_key_column_for(foreign_key.to_table)
else
@connection.foreign_key_column_for(foreign_key.to_table, 'id')
end
end
end
end
end
module ActiveRecord
class SchemaDumper
prepend Ridgepole::Ext::SchemaDumper
end
end
Version data entries
9 entries across 9 versions & 1 rubygems