dohmysql 0.2.22 → 0.2.23

data/lib/doh/mysql/handle.rb

@@ -76,10 +76,18 @@ class Handle
     insert_hash_helper(hash, table, 'INSERT', quote_strings)
   end
 
+  def insert_hashes(hashes, table, quote_strings = true)
+    insert_hashes_helper(hashes, table, 'INSERT', quote_strings)
+  end
+
   def insert_ignore_hash(hash, table, quote_strings = true)
     insert_hash_helper(hash, table, 'INSERT IGNORE', quote_strings)
   end
 
+  def insert_ignore_hashes(hash, table, quote_strings = true)
+    insert_hashes_helper(hash, table, 'INSERT IGNORE', quote_strings)
+  end
+
   def replace_hash(hash, table, quote_strings = true)
     insert_hash_helper(hash, table, 'REPLACE', quote_strings)
   end
@@ -190,6 +198,10 @@ private
     raise
   end
 
+  def get_key_insert_str(keys)
+    "(`#{keys.join('`,`')}`)"
+  end
+
   def insert_hash_helper(hash, table, keyword, quote_strings)
     names = []
     values = []
@@ -201,6 +213,23 @@ private
     insert("#{keyword} INTO #{table} (`#{names.join('`,`')}`) VALUES (#{values.join(',')})")
   end
 
+  def insert_hashes_helper(hashes, table, keyword, quote_strings)
+    valuestrs = []
+    keys = hashes[0].keys
+    keystr = get_key_insert_str(keys)
+    hashes.each do |hash|
+      values = []
+      keys.each do |key|
+        value = hash[key]
+        values.push(if quote_strings || !value.is_a?(String) then value.to_sql else value end)
+      end
+      valuestrs.push("(#{values.join(',')})")
+    end
+
+    insert("#{keyword} INTO #{table} #{keystr} VALUES #{valuestrs.join(",")}")
+  end
+
+
   def get_row_builder(row_builder = nil)
     if row_builder.nil?
       TypedRowBuilder.new

data/test/handle.dt.rb

@@ -52,18 +52,24 @@ class Test_Handle < DohTest::TestGroup
     dbh = get_dbh
     dbh.query("CREATE TEMPORARY TABLE #{tbl} (value INT KEY)")
     hash1 = {'value' => 1}
+    hash2 = {'value' => 2}
+    hash3 = {'value' => 3}
+    hash4 = {'value' => 4}
     assert_equal(0, dbh.insert_hash(hash1, tbl))
     assert_equal(0, dbh.insert_ignore_hash(hash1, tbl))
+    assert_equal(1, dbh.select_field("SELECT COUNT(*) from #{tbl}"))
+    assert_equal(0, dbh.insert_hashes([hash2, hash3, hash4], tbl))
+    assert_equal(4, dbh.select_field("SELECT COUNT(*) from #{tbl}"))
     assert_raises(Mysql2::Error) { dbh.insert_hash(hash1, tbl) }
   end
 
   def test_bad_handle_reconnect
     dbh = get_dbh
     begin
-      Timeout.timeout(0.001) { Doh.db.select_field("SELECT SLEEP(1), 0 FROM mysql.user GROUP BY 2") }
+      Timeout.timeout(0.001) { dbh.select_field("SELECT SLEEP(1), 0 FROM mysql.user GROUP BY 2") }
     rescue => _
     end
-    assert_equal(3, Doh.db.select_field("SELECT 3 FROM mysql.user GROUP BY 1"))
+    assert_equal(3, dbh.select_field("SELECT 3 FROM mysql.user GROUP BY 1"))
   end
 end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: dohmysql
 version: !ruby/object:Gem::Version
-  version: 0.2.22
+  version: 0.2.23
   prerelease: 
 platform: ruby
 authors: