Class: SerializationHelper::Base

Inherits:
Object
  • Object
show all
Defined in:
lib/serialization_helper.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(helper) ⇒ Base

Returns a new instance of Base.



6
7
8
9
10
 
# File 'lib/serialization_helper.rb', line 6

def initialize(helper)
  @dumper = helper.dumper
  @loader = helper.loader
  @extension = helper.extension
end

Instance Attribute Details

#extension ⇒ Object (readonly)

Returns the value of attribute extension.



4
5
6
 
# File 'lib/serialization_helper.rb', line 4

def extension
  @extension
end

Instance Method Details

#disable_logger ⇒ Object 



51
52
53
54
 
# File 'lib/serialization_helper.rb', line 51

def disable_logger
  @@old_logger = ActiveRecord::Base.logger
  ActiveRecord::Base.logger = nil
end

#dump(file) ⇒ Object 



12
13
14
15
16
17
18
19
20
21
 
# File 'lib/serialization_helper.rb', line 12

def dump(file)
  disable_logger
  unless File.exist?(file)
    full_path = (file).match(/^(.*\/)?(?:$|(.+?)(?:(\.[^.]*$)|$))/)[1]
    FileUtils.mkdir_p full_path
  end
  f = File.open(file, "w")
  @dumper.dump(f)
  reenable_logger
end

#dump_to_dir(dirname) ⇒ Object 



23
24
25
26
27
28
29
30
31
32
 
# File 'lib/serialization_helper.rb', line 23

def dump_to_dir(dirname)
  Dir.mkdir(dirname)
  tables = @dumper.tables
  tables.each do |table|
    io = File.new "#{dirname}/#{table}.#{@extension}", "w"
    @dumper.before_table(io, table)
    @dumper.dump_table io, table
    @dumper.after_table(io, table)
  end
end

#load(filename, truncate = false) ⇒ Object 



34
35
36
37
38
39
40
 
# File 'lib/serialization_helper.rb', line 34

def load(filename, truncate = false)
  disable_logger
  ActiveRecord::Base.connection.disable_referential_integrity do
    @loader.load(File.new(filename, "r"), truncate)
  end
  reenable_logger
end

#load_from_dir(dirname, truncate = true) ⇒ Object 



42
43
44
45
46
47
48
49
 
# File 'lib/serialization_helper.rb', line 42

def load_from_dir(dirname, truncate = true)
  Dir.entries(dirname).each do |filename|
    if filename =~ /^[.]/
      next
    end
    @loader.load(File.new("#{dirname}/#{filename}", "r"), truncate)
  end
end

#reenable_logger ⇒ Object 



56
57
58
 
# File 'lib/serialization_helper.rb', line 56

def reenable_logger
  ActiveRecord::Base.logger = @@old_logger
end