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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
module ActiveRecord
class Relation
delegate :to_sql, :to => :relation
attr_reader :relation, :klass
def initialize(klass, relation)
@klass, @relation = klass, relation
@readonly = false
@associations_to_preload = []
@eager_load_associations = []
end
def preload(association)
@associations_to_preload += association
self
end
def eager_load(association)
@eager_load_associations += association
self
end
def readonly
@readonly = true
self
end
def to_a
records = if @eager_load_associations.any?
catch :invalid_query do
return @klass.send(:find_with_associations, {
:select => @relation.send(:select_clauses).join(', '),
:joins => @relation.joins(relation),
:group => @relation.send(:group_clauses).join(', '),
:order => @relation.send(:order_clauses).join(', '),
:conditions => @relation.send(:where_clauses).join("\n\tAND "),
:limit => @relation.taken
},
ActiveRecord::Associations::ClassMethods::JoinDependency.new(@klass, @eager_load_associations, nil))
end
[]
else
@klass.find_by_sql(@relation.to_sql)
end
@klass.send(:preload_associations, records, @associations_to_preload) unless @associations_to_preload.empty?
records.each { |record| record.readonly! } if @readonly
records
end
def each(&block)
to_a.each(&block)
end
def first
@relation = @relation.take(1)
to_a.first
end
def select(selects)
selects.blank? ? self : Relation.new(@klass, @relation.project(selects))
end
def group(groups)
groups.blank? ? self : Relation.new(@klass, @relation.group(groups))
end
def order(orders)
orders.blank? ? self : Relation.new(@klass, @relation.order(orders))
end
def limit(limits)
limits.blank? ? self : Relation.new(@klass, @relation.take(limits))
end
def offset(offsets)
offsets.blank? ? self : Relation.new(@klass, @relation.skip(offsets))
end
def on(join)
join.blank? ? self : Relation.new(@klass, @relation.on(join))
end
def joins(join, join_type = nil)
if join.blank?
self
else
join = case join
when String
@relation.join(join)
when Hash, Array, Symbol
if @klass.send(:array_of_strings?, join)
@relation.join(join.join(' '))
else
@relation.join(@klass.send(:build_association_joins, join))
end
else
@relation.join(join, join_type)
end
Relation.new(@klass, join)
end
end
def conditions(conditions)
if conditions.blank?
self
else
conditions = @klass.send(:merge_conditions, conditions) if [String, Hash, Array].include?(conditions.class)
Relation.new(@klass, @relation.where(conditions))
end
end
def respond_to?(method)
if @relation.respond_to?(method) || Array.instance_methods.include?(method.to_s)
true
else
super
end
end
private
def method_missing(method, *args, &block)
if @relation.respond_to?(method)
@relation.send(method, *args, &block)
elsif Array.instance_methods.include?(method.to_s)
to_a.send(method, *args, &block)
end
end
end
end
|