Class: ActiveRecord::Associations::JoinDependency
- Defined in:
- lib/brick.rb
Overview
For AR >= 4.2
Instance Method Summary collapse
-
#apply_column_aliases(relation) ⇒ Object
An intelligent .eager_load() and .includes() that creates t0_r0 style aliases only for the columns used in .select().
Instance Method Details
#apply_column_aliases(relation) ⇒ Object
An intelligent .eager_load() and .includes() that creates t0_r0 style aliases only for the columns used in .select(). To enable this behaviour, include the flag :_brick_eager_load as the first entry in your .select(). More information: discuss.rubyonrails.org/t/includes-and-select-for-joined-data/81640
1688 1689 1690 1691 1692 1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734 1735 1736 1737 1738 1739 1740 1741 1742 1743 |
# File 'lib/brick.rb', line 1688 def apply_column_aliases(relation) if !(@join_root_alias = relation.select_values.empty?) && relation.select_values.first.to_s == '_brick_eager_load' relation.select_values.shift used_cols = {} # Find and expand out all column names being used in select(...) new_select_values = relation.select_values.map(&:to_s).each_with_object([]) do |col, s| if col.include?(' ') # Some expression? (No chance for a simple column reference) s << col # Just pass it through else col = if (col_parts = col.split('.')).length == 1 [col] else [col_parts[0..-2].join('.'), col_parts.last] end used_cols[col] = nil end end if new_select_values.present? relation.select_values = new_select_values else relation.select_values.clear end @aliases ||= Aliases.new(join_root.each_with_index.map do |join_part, i| join_alias = join_part.table&.table_alias || join_part.table_name keys = [join_part.base_klass.primary_key] # Always include the primary key # # %%% Optional to include all foreign keys: # keys.concat(join_part.base_klass.reflect_on_all_associations.select { |a| a.belongs_to? }.map(&:foreign_key)) # Add foreign keys out to referenced tables that we belongs_to join_part.children.each { |child| keys << child.reflection.foreign_key if child.reflection.belongs_to? } # Add the foreign key that got us here -- "the train we rode in on" -- if we arrived from # a has_many or has_one: if join_part.is_a?(ActiveRecord::Associations::JoinDependency::JoinAssociation) && !join_part.reflection.belongs_to? keys << join_part.reflection.foreign_key end keys = keys.compact # In case we're using composite_primary_keys j = 0 columns = join_part.column_names.each_with_object([]) do |column_name, s| # Include columns chosen in select(...) as well as the PK and any relevant FKs if used_cols.keys.find { |c| (c.length == 1 || c.first == join_alias) && c.last == column_name } || keys.find { |c| c == column_name } s << Aliases::Column.new(column_name, "t#{i}_r#{j}") end j += 1 end Aliases::Table.new(join_part, columns) end) end relation._select!(-> { aliases.columns }) end |