Sha256: 761c7e8759a03cf4458205b4c6ae74b40cd4c0225e14a947b5ab8b6316513a04
Contents?: true
Size: 1.75 KB
Versions: 20
Compression:
Stored size: 1.75 KB
Contents
require 'rom/sql/commands/error_wrapper'
require 'rom/sql/commands/transaction'
module ROM
module SQL
module Commands
# SQL create command
#
# @api public
class Create < ROM::Commands::Create
adapter :sql
include Transaction
include ErrorWrapper
use :associates
use :schema
after :finalize
# Inserts provided tuples into the database table
#
# @api public
def execute(tuples)
insert_tuples = with_input_tuples(tuples) do |tuple|
attributes = input[tuple]
attributes.to_h
end
if insert_tuples.length > 1
multi_insert(insert_tuples)
else
insert(insert_tuples)
end
end
private
# @api private
def finalize(tuples, *)
tuples.map { |t| relation.output_schema[t] }
end
# Executes insert statement and returns inserted tuples
#
# @api private
def insert(tuples)
pks = tuples.map { |tuple| relation.insert(tuple) }
relation.where(relation.primary_key => pks).to_a
end
# Executes multi_insert statement and returns inserted tuples
#
# @api private
def multi_insert(tuples)
pks = relation.multi_insert(tuples, return: :primary_key)
relation.where(relation.primary_key => pks).to_a
end
# Yields tuples for insertion or return an enumerator
#
# @api private
def with_input_tuples(tuples)
input_tuples = Array([tuples]).flatten(1).map
return input_tuples unless block_given?
input_tuples.each { |tuple| yield(tuple) }
end
end
end
end
end
Version data entries
20 entries across 20 versions & 1 rubygems