Class: Perpetuity::MongoDB::Serializer

Inherits:
Object
  • Object
show all
Includes:
DataInjectable
Defined in:
lib/perpetuity/mongodb/serializer.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Methods included from DataInjectable

#give_id_to, #inject_attribute, #inject_data

Constructor Details

#initialize(mapper) ⇒ Serializer

Returns a new instance of Serializer.



10
11
12
13
14
 
# File 'lib/perpetuity/mongodb/serializer.rb', line 10

def initialize(mapper)
  @mapper = mapper
  @class = mapper.mapped_class
  @mapper_registry = mapper.mapper_registry
end

Instance Attribute Details

#mapper ⇒ Object (readonly)

Returns the value of attribute mapper.



8
9
10
 
# File 'lib/perpetuity/mongodb/serializer.rb', line 8

def mapper
  @mapper
end

#mapper_registry ⇒ Object (readonly)

Returns the value of attribute mapper_registry.



8
9
10
 
# File 'lib/perpetuity/mongodb/serializer.rb', line 8

def mapper_registry
  @mapper_registry
end

Instance Method Details

#attribute_for(object, attribute_name) ⇒ Object 



16
17
18
 
# File 'lib/perpetuity/mongodb/serializer.rb', line 16

def attribute_for object, attribute_name
  object.instance_variable_get("@#{attribute_name}")
end

#serialize(object) ⇒ Object 



20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
 
# File 'lib/perpetuity/mongodb/serializer.rb', line 20

def serialize object
  attrs = mapper.class.attribute_set.map do |attrib|
    value = attribute_for object, attrib.name

    serialized_value = if value.is_a? Reference
      serialize_reference value
    elsif value.is_a? Array
      serialize_array(value, attrib.embedded?)
    elsif mapper.data_source.can_serialize? value
      value
    elsif mapper_registry.has_mapper?(value.class)
      serialize_with_foreign_mapper(value, attrib.embedded?)
    else
      Marshal.dump(value)
    end

    [attrib.name.to_s, serialized_value]
  end

  Hash[attrs]
end

#serialize_array(enum, embedded) ⇒ Object 



109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
 
# File 'lib/perpetuity/mongodb/serializer.rb', line 109

def serialize_array enum, embedded
  enum.map do |value|
    if value.is_a? Reference
      serialize_reference value
    elsif value.is_a? Array
      serialize_array(value)
    elsif mapper.data_source.can_serialize? value
      value
    elsif mapper_registry.has_mapper?(value.class)
      if embedded
        {
          '__metadata__' => {
            'class' => value.class.to_s
          }
        }.merge mapper_registry[value.class].serialize(value)
      else
        serialize_reference value
      end
    else
      Marshal.dump(value)
    end
  end
end

#serialize_reference(value) ⇒ Object 



133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
 
# File 'lib/perpetuity/mongodb/serializer.rb', line 133

def serialize_reference value
  if value.is_a? Reference
    reference = value
  else
    unless value.is_a? PersistedObject
      mapper_registry[value.class].insert value
    end
    reference = Reference.new(value.class.to_s, value.id)
  end
  {
    '__metadata__' => {
      'class' => reference.klass.to_s,
      'id'    => reference.id
    }
  }
end

#serialize_with_foreign_mapper(value, embedded = false) ⇒ Object 



98
99
100
101
102
103
104
105
106
107
 
# File 'lib/perpetuity/mongodb/serializer.rb', line 98

def serialize_with_foreign_mapper value, embedded = false
  if embedded
    value_mapper = mapper_registry[value.class]
    value_serializer = Serializer.new(value_mapper)
    attr = value_serializer.serialize(value)
    attr.merge '__metadata__' =>  { 'class' => value.class }
  else
    serialize_reference(value)
  end
end

#unserialize(data) ⇒ Object 



42
43
44
45
46
47
48
49
50
 
# File 'lib/perpetuity/mongodb/serializer.rb', line 42

def unserialize data
  if data.is_a? Array
    unserialize_object_array data
  else
    object = unserialize_object(data)
    give_id_to object
    object
  end
end

#unserialize_attribute(data) ⇒ Object 



72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
 
# File 'lib/perpetuity/mongodb/serializer.rb', line 72

def unserialize_attribute data
  if data.is_a?(String) && data.start_with?("\u0004") # if it's marshaled
    Marshal.load(data)
  elsif data.is_a? Array
    data.map { |i| unserialize_attribute i }
  elsif data.is_a? Hash
    metadata = data.delete('__metadata__')
    if metadata
      klass = metadata['class'].split('::').inject(Kernel) do |scope, const_name|
        scope.const_get(const_name)
      end
      id = metadata['id']

      if id
        object = Reference.new(klass, id)
      else
        object = unserialize_object(data, klass)
      end
    else
      data
    end
  else
    data
  end
end

#unserialize_object(data, klass = @class) ⇒ Object 



52
53
54
55
56
57
58
59
60
61
62
 
# File 'lib/perpetuity/mongodb/serializer.rb', line 52

def unserialize_object data, klass=@class
  if data.is_a? Hash
    object = klass.allocate
    data.each do |attr, value|
      inject_attribute object, attr, unserialize_attribute(value)
    end
    object
  else
    unserialize_attribute data
  end
end

#unserialize_object_array(objects) ⇒ Object 



64
65
66
67
68
69
70
 
# File 'lib/perpetuity/mongodb/serializer.rb', line 64

def unserialize_object_array objects
  objects.map do |data|
    object = unserialize_object data
    give_id_to object
    object
  end
end