Module: ActiveRecord::Import::MysqlAdapter::InstanceMethods

Included in:
ConnectionAdapters::Mysql2Adapter, ConnectionAdapters::MysqlAdapter
Defined in:
lib/activerecord-import/adapters/mysql_adapter.rb

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.included(klass) ⇒ Object 



3
4
5
6
7
8
 
# File 'lib/activerecord-import/adapters/mysql_adapter.rb', line 3

def self.included(klass)
  klass.instance_eval do
    include ActiveRecord::Import::ImportSupport  
    include ActiveRecord::Import::OnDuplicateKeyUpdateSupport
  end
end

Instance Method Details

#duplicate_key_update_error?(exception) ⇒ Boolean

return true if the statement is a duplicate key record error

Returns:

  • (Boolean) 


55
56
57
 
# File 'lib/activerecord-import/adapters/mysql_adapter.rb', line 55

def duplicate_key_update_error?(exception)# :nodoc:
  exception.is_a?(ActiveRecord::StatementInvalid) && exception.to_s.include?('Duplicate entry')
end

#max_allowed_packet ⇒ Object

Returns the maximum number of bytes that the server will allow in a single packet



12
13
14
15
16
17
 
# File 'lib/activerecord-import/adapters/mysql_adapter.rb', line 12

def max_allowed_packet # :nodoc:
  result = execute( "SHOW VARIABLES like 'max_allowed_packet';" )
  # original Mysql gem responds to #fetch_row while Mysql2 responds to #first
  val = result.respond_to?(:fetch_row) ? result.fetch_row[1] : result.first[1]
  val.to_i
end

#sql_for_on_duplicate_key_update(table_name, *args) ⇒ Object

Returns a generated ON DUPLICATE KEY UPDATE statement given the passed in args.



21
22
23
24
25
26
27
28
29
30
31
32
33
34
 
# File 'lib/activerecord-import/adapters/mysql_adapter.rb', line 21

def sql_for_on_duplicate_key_update( table_name, *args ) # :nodoc:
  sql = ' ON DUPLICATE KEY UPDATE '
  arg = args.first
  if arg.is_a?( Array )
    sql << sql_for_on_duplicate_key_update_as_array( table_name, arg )
  elsif arg.is_a?( Hash )
    sql << sql_for_on_duplicate_key_update_as_hash( table_name, arg )
  elsif arg.is_a?( String )
    sql << arg
  else
    raise ArgumentError.new( "Expected Array or Hash" )
  end
  sql
end

#sql_for_on_duplicate_key_update_as_array(table_name, arr) ⇒ Object

:nodoc:



36
37
38
39
40
41
42
 
# File 'lib/activerecord-import/adapters/mysql_adapter.rb', line 36

def sql_for_on_duplicate_key_update_as_array( table_name, arr )  # :nodoc:
  results = arr.map do |column|
    qc = quote_column_name( column )
    "#{table_name}.#{qc}=VALUES(#{qc})"        
  end
  results.join( ',' )
end

#sql_for_on_duplicate_key_update_as_hash(table_name, hsh) ⇒ Object

:nodoc:



44
45
46
47
48
49
50
51
52
 
# File 'lib/activerecord-import/adapters/mysql_adapter.rb', line 44

def sql_for_on_duplicate_key_update_as_hash( table_name, hsh ) # :nodoc:
  sql = ' ON DUPLICATE KEY UPDATE '
  results = hsh.map do |column1, column2|
    qc1 = quote_column_name( column1 )
    qc2 = quote_column_name( column2 )
    "#{table_name}.#{qc1}=VALUES( #{qc2} )"
  end
  results.join( ',')
end