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
|
module ActiveRecord
class Relation
delegate :delete, :to_sql, :to => :relation
CLAUSES_METHODS = ["group", "order", "on"].freeze
attr_reader :relation, :klass
def initialize(klass, table = nil)
@klass = klass
@relation = Arel::Table.new(table || @klass.table_name)
end
def to_a
@klass.find_by_sql(@relation.to_sql)
end
def each(&block)
to_a.each(&block)
end
def first
@relation = @relation.take(1)
to_a.first
end
for clause in CLAUSES_METHODS
class_eval %{
def #{clause}!(_#{clause})
@relation = @relation.#{clause}(_#{clause}) if _#{clause}
self
end
}
end
def select!(selection)
@relation = @relation.project(selection) if selection
self
end
def limit!(limit)
@relation = @relation.take(limit) if limit
self
end
def offset!(offset)
@relation = @relation.skip(offset) if offset
self
end
def joins!(joins, join_type = nil)
if !joins.blank?
@relation = case joins
when String
@relation.join(joins)
when Hash, Array, Symbol
if @klass.send(:array_of_strings?, joins)
@relation.join(joins.join(' '))
else
@relation.join(@klass.send(:build_association_joins, joins))
end
else
@relation.join(joins, join_type)
end
end
self
end
def conditions!(conditions)
if !conditions.blank?
conditions = @klass.send(:merge_conditions, conditions) if [String, Hash, Array].include?(conditions.class)
@relation = @relation.where(conditions)
end
self
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
|