class SerializationHelper::Load
Public Class Methods
load(io, truncate = true)
click to toggle source
# File lib/serialization_helper.rb, line 62 def self.load(io, truncate = true) ActiveRecord::Base.connection.transaction do load_documents(io, truncate) end end
load_records(table, column_names, records)
click to toggle source
# File lib/serialization_helper.rb, line 87 def self.load_records(table, column_names, records) if column_names.nil? return end columns = column_names.map{|cn| ActiveRecord::Base.connection.columns(table).detect{|c| c.name == cn}} quoted_column_names = column_names.map { |column| ActiveRecord::Base.connection.quote_column_name(column) }.join(',') quoted_table_name = SerializationHelper::Utils.quote_table(table) records.each do |record| quoted_values = record.zip(columns).map{|c| ActiveRecord::Base.connection.quote(c.first, c.last)}.join(',') ActiveRecord::Base.connection.execute("INSERT INTO #{quoted_table_name} (#{quoted_column_names}) VALUES (#{quoted_values})") end end
load_table(table, data, truncate = true)
click to toggle source
# File lib/serialization_helper.rb, line 78 def self.load_table(table, data, truncate = true) column_names = data['columns'] if truncate truncate_table(table) end load_records(table, column_names, data['records']) reset_pk_sequence!(table) end
reset_pk_sequence!(table_name)
click to toggle source
# File lib/serialization_helper.rb, line 100 def self.reset_pk_sequence!(table_name) if ActiveRecord::Base.connection.respond_to?(:reset_pk_sequence!) ActiveRecord::Base.connection.reset_pk_sequence!(table_name) end end
truncate_table(table)
click to toggle source
# File lib/serialization_helper.rb, line 68 def self.truncate_table(table) begin ActiveRecord::Base.connection.execute("SAVEPOINT before_truncation") ActiveRecord::Base.connection.execute("TRUNCATE #{SerializationHelper::Utils.quote_table(table)} CASCADE") rescue Exception ActiveRecord::Base.connection.execute("ROLLBACK TO SAVEPOINT before_truncation") ActiveRecord::Base.connection.execute("DELETE FROM #{SerializationHelper::Utils.quote_table(table)}") end end