Class: Psych::Visitors::ToRuby

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

Instance Method Summary collapse

Instance Method Details

#resolve_class_with_constantize(klass_name) ⇒ Object



138
139
140
141
142
# File 'lib/delayed/psych_ext.rb', line 138

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



101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
# File 'lib/delayed/psych_ext.rb', line 101

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
      klass.unscoped.find(id)
    rescue ActiveRecord::RecordNotFound
      raise Delayed::DeserializationError
    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 Delayed::DeserializationError
    end
  when /^!ruby\/DataMapper:(.+)$/
    klass = resolve_class($1)
    payload = Hash[*object.children.map { |c| accept c }]
    begin
      primary_keys = klass.properties.select { |p| p.key? }
      key_names = primary_keys.map { |p| p.name.to_s }
      klass.get!(*key_names.map { |k| payload["attributes"][k] })
    rescue DataMapper::ObjectNotFoundError
      raise Delayed::DeserializationError
    end
  else
    visit_Psych_Nodes_Mapping_without_class(object)
  end
end

#visit_Psych_Nodes_Scalar(o) ⇒ Object



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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
# File 'lib/delayed/psych_ext.rb', line 39

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