Class: Arel::SelectManager

22
23
24

# File 'lib/arel/select_manager.rb', line 22

def constraints
  @ctx.wheres
end

148
149
150
151
152
153
154

# File 'lib/arel/select_manager.rb', line 148

def distinct(value = true)
  if value
    @ctx.set_quantifier = Arel::Nodes::Distinct.new
  else
    @ctx.set_quantifier = nil
  end
end

195
196
197

# File 'lib/arel/select_manager.rb', line 195

def except other
  Nodes::Except.new ast, other.ast
end

42
43
44

# File 'lib/arel/select_manager.rb', line 42

def exists
  Arel::Nodes::Exists.new @ast
end

91
92
93
94
95
96
97
98
99
100
101
102
103
104
105

# File 'lib/arel/select_manager.rb', line 91

def from table
  table = Nodes::SqlLiteral.new(table) if String === table
  # FIXME: this is a hack to support
  # test_with_two_tables_in_from_without_getting_double_quoted
  # from the AR tests.

  case table
  when Nodes::Join
    @ctx.source.right << table
  else
    @ctx.source.left = table
  end

  self
end

107
108
109

# File 'lib/arel/select_manager.rb', line 107

def froms
  @ast.cores.map { |x| x.from }.compact
end

80
81
82
83
84
85
86
87
88
89

# File 'lib/arel/select_manager.rb', line 80

def group *columns
  columns.each do |column|
    # FIXME: backwards compat
    column = Nodes::SqlLiteral.new(column) if String === column
    column = Nodes::SqlLiteral.new(column.to_s) if Symbol === column

    @ctx.groups.push Nodes::Group.new column
  end
  self
end

124
125
126
127

# File 'lib/arel/select_manager.rb', line 124

def having *exprs
  @ctx.having = Nodes::Having.new(collapse(exprs, @ctx.having))
  self
end

12
13
14
15

# File 'lib/arel/select_manager.rb', line 12

def initialize_copy other
  super
  @ctx = @ast.cores.last
end

272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294

# File 'lib/arel/select_manager.rb', line 272

def insert values
  if $VERBOSE
    warn <<-eowarn
insert (#{caller.first}) is deprecated and will be removed in ARel 4.0.0. Please
switch to `compile_insert`
    eowarn
  end

  im = compile_insert(values)
  table = @ctx.froms

  primary_key      = table.primary_key
  primary_key_name = primary_key.name if primary_key

  # FIXME: in AR tests values sometimes were Array and not Hash therefore is_a?(Hash) check is added
  primary_key_value = primary_key && values.is_a?(Hash) && values[primary_key]
  im.into table
  # Oracle adapter needs primary key name to generate RETURNING ... INTO ... clause
  # for tables which assign primary key value using trigger.
  # RETURNING ... INTO ... clause will be added only if primary_key_value is nil
  # therefore it is necessary to pass primary key value as well
  @engine.connection.insert im.to_sql, 'AREL', primary_key_name, primary_key_value
end

191
192
193

# File 'lib/arel/select_manager.rb', line 191

def intersect other
  Nodes::Intersect.new ast, other.ast
end

111
112
113
114
115
116
117
118
119
120
121
122

# File 'lib/arel/select_manager.rb', line 111

def join relation, klass = Nodes::InnerJoin
  return self unless relation

  case relation
  when String, Nodes::SqlLiteral
    raise if relation.blank?
    klass = Nodes::StringJoin
  end

  @ctx.source.right << create_join(relation, nil, klass)
  self
end

237
238
239

# File 'lib/arel/select_manager.rb', line 237

def join_sources
  @ctx.source.right
end

223
224
225
226
227
228

# File 'lib/arel/select_manager.rb', line 223

def join_sql
  return nil if @ctx.source.right.empty?

  sql = visitor.dup.extend(Visitors::JoinSql).accept @ctx
  Nodes::SqlLiteral.new sql
end

245
246
247
248
249
250
251

# File 'lib/arel/select_manager.rb', line 245

def joins manager
  if $VERBOSE
    warn "joins is deprecated and will be removed in 4.0.0"
    warn "please remove your call to joins from #{caller.first}"
  end
  manager.join_sql
end

17
18
19

# File 'lib/arel/select_manager.rb', line 17

def limit
  @ast.limit && @ast.limit.expr
end

58
59
60
61
62
63
64
65
66
67
68
69

# File 'lib/arel/select_manager.rb', line 58

def lock locking = Arel.sql('FOR UPDATE')
  case locking
  when true
    locking = Arel.sql('FOR UPDATE')
  when Arel::Nodes::SqlLiteral
  when String
    locking = Arel.sql locking
  end

  @ast.lock = Nodes::Lock.new(locking)
  self
end

71
72
73

# File 'lib/arel/select_manager.rb', line 71

def locked
  @ast.lock
end

26
27
28

# File 'lib/arel/select_manager.rb', line 26

def offset
  @ast.offset && @ast.offset.expr
end

75
76
77
78

# File 'lib/arel/select_manager.rb', line 75

def on *exprs
  @ctx.source.right.last.right = Nodes::On.new(collapse(exprs))
  self
end

156
157
158
159
160
161
162

# File 'lib/arel/select_manager.rb', line 156

def order *expr
  # FIXME: We SHOULD NOT be converting these to SqlLiteral automatically
  @ast.orders.concat expr.map { |x|
    String === x || Symbol === x ? Nodes::SqlLiteral.new(x.to_s) : x
  }
  self
end

230
231
232
233
234
235

# File 'lib/arel/select_manager.rb', line 230

def order_clauses
  visitor = Visitors::OrderClauses.new(@engine.connection)
  visitor.accept(@ast).map { |x|
    Nodes::SqlLiteral.new x
  }
end

164
165
166

# File 'lib/arel/select_manager.rb', line 164

def orders
  @ast.orders
end

135
136
137
138
139
140
141
142

# File 'lib/arel/select_manager.rb', line 135

def project *projections
  # FIXME: converting these to SQLLiterals is probably not good, but
  # rails tests require it.
  @ctx.projections.concat projections.map { |x|
    [Symbol, String].include?(x.class) ? SqlLiteral.new(x.to_s) : x
  }
  self
end

144
145
146

# File 'lib/arel/select_manager.rb', line 144

def projections= projections
  @ctx.projections = projections
end

30
31
32
33
34
35
36
37

# File 'lib/arel/select_manager.rb', line 30

def skip amount
  if amount
    @ast.offset = Nodes::Offset.new(amount)
  else
    @ast.offset = nil
  end
  self
end

241
242
243

# File 'lib/arel/select_manager.rb', line 241

def source
  @ctx.source
end

211
212
213
214
215
216
217
218
219
220

# File 'lib/arel/select_manager.rb', line 211

def take limit
  if limit
    @ast.limit = Nodes::Limit.new(limit)
    @ctx.top   = Nodes::Top.new(limit)
  else
    @ast.limit = nil
    @ctx.top   = nil
  end
  self
end

265
266
267
268
269

# File 'lib/arel/select_manager.rb', line 265

def to_a # :nodoc:
  warn "to_a is deprecated. Please remove it from #{caller[0]}"
  # FIXME: I think `select` should be made public...
  @engine.connection.send(:select, to_sql, 'AREL').map { |x| Row.new(x) }
end

180
181
182
183
184
185
186
187
188
189

# File 'lib/arel/select_manager.rb', line 180

def union operation, other = nil
  if other
    node_class = Nodes.const_get("Union#{operation.to_s.capitalize}")
  else
    other = operation
    node_class = Nodes::Union
  end

  node_class.new self.ast, other.ast
end

50
51
52
53
54
55
56

# File 'lib/arel/select_manager.rb', line 50

def where_clauses
  if $VERBOSE
    warn "(#{caller.first}) where_clauses is deprecated and will be removed in arel 4.0.0 with no replacement"
  end
  to_sql = Visitors::ToSql.new @engine.connection
  @ctx.wheres.map { |c| to_sql.accept c }
end

173
174
175
176
177
178

# File 'lib/arel/select_manager.rb', line 173

def where_sql
  return if @ctx.wheres.empty?

  viz = Visitors::WhereSql.new @engine.connection
  Nodes::SqlLiteral.new viz.accept @ctx
end

168
169
170
171

# File 'lib/arel/select_manager.rb', line 168

def wheres
  warn "#{caller[0]}: SelectManager#wheres is deprecated and will be removed in ARel 4.0.0 with no replacement"
  Compatibility::Wheres.new @engine.connection, @ctx.wheres
end

129
130
131
132
133

# File 'lib/arel/select_manager.rb', line 129

def window name
  window = Nodes::NamedWindow.new(name)
  @ctx.windows.push window
  window
end

200
201
202
203
204
205
206
207
208
209

# File 'lib/arel/select_manager.rb', line 200

def with *subqueries
  if subqueries.first.is_a? Symbol
    node_class = Nodes.const_get("With#{subqueries.shift.to_s.capitalize}")
  else
    node_class = Nodes::With
  end
  @ast.with = node_class.new(subqueries.flatten)

  self
end