Class: Psych::Visitors::ToRuby

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

Instance Method Summary collapse

Instance Method Details

#resolve_class_with_constantize(klass_name) ⇒ Object 



119
120
121
122
123
 
# File 'lib/queue_dispatcher/psych_ext.rb', line 119

def resolve_class_with_constantize(klass_name)
  klass_name.constantize
rescue
  resolve_class_without_constantize(klass_name)
end

#visit_Psych_Nodes_Mapping_with_class(object) ⇒ Object 



82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
 
# File 'lib/queue_dispatcher/psych_ext.rb', line 82

def visit_Psych_Nodes_Mapping_with_class(object)
  return revive(Psych.load_tags[object.tag], object) if Psych.load_tags[object.tag]

  case object.tag
  when /^!ruby\/ActiveRecord:(.+)$/
    klass = resolve_class($1)
    payload = Hash[*object.children.map { |c| accept c }]
    id = payload["attributes"][klass.primary_key]
    begin
      if ActiveRecord::VERSION::MAJOR == 3
        klass.unscoped.find(id)
      else # Rails 2
        klass.with_exclusive_scope { klass.find(id) }
      end
    rescue ActiveRecord::RecordNotFound
      # raise QueueDispatcher::DeserializationError

      # Return ActiveRecord without reload
      revive(klass, object)
    end
  when /^!ruby\/Mongoid:(.+)$/
    klass = resolve_class($1)
    payload = Hash[*object.children.map { |c| accept c }]
    begin
      klass.find(payload["attributes"]["_id"])
    rescue Mongoid::Errors::DocumentNotFound
      # raise QueueDispatcher::DeserializationError

      # Return ActiveRecord without reload
      revive(klass, object)
    end
  else
    visit_Psych_Nodes_Mapping_without_class(object)
  end
end

#visit_Psych_Nodes_Scalar(o) ⇒ Object 



20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
 
# File 'lib/queue_dispatcher/psych_ext.rb', line 20

def visit_Psych_Nodes_Scalar(o)
  @st[o.anchor] = o.value if o.anchor

  if klass = Psych.load_tags[o.tag]
    instance = klass.allocate

    if instance.respond_to?(:init_with)
      coder = Psych::Coder.new(o.tag)
      coder.scalar = o.value
      instance.init_with coder
    end

    return instance
  end

  return o.value if o.quoted
  return @ss.tokenize(o.value) unless o.tag

  case o.tag
  when '!binary', 'tag:yaml.org,2002:binary'
    o.value.unpack('m').first
  when '!str', 'tag:yaml.org,2002:str'
    o.value
  when "!ruby/object:DateTime"
    require 'date'
    @ss.parse_time(o.value).to_datetime
  when "!ruby/object:Complex"
    Complex(o.value)
  when "!ruby/object:Rational"
    Rational(o.value)
  when "!ruby/class", "!ruby/module"
    resolve_class o.value
  when "tag:yaml.org,2002:float", "!float"
    Float(@ss.tokenize(o.value))
  when "!ruby/regexp"
    o.value =~ /^\/(.*)\/([mixn]*)$/
    source  = $1
    options = 0
    lang    = nil
    ($2 || '').split('').each do |option|
      case option
      when 'x' then options |= Regexp::EXTENDED
      when 'i' then options |= Regexp::IGNORECASE
      when 'm' then options |= Regexp::MULTILINE
      when 'n' then options |= Regexp::NOENCODING
      else lang = option
      end
    end
    Regexp.new(*[source, options, lang].compact)
  when "!ruby/range"
    args = o.value.split(/([.]{2,3})/, 2).map { |s|
      accept Nodes::Scalar.new(s)
    }
    args.push(args.delete_at(1) == '...')
    Range.new(*args)
  when /^!ruby\/sym(bol)?:?(.*)?$/
    o.value.to_sym
  else
    @ss.tokenize o.value
  end
end