class SerializationHelper::Load
Public Class Methods
load(io, truncate = true)
click to toggle source
# File lib/serialization_helper.rb, line 65 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 88 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(parse_erb(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 79 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
parse_erb(input)
click to toggle source
# File lib/serialization_helper.rb, line 61 def self.parse_erb(input) ERB.new(input).result rescue input end
reset_pk_sequence!(table_name)
click to toggle source
# File lib/serialization_helper.rb, line 101 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 71 def self.truncate_table(table) begin ActiveRecord::Base.connection.execute("TRUNCATE #{SerializationHelper::Utils.quote_table(table)}") rescue Exception ActiveRecord::Base.connection.execute("DELETE FROM #{SerializationHelper::Utils.quote_table(table)}") end end