Class: YamlDb::SerializationHelper::Load
- Inherits:
-
Object
- Object
- YamlDb::SerializationHelper::Load
- Defined in:
- lib/yaml_db/serialization_helper.rb
Direct Known Subclasses
Class Method Summary collapse
- .load(io, truncate = true) ⇒ Object
- .load_records(table, column_names, records) ⇒ Object
- .load_table(table, data, truncate = true) ⇒ Object
- .reset_pk_sequence!(table_name) ⇒ Object
- .truncate_table(table) ⇒ Object
Class Method Details
.load(io, truncate = true) ⇒ Object
61 62 63 64 65 |
# File 'lib/yaml_db/serialization_helper.rb', line 61 def self.load(io, truncate = true) ActiveRecord::Base.connection.transaction do load_documents(io, truncate) end end |
.load_records(table, column_names, records) ⇒ Object
84 85 86 87 88 89 90 91 92 93 94 |
# File 'lib/yaml_db/serialization_helper.rb', line 84 def self.load_records(table, column_names, records) if column_names.nil? return end quoted_column_names = column_names.map { |column| ActiveRecord::Base.connection.quote_column_name(column) }.join(',') quoted_table_name = Utils.quote_table(table) records.each do |record| quoted_values = record.map{|c| ActiveRecord::Base.connection.quote(c)}.join(',') ActiveRecord::Base.connection.execute("INSERT INTO #{quoted_table_name} (#{quoted_column_names}) VALUES (#{quoted_values})") end end |
.load_table(table, data, truncate = true) ⇒ Object
75 76 77 78 79 80 81 82 |
# File 'lib/yaml_db/serialization_helper.rb', line 75 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) ⇒ Object
96 97 98 99 100 |
# File 'lib/yaml_db/serialization_helper.rb', line 96 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) ⇒ Object
67 68 69 70 71 72 73 |
# File 'lib/yaml_db/serialization_helper.rb', line 67 def self.truncate_table(table) begin ActiveRecord::Base.connection.execute("TRUNCATE #{Utils.quote_table(table)}") rescue Exception ActiveRecord::Base.connection.execute("DELETE FROM #{Utils.quote_table(table)}") end end |