multi_insert 0.1.7 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 14fe30a32e5c6287987c5ef9c096ac641711c65a
-  data.tar.gz: 5c08654357b2f14e7599a9a63f9246d821c31223
+  metadata.gz: 832a9abb0d1f4a46f8f0a751ec1dce7a93cc01de
+  data.tar.gz: 415a740b0be619998f708ae58fd8144155327be9
 SHA512:
-  metadata.gz: f5b0e2453ac9474b8d7f37d69af7911bbe800e3c46f55949ce66e45f465b91bb3228ba1b392be125343b1650accb033d9cec751dc7d29529d3d6190b61e3bc8f
-  data.tar.gz: 66f3ccccc3d543da757f4f627b1b165dde379f9ccf70c15ebac9ab1ea078cd008e226795da6fdd306e5a5b6f0cd75692c2e0d35dc7fb1f7ba5d196939174e2ff
+  metadata.gz: 1dbe2dd054728c86813136dbc22ce6484c6cdd3c508fea3dcc6bde96ed3fa73a197bfcbccdd0dc5442ab3cb9971e138bb2c0d3a481a498c80d2c0c0b5f0a37aa
+  data.tar.gz: 8a6474bcd0c7099ab5737b3a38736167e1eea0e0516914d3ed4a88b2e2553bcf7b994c289fa6cfd77891b2313291daf0805d85ef27e7143f7748190f8f54b277

data/lib/multi_insert/query.rb

@@ -3,8 +3,26 @@ require 'multi_insert/query_builder'
 
 module MultiInsert
   class Query
+    class OnConflict
+      def initialize(query, column)
+        @query = query
+        @column = column.to_sym
+      end
+
+      def do_nothing
+        @query.on_conflict_sql(::MultiInsert::QueryBuilder.on_conflict_do_nothing(@column))
+      end
+
+      def do_update(values)
+        @query.on_conflict_sql(::MultiInsert::QueryBuilder.on_conflict_do_update(@column, values, @query.opts))
+      end
+    end
+
+    attr_reader :opts
+
     def initialize(table, columns, values, opts = {})
       @table = table.to_sym
+      @opts = opts
       @sql_insert = ::MultiInsert::QueryBuilder.insert(table, columns, values, opts)
     end
 
@@ -20,10 +38,17 @@ module MultiInsert
       self
     end
 
+    def on_conflict(column)
+      ::MultiInsert::Query::OnConflict.new(self, column)
+    end
+
+    def on_conflict_sql(sql)
+      @sql_on_conflict = sql
+      self
+    end
+
     def to_sql
-      sql = @sql_insert
-      sql = "#{sql} #{@sql_returning}" unless @sql_returning.nil?
-      sql
+      [@sql_insert, @sql_on_conflict, @sql_returning].reject(&:nil?).join(' ')
     end
 
     def to_s

data/lib/multi_insert/query_builder.rb

@@ -33,6 +33,41 @@ module MultiInsert
       "RETURNING #{columns}"
     end
 
+    def self.on_conflict(column)
+      column = ActiveRecord::Base.connection.quote_column_name(column.to_s)
+      "ON CONFLICT (#{column})"
+    end
+
+    def self.on_conflict_do_nothing(column)
+      "#{on_conflict(column)} DO NOTHING"
+    end
+
+    def self.on_conflict_do_update(column, values, opts = {})
+      opts = INSERT_DEFAULTS.merge(opts)
+      if values.is_a?(Symbol) || values.is_a?(String)
+        tmp = {}
+        tmp[values] = :excluded
+      elsif values.is_a?(Array)
+        values = values.product([:excluded]).to_h
+      end
+      if opts[:time]
+        now = Time.now.to_s(:db)
+        values[:updated_at] = now
+      end
+      values.map! do |key, value|
+        v = nil
+        key = ActiveRecord::Base.connection.quote_column_name(key)
+        if value == :excluded
+          v = "excluded.#{key}"
+        else
+          v = ActiveRecord::Base.connection.quote(value)
+        end
+        "#{key} = #{v}"
+      end
+      values = values.join(', ')
+      "#{on_conflict(column)} DO UPDATE SET #{values}"
+    end
+
     def self.join_params(params)
       "(" + params.join(',') + ")"
     end

data/lib/multi_insert/version.rb

@@ -1,3 +1,3 @@
 module MultiInsert
-  VERSION = "0.1.7"
+  VERSION = "0.2.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: multi_insert
 version: !ruby/object:Gem::Version
-  version: 0.1.7
+  version: 0.2.0
 platform: ruby
 authors:
 - Nax