From 5e6312e1745dc278ba0a99bf2bc7b78977785d35 Mon Sep 17 00:00:00 2001
From: Kir Shatrov <shatrov@me.com>
Date: Sun, 21 May 2017 13:12:08 +0100
Subject: Support INSERT with multiple values

---
 lib/arel/insert_manager.rb    |  4 ++++
 lib/arel/nodes.rb             |  1 +
 lib/arel/nodes/values_list.rb | 13 +++++++++++++
 lib/arel/visitors/to_sql.rb   | 22 ++++++++++++++++++++++
 4 files changed, 40 insertions(+)
 create mode 100644 lib/arel/nodes/values_list.rb

(limited to 'lib')

diff --git a/lib/arel/insert_manager.rb b/lib/arel/insert_manager.rb
index f9a598e8b7..dcbac6cb43 100644
--- a/lib/arel/insert_manager.rb
+++ b/lib/arel/insert_manager.rb
@@ -40,5 +40,9 @@ module Arel
     def create_values values, columns
       Nodes::Values.new values, columns
     end
+
+    def create_values_list(rows)
+      Nodes::ValuesList.new(rows)
+    end
   end
 end
diff --git a/lib/arel/nodes.rb b/lib/arel/nodes.rb
index 8c9815a96b..bb8ad6500b 100644
--- a/lib/arel/nodes.rb
+++ b/lib/arel/nodes.rb
@@ -44,6 +44,7 @@ require 'arel/nodes/function'
 require 'arel/nodes/count'
 require 'arel/nodes/extract'
 require 'arel/nodes/values'
+require 'arel/nodes/values_list'
 require 'arel/nodes/named_function'
 
 # windows
diff --git a/lib/arel/nodes/values_list.rb b/lib/arel/nodes/values_list.rb
new file mode 100644
index 0000000000..b39aaa1465
--- /dev/null
+++ b/lib/arel/nodes/values_list.rb
@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+module Arel
+  module Nodes
+    class ValuesList < Node
+      attr_reader :rows
+
+      def initialize(rows)
+        @rows = rows
+        super()
+      end
+    end
+  end
+end
diff --git a/lib/arel/visitors/to_sql.rb b/lib/arel/visitors/to_sql.rb
index 486c51a183..3f1e390dcc 100644
--- a/lib/arel/visitors/to_sql.rb
+++ b/lib/arel/visitors/to_sql.rb
@@ -166,6 +166,28 @@ module Arel
         collector << "FALSE"
       end
 
+      def visit_Arel_Nodes_ValuesList o, collector
+        collector << "VALUES "
+
+        len = o.rows.length - 1
+        o.rows.each_with_index { |row, i|
+          collector << '('
+          row_len = row.length - 1
+          row.each_with_index do |value, k|
+            case value
+            when Nodes::SqlLiteral, Nodes::BindParam
+              collector = visit(value, collector)
+            else
+              collector << quote(value)
+            end
+            collector << COMMA unless k == row_len
+          end
+          collector << ')'
+          collector << COMMA unless i == len
+        }
+        collector
+      end
+
       def visit_Arel_Nodes_Values o, collector
         collector << "VALUES ("
 
-- 
cgit v1.2.3