activerecord-bulkwrite 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: baa1dc1f6bd651873a0746548960ef3a70d12d6c
+  data.tar.gz: 13c6220a44d36c96e7b6d2a0187b655f7e567fed
+SHA512:
+  metadata.gz: 31fbe026d454d4f940e7e647d09613450d6300e95e537883fe0d286208ab27ed07547561e626206be456c00c377760328b3b10e4188f27cc051ab96611aed2ba
+  data.tar.gz: 19c1612d0e2988b0796971ff99ff7186e73ea5229eb8df075e551cffc5926fd4dc2a3b7f707f59ef5d516f7e32e3c609328068b59829567f36b2263459dc6889

data/lib/activerecord/bulkwrite.rb

@@ -0,0 +1,63 @@
+require 'active_record'
+
+module ActiveRecord
+  class Base
+    # Insert or update a batch of rows.
+    # When upsert is nil, insert only. Otherwise insert or update on conflict.
+    #
+    # Parameters:
+    # fields: an array of field names to insert value
+    # rows:   an array of array of values to insert. The element, values array,
+    #         must correspond with the fields parameter.
+    # upsert:
+    #   A hash with the following keys::
+    #   conflict: an array of field names on which a unique constrain may fail the
+    #             insert
+    #   update:   an array of field names to update when conflict happens. If
+    #             omitted, it will be "fields - upsert[:conflict]".
+    #   where:    where clause to determine if a row will be updated when
+    #             conflict happens. "EXCLUDED" is used for referencing the row
+    #             that failed in insert. If omitted, update all rows that have
+    #             conflict.
+    #
+    # Return:
+    # The number of affected rows
+    def self.bulk_write(fields, rows, upsert = nil)
+      return 0 if rows.empty?
+
+      columns = fields.map {|name| column_for_attribute(name) }
+      rows = rows.map do |row|
+        values = row.map.with_index do |val, i|
+          # NOTE: Here quote method treats val as a Ruby value from
+          # Value#type_cast_from_user, and thus won't convert a String time to
+          # Time object before passing the string time to database.
+          # That's OK for a time string in UTC, but NOT FOR LOCAL TIME, since
+          # ActiveRecord saves time as type "DATETIME WITHOUT TIMEZONE" in database!
+          connection.quote(val, columns[i])
+        end.join ', '
+        "(#{ values })"
+      end
+      field_list = fields.map{|e| %Q("#{e}") }.join ', '
+      sql = "INSERT INTO #{ table_name } (#{ field_list }) VALUES #{ rows.join ', ' }"
+      if upsert
+        if !upsert[:update]
+          update = fields.map(&:to_s) - upsert[:conflict].map(&:to_s)
+        else
+          update = upsert[:update]
+        end
+        update = update.map {|field| "#{ field } = EXCLUDED.#{ field }" }
+        sql += " ON CONFLICT (#{ upsert[:conflict].join ', ' }) DO UPDATE SET #{ update.join ', ' }"
+        if upsert[:where]
+          sql += " WHERE #{ upsert[:where] }"
+        end
+      end
+      # res is a PG::Result object. See
+      # https://deveiate.org/code/pg/PG/Result.html
+      # for its details.
+      res = connection.execute(sql)
+      res.cmd_tuples
+    end
+
+  end
+end
+

metadata

@@ -0,0 +1,64 @@
+--- !ruby/object:Gem::Specification
+name: activerecord-bulkwrite
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Robert Zhang
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-10-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.0'
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.0'
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+description: 
+email: louirobert@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/activerecord/bulkwrite.rb
+homepage: https://github.com/coin8086/activerecord-bulkwrite
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.1
+signing_key: 
+specification_version: 4
+summary: Bulk write/upsert for ActiveRecord
+test_files: []