Class: Replicate::Dumper

Inherits:
Emitter show all
Defined in:
lib/replicate/dumper.rb

Overview

Dump replicants in a streaming fashion.

The Dumper takes objects and generates one or more replicant objects. A replicant has the form [type, id, attributes] and describes exactly one addressable record in a datastore. The type and id identify the model class name and model primary key id. The attributes Hash is a set of attribute name to primitively typed object value mappings.

Example dump session:

>> Replicate::Dumper.new do |dumper|
>>   dumper.marshal_to $stdout
>>   dumper.log_to $stderr
>>   dumper.dump User.find(1234)
>> end

Instance Method Summary collapse

Methods inherited from Emitter

#complete, #emit, #listen, #use

Constructor Details

#initialize(io = nil) ⇒ Dumper

Create a new Dumper.

io - IO object to write marshalled replicant objects to. block - Dump context block. If given, the end of the block’s execution

is assumed to be the end of the dump stream.



24
25
26
27
28
29
30
 
# File 'lib/replicate/dumper.rb', line 24

def initialize(io=nil)
  @memo = Hash.new { |hash,k| hash[k] = {} }
  super() do
    marshal_to io if io
    yield self if block_given?
  end
end

Instance Method Details

#dump(*objects) ⇒ Object

Dump one or more objects to the internal array or provided dump stream. This method guarantees that the same object will not be dumped more than once.

objects - ActiveRecord object instances.

Returns nothing.



59
60
61
62
63
64
65
66
67
68
69
 
# File 'lib/replicate/dumper.rb', line 59

def dump(*objects)
  objects = objects[0] if objects.size == 1 && objects[0].respond_to?(:to_ary)
  objects.each do |object|
    next if object.nil? || dumped?(object)
    if object.respond_to?(:dump_replicant)
      object.dump_replicant(self)
    else
      raise NoMethodError, "#{object.class} must respond to #dump_replicant"
    end
  end
end

#dumped?(object) ⇒ Boolean

Check if object has been written yet.

Returns:

  • (Boolean) 


72
73
74
75
76
77
78
79
80
81
 
# File 'lib/replicate/dumper.rb', line 72

def dumped?(object)
  if object.respond_to?(:replicant_id)
    type, id = object.replicant_id
  elsif object.is_a?(Array)
    type, id = object
  else
    return false
  end
  @memo[type.to_s][id]
end

#log_to(out = $stderr, verbose = false, quiet = false) ⇒ Object

Register a filter to write status information to the given stream. By default, a single line is used to report object counts while the dump is in progress; dump counts for each class are written when complete. The verbose and quiet options can be used to increase or decrease verbosity.

out - An IO object to write to, like stderr. verbose - Whether verbose output should be enabled. quiet - Whether quiet output should be enabled.

Returns the Replicate::Status object.



48
49
50
 
# File 'lib/replicate/dumper.rb', line 48

def log_to(out=$stderr, verbose=false, quiet=false)
  use Replicate::Status, 'dump', out, verbose, quiet
end

#marshal_to(io) ⇒ Object

Register a filter to write marshalled data to the given IO object.



33
34
35
 
# File 'lib/replicate/dumper.rb', line 33

def marshal_to(io)
  listen { |type, id, attrs, obj| Marshal.dump([type, id, attrs], io) }
end

#stats ⇒ Object

Retrieve dumped object counts for all classes.

Returns a Hash of { class_name => count } where count is the number of objects dumped with a class of class_name.



103
104
105
106
107
 
# File 'lib/replicate/dumper.rb', line 103

def stats
  stats = {}
  @memo.each { |class_name, items| stats[class_name] = items.size }
  stats
end

#write(type, id, attributes, object) ⇒ Object

Called exactly once per unique type and id. Emits to all listeners.

type - The model class name as a String. id - The record’s id. Usually an integer. attributes - All model attributes. object - The object this dump is generated for.

Returns the object.



91
92
93
94
95
96
97
 
# File 'lib/replicate/dumper.rb', line 91

def write(type, id, attributes, object)
  type = type.to_s
  return if dumped?([type, id])
  @memo[type][id] = true

  emit type, id, attributes, object
end