1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
# frozen_string_literal: true
module ActiveRecord::Import::AbstractAdapter
module InstanceMethods
def next_value_for_sequence(sequence_name)
%(#{sequence_name}.nextval)
end
def insert_many( sql, values, _options = {}, *args ) # :nodoc:
number_of_inserts = 1
base_sql, post_sql = if sql.is_a?( String )
[sql, '']
elsif sql.is_a?( Array )
[sql.shift, sql.join( ' ' )]
end
sql2insert = base_sql + values.join( ',' ) + post_sql
insert( sql2insert, *args )
ActiveRecord::Import::Result.new([], number_of_inserts, [], [])
end
def pre_sql_statements(options)
sql = []
sql << options[:pre_sql] if options[:pre_sql]
sql << options[:command] if options[:command]
# add keywords like IGNORE or DELAYED
if options[:keywords].is_a?(Array)
sql.concat(options[:keywords])
elsif options[:keywords]
sql << options[:keywords].to_s
end
sql
end
# Synchronizes the passed in ActiveRecord instances with the records in
# the database by calling +reload+ on each instance.
def after_import_synchronize( instances )
instances.each(&:reload)
end
# Returns an array of post SQL statements given the passed in options.
def post_sql_statements( table_name, options ) # :nodoc:
post_sql_statements = []
if supports_on_duplicate_key_update? && options[:on_duplicate_key_update]
post_sql_statements << sql_for_on_duplicate_key_update( table_name, options[:on_duplicate_key_update], options[:primary_key], options[:locking_column] )
elsif logger && options[:on_duplicate_key_update]
logger.warn "Ignoring on_duplicate_key_update because it is not supported by the database."
end
# custom user post_sql
post_sql_statements << options[:post_sql] if options[:post_sql]
# with rollup
post_sql_statements << rollup_sql if options[:rollup]
post_sql_statements
end
def increment_locking_column!(table_name, results, locking_column)
if locking_column.present?
results << "\"#{locking_column}\"=#{table_name}.\"#{locking_column}\"+1"
end
end
def supports_on_duplicate_key_update?
false
end
end
end
|
