rails--rails/lib/arel/insert_manager.rb

35 lines
698 B
Ruby
Raw Normal View History

module Arel
class InsertManager < Arel::TreeManager
def initialize engine
super
2010-11-05 17:09:09 -04:00
@ast = Nodes::InsertStatement.new
end
def into table
2010-11-05 17:09:09 -04:00
@ast.relation = table
self
end
2010-08-13 18:43:32 -04:00
2010-11-05 17:09:09 -04:00
def columns; @ast.columns end
def values= val; @ast.values = val; end
2010-08-13 21:28:38 -04:00
def insert fields
return if fields.empty?
if String === fields
2010-11-05 17:09:09 -04:00
@ast.values = SqlLiteral.new(fields)
else
2010-11-05 17:09:09 -04:00
@ast.relation ||= fields.first.first.relation
2010-08-13 21:28:38 -04:00
values = []
fields.each do |column, value|
2010-11-05 17:09:09 -04:00
@ast.columns << column
values << value
end
2010-11-05 17:09:09 -04:00
@ast.values = Nodes::Values.new values, @ast.columns
2010-08-13 21:28:38 -04:00
end
end
end
end