blob: f48efabec2c3033b14e3e0b1f41cd88ed7dba0ed (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
require 'active_support/core_ext/string/conversions'
module ActiveRecord
module Associations
# Keeps track of table aliases for ActiveRecord::Associations::ClassMethods::JoinDependency
class AliasTracker # :nodoc:
# other_sql is some other sql which might conflict with the aliases we assign here. Therefore
# we store other_sql so that we can scan it before assigning a specific name.
def initialize(other_sql)
@aliases = Hash.new
@other_sql = other_sql.to_s.downcase
end
def aliased_name_for(table_name, aliased_name = nil)
aliased_name ||= table_name
initialize_count_for(table_name) if @aliases[table_name].nil?
if @aliases[table_name].zero?
# If it's zero, we can have our table_name
@aliases[table_name] = 1
table_name
else
# Otherwise, we need to use an alias
aliased_name = connection.table_alias_for(aliased_name)
initialize_count_for(aliased_name) if @aliases[aliased_name].nil?
# Update the count
@aliases[aliased_name] += 1
if @aliases[aliased_name] > 1
"#{truncate(aliased_name)}_#{@aliases[aliased_name]}"
else
aliased_name
end
end
end
private
def initialize_count_for(name)
@aliases[name] = 0
unless @other_sql.blank?
# quoted_name should be downcased as some database adapters (Oracle) return quoted name in uppercase
quoted_name = connection.quote_table_name(name.downcase).downcase
# Table names
@aliases[name] += @other_sql.scan(/join(?:\s+\w+)?\s+#{quoted_name}\son/).size
# Table aliases
@aliases[name] += @other_sql.scan(/join(?:\s+\w+)?\s+\S+\s+#{quoted_name}\son/).size
end
@aliases[name]
end
def truncate(name)
name[0..connection.table_alias_length-3]
end
def connection
ActiveRecord::Base.connection
end
end
end
end
|
