Sha256: ddbd9588477819826931e0bb537fc5b449299c5f49ae9d6efc0719c6c5bd3637
Contents?: true
Size: 1.33 KB
Versions: 7
Compression:
Stored size: 1.33 KB
Contents
# frozen_string_literal: true
require 'table_saw/foreign_key'
require 'set'
module TableSaw
module Queries
class ForeignKeyRelationships
QUERY = <<~SQL
select
tc.constraint_name,
tc.table_name as from_table,
kcu.column_name as from_column,
ccu.table_name as to_table,
ccu.column_name as to_column
from information_schema.table_constraints tc
join information_schema.key_column_usage kcu on tc.constraint_name = kcu.constraint_name
join information_schema.constraint_column_usage ccu on tc.constraint_name = ccu.constraint_name
where tc.constraint_type = 'FOREIGN KEY'
SQL
def constraint_names
@constraint_names ||= result.each_with_object(Hash.new { |h, k| h[k] = [] }) do |row, memo|
memo[row['from_table']].push(row['constraint_name'])
end
end
def foreign_keys
@foreign_keys ||= result.map do |row|
TableSaw::ForeignKey.new(name: row['constraint_name'],
from_table: row['from_table'], from_column: row['from_column'],
to_table: row['to_table'], to_column: row['to_column'])
end
end
private
def result
@result ||= TableSaw::Connection.exec(QUERY)
end
end
end
end
Version data entries
7 entries across 7 versions & 1 rubygems