Sha256: bc7b87a743b23eb38063c36d012f4e2faaf0ce88cdbb807339f445d2a9d66ddf
Contents?: true
Size: 1.14 KB
Versions: 7
Compression:
Stored size: 1.14 KB
Contents
# frozen_string_literal: true
module TableSaw
module DependencyGraph
class BelongsToDirectives
attr_reader :directive
def initialize(directive)
@directive = directive
end
def call
associations.map do |from_column, to_table|
TableSaw::DependencyGraph::AddDirective.new(to_table, ids: ids[from_column].to_a, partial: directive.partial?)
end
end
private
def associations
TableSaw.information_schema.belongs_to.fetch(directive.table_name, {})
end
def ids
@ids ||= query_result.each_with_object(Hash.new { |h, k| h[k] = Set.new }) do |row, memo|
associations.each_key { |key| memo[key].add row[key] unless row[key].nil? }
end
end
def query_result
return [] unless directive.selectable?
TableSaw::Connection.exec(
format('select %{columns} from %{table_name} where %{primary_key} in (%{ids})',
primary_key: directive.primary_key, columns: associations.keys.join(','),
table_name: directive.table_name, ids: directive.ids.join(','))
)
end
end
end
end
Version data entries
7 entries across 7 versions & 1 rubygems