module MultiInsert::QueryBuilder
Constants
- INSERT_DEFAULTS
Public Class Methods
insert(table, columns, values, opts = {})
click to toggle source
# File lib/multi_insert/query_builder.rb, line 9 def self.insert(table, columns, values, opts = {}) ar = ActiveRecord::Base.connection opts = INSERT_DEFAULTS.merge(opts) now = Time.now.to_s(:db) if opts[:time] table = ar.quote_table_name(table.to_s) # Format columns columns = columns + [:created_at, :updated_at] if opts[:time] columns = columns.map!{|c| ar.quote_column_name(c.to_s)} columns = join_params(columns) # Format values if opts[:time] values = values.map{|v| v + [now, now]} end values = values.map{|v| join_params(v.map{|vv| ar.quote(vv)})}.join(',') "INSERT INTO #{table} #{columns} VALUES #{values}" end
join_params(params)
click to toggle source
# File lib/multi_insert/query_builder.rb, line 76 def self.join_params(params) "(" + params.join(',') + ")" end
on_conflict(column)
click to toggle source
# File lib/multi_insert/query_builder.rb, line 36 def self.on_conflict(column) if column.nil? "ON CONFLICT" else column = [column] unless column.kind_of?(Array) columns = column.map{|c| ActiveRecord::Base.connection.quote_column_name(c.to_s)}.join(',') "ON CONFLICT (#{columns})" end end
on_conflict_do_nothing(column)
click to toggle source
# File lib/multi_insert/query_builder.rb, line 46 def self.on_conflict_do_nothing(column) "#{on_conflict(column)} DO NOTHING" end
on_conflict_do_update(column, values, opts = {})
click to toggle source
# File lib/multi_insert/query_builder.rb, line 50 def self.on_conflict_do_update(column, values, opts = {}) opts = INSERT_DEFAULTS.merge(opts) if values.is_a?(Symbol) || values.is_a?(String) values = { 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 arr = [] values.each do |key, value| v = nil key = ActiveRecord::Base.connection.quote_column_name(key.to_s) if value == :excluded v = "excluded.#{key}" else v = ActiveRecord::Base.connection.quote(value) end arr << "#{key} = #{v}" end values = arr.join(', ') "#{on_conflict(column)} DO UPDATE SET #{values}" end
returning(columns)
click to toggle source
# File lib/multi_insert/query_builder.rb, line 31 def self.returning(columns) columns = columns.map{|c| ActiveRecord::Base.connection.quote_column_name(c.to_s)}.join(',') "RETURNING #{columns}" end