Class: Joiner::AliasTracker

Inherits:
Object
  • Object
show all
Defined in:
lib/joiner/alias_tracker.rb

Overview

This code is taken straight from Rails, prior to v6.1.0. I’m maintaining a copy here to save myself having to work through aliasing logic myself - there’s a good chance I don’t need all of thiis, but it’ll do to get this gem working with Rails 6.1.

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(connection, aliases) ⇒ AliasTracker

table_joins is an array of arel joins which might conflict with the aliases we assign here



51
52
53
54
 
# File 'lib/joiner/alias_tracker.rb', line 51

def initialize(connection, aliases)
  @aliases    = aliases
  @connection = connection
end

Instance Attribute Details

#aliases ⇒ Object (readonly)

Returns the value of attribute aliases.



77
78
79
 
# File 'lib/joiner/alias_tracker.rb', line 77

def aliases
  @aliases
end

Class Method Details

.create(connection, initial_table, joins, aliases = nil) ⇒ Object

:nodoc:



11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
 
# File 'lib/joiner/alias_tracker.rb', line 11

def self.create(connection, initial_table, joins, aliases = nil)
  if joins.empty?
    aliases ||= Hash.new(0)
  elsif aliases
    default_proc = aliases.default_proc || proc { 0 }
    aliases.default_proc = proc { |h, k|
      h[k] = initial_count_for(connection, k, joins) + default_proc.call(h, k)
    }
  else
    aliases = Hash.new { |h, k|
      h[k] = initial_count_for(connection, k, joins)
    }
  end
  aliases[initial_table] = 1
  new(connection, aliases)
end

.initial_count_for(connection, name, table_joins) ⇒ Object 



28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
 
# File 'lib/joiner/alias_tracker.rb', line 28

def self.initial_count_for(connection, name, table_joins)
  quoted_name = nil

  counts = table_joins.map do |join|
    if join.is_a?(Arel::Nodes::StringJoin)
      # quoted_name should be case ignored as some database adapters (Oracle) return quoted name in uppercase
      quoted_name ||= connection.quote_table_name(name)

      # Table names + table aliases
      join.left.scan(
        /JOIN(?:\s+\w+)?\s+(?:\S+\s+)?(?:#{quoted_name}|#{name})\sON/i
      ).size
    elsif join.is_a?(Arel::Nodes::Join)
      join.left.name == name ? 1 : 0
    else
      raise ArgumentError, "joins list should be initialized by list of Arel::Nodes::Join"
    end
  end

  counts.sum
end

Instance Method Details

#aliased_table_for(table_name, aliased_name, type_caster) ⇒ Object 



56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
 
# File 'lib/joiner/alias_tracker.rb', line 56

def aliased_table_for(table_name, aliased_name, type_caster)
  if aliases[table_name].zero?
    # If it's zero, we can have our table_name
    aliases[table_name] = 1
    Arel::Table.new(table_name, type_caster: type_caster)
  else
    # Otherwise, we need to use an alias
    aliased_name = @connection.table_alias_for(aliased_name)

    # Update the count
    aliases[aliased_name] += 1

    table_alias = if aliases[aliased_name] > 1
      "#{truncate(aliased_name)}_#{aliases[aliased_name]}"
    else
      aliased_name
    end
    Arel::Table.new(table_name, type_caster: type_caster).alias(table_alias)
  end
end