Class: Arel::SelectManager

Inherits:
TreeManager show all
Includes:
Crud
Defined in:
lib/arel/select_manager.rb

Defined Under Namespace

Classes: Row

Instance Attribute Summary

Attributes inherited from TreeManager

#ast, #engine

Instance Method Summary collapse

Methods included from Crud

#compile_delete, #compile_insert, #compile_update, #create_insert

Methods inherited from TreeManager

#to_dot, #to_sql, #visitor, #where

Methods included from FactoryMethods

#create_and, #create_false, #create_join, #create_on, #create_string_join, #create_table_alias, #create_true, #grouping, #lower

Constructor Details

#initialize(engine, table = nil) ⇒ SelectManager

Returns a new instance of SelectManager.


5
6
7
8
9
10
# File 'lib/arel/select_manager.rb', line 5

def initialize engine, table = nil
  super(engine)
  @ast   = Nodes::SelectStatement.new
  @ctx    = @ast.cores.last
  from table
end

Instance Method Details

#as(other) ⇒ Object


46
47
48
# File 'lib/arel/select_manager.rb', line 46

def as other
  create_table_alias grouping(@ast), Nodes::SqlLiteral.new(other)
end

#constraintsObject


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

def constraints
  @ctx.wheres
end

#distinct(value = true) ⇒ Object


144
145
146
147
148
149
150
# File 'lib/arel/select_manager.rb', line 144

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

#except(other) ⇒ Object Also known as: minus


186
187
188
# File 'lib/arel/select_manager.rb', line 186

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

#existsObject

Produces an Arel::Nodes::Exists node


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

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

#from(table) ⇒ Object


83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
# File 'lib/arel/select_manager.rb', line 83

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

#fromsObject


99
100
101
# File 'lib/arel/select_manager.rb', line 99

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

#group(*columns) ⇒ Object


72
73
74
75
76
77
78
79
80
81
# File 'lib/arel/select_manager.rb', line 72

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

#having(*exprs) ⇒ Object


116
117
118
119
# File 'lib/arel/select_manager.rb', line 116

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

#initialize_copy(other) ⇒ Object


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

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

#intersect(other) ⇒ Object


182
183
184
# File 'lib/arel/select_manager.rb', line 182

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

#join(relation, klass = Nodes::InnerJoin) ⇒ Object


103
104
105
106
107
108
109
110
111
112
113
114
# File 'lib/arel/select_manager.rb', line 103

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

#join_sourcesObject


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

def join_sources
  @ctx.source.right
end

#join_sqlObject


214
215
216
217
218
219
# File 'lib/arel/select_manager.rb', line 214

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

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

#joins(manager) ⇒ Object


236
237
238
239
240
241
242
# File 'lib/arel/select_manager.rb', line 236

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

#limitObject Also known as: taken


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

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

#lock(locking = Arel.sql('FOR UPDATE')) ⇒ Object


50
51
52
53
54
55
56
57
58
59
60
61
# File 'lib/arel/select_manager.rb', line 50

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

#lockedObject


63
64
65
# File 'lib/arel/select_manager.rb', line 63

def locked
  @ast.lock
end

#offsetObject


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

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

#on(*exprs) ⇒ Object


67
68
69
70
# File 'lib/arel/select_manager.rb', line 67

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

#order(*expr) ⇒ Object


152
153
154
155
156
157
158
# File 'lib/arel/select_manager.rb', line 152

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

#order_clausesObject


221
222
223
224
225
226
# File 'lib/arel/select_manager.rb', line 221

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

#ordersObject


160
161
162
# File 'lib/arel/select_manager.rb', line 160

def orders
  @ast.orders
end

#project(*projections) ⇒ Object


127
128
129
130
131
132
133
134
# File 'lib/arel/select_manager.rb', line 127

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

#projectionsObject


136
137
138
# File 'lib/arel/select_manager.rb', line 136

def projections
  @ctx.projections
end

#projections=(projections) ⇒ Object


140
141
142
# File 'lib/arel/select_manager.rb', line 140

def projections= projections
  @ctx.projections = projections
end

#skip(amount) ⇒ Object Also known as: offset=


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

#sourceObject


232
233
234
# File 'lib/arel/select_manager.rb', line 232

def source
  @ctx.source
end

#take(limit) ⇒ Object Also known as: limit=


202
203
204
205
206
207
208
209
210
211
# File 'lib/arel/select_manager.rb', line 202

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

#to_aObject

:nodoc:


256
257
258
259
260
# File 'lib/arel/select_manager.rb', line 256

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

#union(operation, other = nil) ⇒ Object


171
172
173
174
175
176
177
178
179
180
# File 'lib/arel/select_manager.rb', line 171

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

#where_sqlObject


164
165
166
167
168
169
# File 'lib/arel/select_manager.rb', line 164

def where_sql
  return if @ctx.wheres.empty?

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

#window(name) ⇒ Object


121
122
123
124
125
# File 'lib/arel/select_manager.rb', line 121

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

#with(*subqueries) ⇒ Object


191
192
193
194
195
196
197
198
199
200
# File 'lib/arel/select_manager.rb', line 191

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