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
|
# frozen_string_literal: true
module Arel # :nodoc: all
module Visitors
class PostgreSQL < Arel::Visitors::ToSql
CUBE = "CUBE"
ROLLUP = "ROLLUP"
GROUPING_SETS = "GROUPING SETS"
LATERAL = "LATERAL"
private
def visit_Arel_Nodes_Matches(o, collector)
op = o.case_sensitive ? " LIKE " : " ILIKE "
collector = infix_value o, collector, op
if o.escape
collector << " ESCAPE "
visit o.escape, collector
else
collector
end
end
def visit_Arel_Nodes_DoesNotMatch(o, collector)
op = o.case_sensitive ? " NOT LIKE " : " NOT ILIKE "
collector = infix_value o, collector, op
if o.escape
collector << " ESCAPE "
visit o.escape, collector
else
collector
end
end
def visit_Arel_Nodes_Regexp(o, collector)
op = o.case_sensitive ? " ~ " : " ~* "
infix_value o, collector, op
end
def visit_Arel_Nodes_NotRegexp(o, collector)
op = o.case_sensitive ? " !~ " : " !~* "
infix_value o, collector, op
end
def visit_Arel_Nodes_DistinctOn(o, collector)
collector << "DISTINCT ON ( "
visit(o.expr, collector) << " )"
end
def visit_Arel_Nodes_BindParam(o, collector)
collector.add_bind(o.value) { |i| "$#{i}" }
end
def visit_Arel_Nodes_GroupingElement(o, collector)
collector << "( "
visit(o.expr, collector) << " )"
end
def visit_Arel_Nodes_Cube(o, collector)
collector << CUBE
grouping_array_or_grouping_element o, collector
end
def visit_Arel_Nodes_RollUp(o, collector)
collector << ROLLUP
grouping_array_or_grouping_element o, collector
end
def visit_Arel_Nodes_GroupingSet(o, collector)
collector << GROUPING_SETS
grouping_array_or_grouping_element o, collector
end
def visit_Arel_Nodes_Lateral(o, collector)
collector << LATERAL
collector << SPACE
grouping_parentheses o, collector
end
def visit_Arel_Nodes_IsNotDistinctFrom(o, collector)
collector = visit o.left, collector
collector << " IS NOT DISTINCT FROM "
visit o.right, collector
end
def visit_Arel_Nodes_IsDistinctFrom(o, collector)
collector = visit o.left, collector
collector << " IS DISTINCT FROM "
visit o.right, collector
end
# Used by Lateral visitor to enclose select queries in parentheses
def grouping_parentheses(o, collector)
if o.expr.is_a? Nodes::SelectStatement
collector << "("
visit o.expr, collector
collector << ")"
else
visit o.expr, collector
end
end
# Utilized by GroupingSet, Cube & RollUp visitors to
# handle grouping aggregation semantics
def grouping_array_or_grouping_element(o, collector)
if o.expr.is_a? Array
collector << "( "
visit o.expr, collector
collector << " )"
else
visit o.expr, collector
end
end
end
end
end
|
