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
59 60 61 62 63 |
# File 'lib/yaml_db/serialization_helper.rb', line 59 def self.load(io, truncate = true) ActiveRecord::Base.connection.transaction do load_documents(io, truncate) end end |
.load_records(table, column_names, records) ⇒ Object
82 83 84 85 86 87 88 89 90 91 92 93 |
# File 'lib/yaml_db/serialization_helper.rb', line 82 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 = 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) ⇒ Object
73 74 75 76 77 78 79 80 |
# File 'lib/yaml_db/serialization_helper.rb', line 73 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
95 96 97 98 99 |
# File 'lib/yaml_db/serialization_helper.rb', line 95 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
65 66 67 68 69 70 71 |
# File 'lib/yaml_db/serialization_helper.rb', line 65 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 |