forked from adamwiggins/yaml_db
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Browse files
Browse the repository at this point in the history
moved serializer into separate file
- Loading branch information
Showing
6 changed files
with
130 additions
and
75 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,24 @@ | ||
module CsvDb | ||
module Helper | ||
def self.loader | ||
Load | ||
end | ||
|
||
def self.dumper | ||
Dump | ||
end | ||
|
||
def self.extension | ||
"csv" | ||
end | ||
end | ||
|
||
module Load | ||
|
||
end | ||
|
||
module Dump | ||
end | ||
|
||
|
||
end |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,39 +1,68 @@ | ||
class SerializationHelper | ||
attr_reader :extension | ||
attr_reader :extension | ||
|
||
def initialize(helper) | ||
@dumper = helper.dumper | ||
@loader = helper.loader | ||
@extension = helper.extension | ||
def initialize(helper) | ||
@dumper = helper.dumper | ||
@loader = helper.loader | ||
@extension = helper.extension | ||
end | ||
|
||
def dump(filename) | ||
disable_logger | ||
@dumper.dump(File.new(filename, "w")) | ||
reenable_logger | ||
end | ||
|
||
def dump_to_dir(dirname) | ||
Dir.mkdir(dirname) | ||
tables = @dumper.tables | ||
tables.each do |table| | ||
file = File.new "#{dirname}/#{table}.#{@extension}", "w" | ||
@dumper.dump_table file, table | ||
end | ||
end | ||
|
||
def load(filename) | ||
disable_logger | ||
@loader.load(File.new(filename, "r")) | ||
reenable_logger | ||
end | ||
|
||
def dump(filename) | ||
disable_logger | ||
@dumper.dump(File.new(filename, "w")) | ||
reenable_logger | ||
end | ||
def disable_logger | ||
@@old_logger = ActiveRecord::Base.logger | ||
ActiveRecord::Base.logger = nil | ||
end | ||
|
||
def dump_to_dir(dirname) | ||
Dir.mkdir(dirname) | ||
tables = @dumper.tables | ||
def reenable_logger | ||
ActiveRecord::Base.logger = @@old_logger | ||
end | ||
|
||
class Load | ||
|
||
end | ||
|
||
class Dump | ||
def self.dump(io) | ||
tables.each do |table| | ||
file = File.new "#{dirname}/#{table}.#{@extension}", "w" | ||
@dumper.dump_table file, table | ||
dump_table(io, table) | ||
end | ||
end | ||
|
||
def load(filename) | ||
disable_logger | ||
@loader.load(File.new(filename, "r")) | ||
reenable_logger | ||
end | ||
def self.tables | ||
ActiveRecord::Base.connection.tables.reject { |table| ['schema_info', 'schema_migrations'].include?(table) } | ||
end | ||
|
||
def self.dump_table(io, table) | ||
return if table_record_count(table).zero? | ||
|
||
dump_table_columns(io, table) | ||
dump_table_records(io, table) | ||
end | ||
|
||
def self.table_column_names(table) | ||
ActiveRecord::Base.connection.columns(table).map { |c| c.name } | ||
end | ||
|
||
def disable_logger | ||
@@old_logger = ActiveRecord::Base.logger | ||
ActiveRecord::Base.logger = nil | ||
end | ||
end | ||
|
||
def reenable_logger | ||
ActiveRecord::Base.logger = @@old_logger | ||
end | ||
end |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,42 @@ | ||
require File.dirname(__FILE__) + '/base' | ||
|
||
describe SerializationHelper do | ||
|
||
before do | ||
ActiveRecord::Base = mock('ActiveRecord::Base', :null_object => true) | ||
ActiveRecord::Base.connection = mock('connection') | ||
ActiveRecord::Base.connection.stub!(:tables).and_return([ 'mytable', 'schema_info', 'schema_migrations' ]) | ||
end | ||
|
||
def stub_helper! | ||
@helper = mock("MyHelper") | ||
@helper.stub!(:dumper).and_return(@dumper) | ||
@helper.stub!(:loader) | ||
@helper.stub!(:extension).and_return("yml") | ||
@dumper.stub!(:tables).and_return([ActiveRecord::Base.connection.tables[0]]) | ||
end | ||
|
||
context "for multi-file dumps" do | ||
before do | ||
@io = StringIO.new | ||
File.should_receive(:new).once.with("dir_name/mytable.yml", "w").and_return(@io) | ||
Dir.should_receive(:mkdir).once.with("dir_name") | ||
stub_helper! | ||
@dumper.should_receive(:dump_table).once.with(@io, "mytable") | ||
end | ||
|
||
it "should create the number of files that there are tables" do | ||
SerializationHelper.new(@helper).dump_to_dir "dir_name" | ||
end | ||
|
||
end | ||
|
||
context "for multi-file loads" do | ||
before do | ||
|
||
end | ||
|
||
|
||
end | ||
|
||
end |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters