2010-08-14 21:29:32 -04:00
|
|
|
module Arel
|
|
|
|
class UpdateManager < Arel::TreeManager
|
|
|
|
def initialize engine
|
|
|
|
super
|
2010-11-05 17:09:09 -04:00
|
|
|
@ast = Nodes::UpdateStatement.new
|
2010-12-07 13:46:30 -05:00
|
|
|
@ctx = @ast
|
2010-08-14 21:29:32 -04:00
|
|
|
end
|
|
|
|
|
2010-09-21 12:20:14 -04:00
|
|
|
def take limit
|
2014-03-24 19:26:09 -04:00
|
|
|
@ast.limit = Nodes::Limit.new(Nodes.build_quoted(limit)) if limit
|
2010-09-21 12:20:14 -04:00
|
|
|
self
|
|
|
|
end
|
|
|
|
|
2010-12-24 17:48:33 -05:00
|
|
|
def key= key
|
2014-03-24 19:26:09 -04:00
|
|
|
@ast.key = Nodes.build_quoted(key)
|
2010-12-24 17:48:33 -05:00
|
|
|
end
|
|
|
|
|
2011-08-11 03:14:51 -04:00
|
|
|
def key
|
|
|
|
@ast.key
|
|
|
|
end
|
|
|
|
|
2010-09-21 12:20:14 -04:00
|
|
|
def order *expr
|
2010-11-05 17:09:09 -04:00
|
|
|
@ast.orders = expr
|
2010-09-21 12:20:14 -04:00
|
|
|
self
|
|
|
|
end
|
|
|
|
|
2010-08-14 21:29:32 -04:00
|
|
|
###
|
|
|
|
# UPDATE +table+
|
|
|
|
def table table
|
2010-11-05 17:09:09 -04:00
|
|
|
@ast.relation = table
|
2010-08-14 21:29:32 -04:00
|
|
|
self
|
|
|
|
end
|
|
|
|
|
2010-08-14 22:48:33 -04:00
|
|
|
def wheres= exprs
|
2010-11-05 17:09:09 -04:00
|
|
|
@ast.wheres = exprs
|
2010-08-14 22:48:33 -04:00
|
|
|
end
|
|
|
|
|
2010-08-14 21:29:32 -04:00
|
|
|
def where expr
|
2010-11-05 17:09:09 -04:00
|
|
|
@ast.wheres << expr
|
2010-08-14 21:29:32 -04:00
|
|
|
self
|
|
|
|
end
|
2010-08-14 22:12:52 -04:00
|
|
|
|
|
|
|
def set values
|
2010-08-18 12:52:16 -04:00
|
|
|
if String === values
|
2010-11-05 17:09:09 -04:00
|
|
|
@ast.values = [values]
|
2010-08-18 12:52:16 -04:00
|
|
|
else
|
2010-11-05 17:09:09 -04:00
|
|
|
@ast.values = values.map { |column,value|
|
2010-09-10 16:36:42 -04:00
|
|
|
Nodes::Assignment.new(
|
2010-08-18 12:52:16 -04:00
|
|
|
Nodes::UnqualifiedColumn.new(column),
|
|
|
|
value
|
|
|
|
)
|
|
|
|
}
|
|
|
|
end
|
2010-08-14 22:12:52 -04:00
|
|
|
self
|
|
|
|
end
|
2010-08-14 21:29:32 -04:00
|
|
|
end
|
|
|
|
end
|