From e1155aa922a04e1cce60a660b4b779bfe3d7d7f5 Mon Sep 17 00:00:00 2001
From: Aaron Patterson <aaron.patterson@gmail.com>
Date: Sat, 14 Aug 2010 19:48:33 -0700
Subject: update is working on the select manager

---
 lib/arel/select_manager.rb  | 9 +++++++++
 lib/arel/update_manager.rb  | 4 ++++
 lib/arel/visitors/to_sql.rb | 2 +-
 3 files changed, 14 insertions(+), 1 deletion(-)

(limited to 'lib/arel')

diff --git a/lib/arel/select_manager.rb b/lib/arel/select_manager.rb
index 444ca8f716..7379596aba 100644
--- a/lib/arel/select_manager.rb
+++ b/lib/arel/select_manager.rb
@@ -26,6 +26,15 @@ module Arel
       self
     end
 
+    # FIXME: this method should go away
+    def update values
+      um = UpdateManager.new @engine
+      um.table values.first.first.relation
+      um.set values
+      um.wheres = @ctx.wheres
+      @engine.connection.execute um.to_sql
+    end
+
     # FIXME: this method should go away
     def insert values
       im = InsertManager.new @engine
diff --git a/lib/arel/update_manager.rb b/lib/arel/update_manager.rb
index cc2f973842..f712103f98 100644
--- a/lib/arel/update_manager.rb
+++ b/lib/arel/update_manager.rb
@@ -12,6 +12,10 @@ module Arel
       self
     end
 
+    def wheres= exprs
+      @head.wheres = exprs
+    end
+
     def where expr
       @head.wheres << expr
       self
diff --git a/lib/arel/visitors/to_sql.rb b/lib/arel/visitors/to_sql.rb
index 4c2a6a10e4..5536ba0267 100644
--- a/lib/arel/visitors/to_sql.rb
+++ b/lib/arel/visitors/to_sql.rb
@@ -16,7 +16,7 @@ module Arel
         [
           "UPDATE #{visit o.relation}",
           ("SET #{o.values.map { |column,value|
-              "#{quote_column_name(column.name)} = #{quote visit value}"
+              "#{quote_column_name(column.name)} = #{value ? quote(visit(value)) : 'NULL'}"
             }.join ', '}" unless o.values.empty?),
           ("WHERE #{o.wheres.map { |x| visit x }.join ' AND '}" unless o.wheres.empty?)
         ].compact.join ' '
-- 
cgit v1.2.3