blob: aaf0324fd72f25d3e673be797cd4ac629317bc36 (
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
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
|
module Arel
module Visitors
class Oracle < Arel::Visitors::ToSql
private
def visit_Arel_Nodes_SelectStatement o
o = order_hacks(o)
# if need to select first records without ORDER BY and GROUP BY and without DISTINCT
# then can use simple ROWNUM in WHERE clause
if o.limit && o.orders.empty? && !o.offset && o.cores.first.projections.first !~ /^DISTINCT /
o.cores.last.wheres.push Nodes::LessThanOrEqual.new(
Nodes::SqlLiteral.new('ROWNUM'), o.limit.expr
)
return super
end
if o.limit && o.offset
o = o.dup
limit = o.limit.expr.to_i
offset = o.offset
o.offset = nil
sql = super(o)
return <<-eosql
SELECT * FROM (
SELECT raw_sql_.*, rownum raw_rnum_
FROM (#{sql}) raw_sql_
WHERE rownum <= #{offset.expr.to_i + limit}
)
WHERE #{visit offset}
eosql
end
if o.limit
o = o.dup
limit = o.limit.expr
return "SELECT * FROM (#{super(o)}) WHERE ROWNUM <= #{visit limit}"
end
if o.offset
o = o.dup
offset = o.offset
o.offset = nil
sql = super(o)
return <<-eosql
SELECT * FROM (
SELECT raw_sql_.*, rownum raw_rnum_
FROM (#{sql}) raw_sql_
)
WHERE #{visit offset}
eosql
end
super
end
def visit_Arel_Nodes_Limit o
end
def visit_Arel_Nodes_Offset o
"raw_rnum_ > #{visit o.expr}"
end
###
# Hacks for the order clauses specific to Oracle
def order_hacks o
return o if o.orders.empty?
return o unless o.cores.any? do |core|
core.projections.any? do |projection|
/DISTINCT.*FIRST_VALUE/ === projection
end
end
# Previous version with join and split broke ORDER BY clause
# if it contained functions with several arguments (separated by ',').
#
# orders = o.orders.map { |x| visit x }.join(', ').split(',')
orders = o.orders.map do |x|
string = visit x
if string.include?(',')
split_order_string(string)
else
string
end
end.flatten
o.orders = []
orders.each_with_index do |order, i|
o.orders <<
Nodes::SqlLiteral.new("alias_#{i}__#{' DESC' if /\bdesc$/i === order}")
end
o
end
# Split string by commas but count opening and closing brackets
# and ignore commas inside brackets.
def split_order_string(string)
array = []
i = 0
string.split(',').each do |part|
if array[i]
array[i] << ',' << part
else
# to ensure that array[i] will be String and not Arel::Nodes::SqlLiteral
array[i] = '' << part
end
i += 1 if array[i].count('(') == array[i].count(')')
end
array
end
end
end
end
|
