Class: Simple::SQL::Inserter

Inherits:
Object
  • Object
show all
Defined in:
lib/simple/sql/insert.rb

Constant Summary collapse

SQL = 
::Simple::SQL
CONFICT_HANDLING = 
{
  nil      => "",
  :nothing => "ON CONFLICT DO NOTHING",
  :ignore  => "ON CONFLICT DO NOTHING"
}
@@inserters = 
{}

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(table_name:, columns:, on_conflict:, into:) ⇒ Inserter

  • table_name - the name of the table

  • columns - name of columns, as Array or Array

Raises:

  • (ArgumentError) 


42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
 
# File 'lib/simple/sql/insert.rb', line 42

def initialize(table_name:, columns:, on_conflict:, into:)
  raise ArgumentError, "Cannot insert a record without attributes" if columns.empty?

  @columns = columns
  @into = into

  cols = []
  vals = []

  cols += columns
  vals += columns.each_with_index.map { |_, idx| "$#{idx + 1}" }

  timestamp_columns = SQL::Reflection.timestamp_columns(table_name) - columns.map(&:to_s)

  cols += timestamp_columns
  vals += timestamp_columns.map { "now()" }

  returning = into ? "*" : "id"

  @sql = "INSERT INTO #{table_name} (#{cols.join(',')}) VALUES(#{vals.join(',')}) #{confict_handling(on_conflict)} RETURNING #{returning}"
end

Class Method Details

.create(table_name:, columns:, on_conflict:, into:) ⇒ Object 



32
33
34
35
36
 
# File 'lib/simple/sql/insert.rb', line 32

def self.create(table_name:, columns:, on_conflict:, into:)
  expect! on_conflict => CONFICT_HANDLING.keys

  @@inserters[[table_name, columns, on_conflict, into]] ||= new(table_name: table_name, columns: columns, on_conflict: on_conflict, into: into)
end

Instance Method Details

#confict_handling(on_conflict) ⇒ Object 



70
71
72
73
74
 
# File 'lib/simple/sql/insert.rb', line 70

def confict_handling(on_conflict)
  CONFICT_HANDLING.fetch(on_conflict) do
    raise(ArgumentError, "Invalid on_conflict value #{on_conflict.inspect}")
  end
end

#insert(records:) ⇒ Object 



76
77
78
79
80
81
82
 
# File 'lib/simple/sql/insert.rb', line 76

def insert(records:)
  SQL.transaction do
    records.map do |record|
      SQL.ask @sql, *record.values_at(*@columns), into: @into
    end
  end
end