mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
096719ce2a
`TreeManager` is inherited by `InsertManager` but `where` on `InsertManager` doesn't work. And also, remove `@ctx` since the instance variable is used only for the `SelectManager`.
32 lines
630 B
Ruby
32 lines
630 B
Ruby
# frozen_string_literal: true
|
|
|
|
module Arel # :nodoc: all
|
|
class UpdateManager < Arel::TreeManager
|
|
include TreeManager::StatementMethods
|
|
|
|
def initialize(table = nil)
|
|
@ast = Nodes::UpdateStatement.new(table)
|
|
end
|
|
|
|
###
|
|
# UPDATE +table+
|
|
def table(table)
|
|
@ast.relation = table
|
|
self
|
|
end
|
|
|
|
def set(values)
|
|
if String === values
|
|
@ast.values = [values]
|
|
else
|
|
@ast.values = values.map { |column, value|
|
|
Nodes::Assignment.new(
|
|
Nodes::UnqualifiedColumn.new(column),
|
|
value
|
|
)
|
|
}
|
|
end
|
|
self
|
|
end
|
|
end
|
|
end
|