mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
71 lines
1.4 KiB
Ruby
71 lines
1.4 KiB
Ruby
module Arel
|
|
class SelectManager < Arel::TreeManager
|
|
include Arel::Crud
|
|
|
|
def initialize engine
|
|
super
|
|
@head = Nodes::SelectStatement.new
|
|
@ctx = @head.cores.last
|
|
end
|
|
|
|
def on expr
|
|
@ctx.froms.last.constraint = Nodes::On.new(expr)
|
|
self
|
|
end
|
|
|
|
def from table
|
|
@ctx.froms << table
|
|
self
|
|
end
|
|
|
|
def project *projections
|
|
# FIXME: converting these to SQLLiterals is probably not good, but
|
|
# rails tests require it.
|
|
@ctx.projections.concat projections.map { |x|
|
|
String == x.class ? SqlLiteral.new(x) : x
|
|
}
|
|
self
|
|
end
|
|
|
|
def where expr
|
|
@ctx.wheres << expr
|
|
self
|
|
end
|
|
|
|
def order *expr
|
|
# FIXME: We SHOULD NOT be converting these to SqlLiteral automatically
|
|
@head.orders.concat expr.map { |x|
|
|
String === x ? Nodes::SqlLiteral.new(x) : x
|
|
}
|
|
self
|
|
end
|
|
|
|
def wheres
|
|
Compatibility::Wheres.new @engine, @ctx.wheres
|
|
end
|
|
|
|
def take limit
|
|
@head.limit = limit
|
|
self
|
|
end
|
|
|
|
def join_sql
|
|
viz = Visitors::JoinSql.new @engine
|
|
Nodes::SqlLiteral.new viz.accept @ctx
|
|
end
|
|
|
|
def order_clauses
|
|
Visitors::OrderClauses.new(@engine).accept(@head).map { |x|
|
|
Nodes::SqlLiteral.new x
|
|
}
|
|
end
|
|
|
|
def joins manager
|
|
manager.join_sql
|
|
end
|
|
|
|
def to_a
|
|
raise NotImplementedError
|
|
end
|
|
end
|
|
end
|