Class: Delayed::PsychExt::ToRuby

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

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.create ⇒ Object 



48
49
50
 
# File 'lib/delayed/psych_ext.rb', line 48

def self.create
  new
end

Instance Method Details

#accept(target) ⇒ Object 



53
54
55
56
57
 
# File 'lib/delayed/psych_ext.rb', line 53

def accept(target)
  super.tap do |value|
    register(target, value) if value.class.include?(Singleton)
  end
end

#jruby_is_seriously_borked ⇒ Object

defined? is triggering something really messed up in jruby causing both the if AND else clauses to execute, however if the check is run here, everything is fine



119
120
121
 
# File 'lib/delayed/psych_ext.rb', line 119

def jruby_is_seriously_borked
  defined?(ActiveRecord::Base)
end

#resolve_class(klass_name) ⇒ Object 



123
124
125
126
127
128
129
 
# File 'lib/delayed/psych_ext.rb', line 123

def resolve_class(klass_name)
  return nil if klass_name.blank?

  klass_name.constantize
rescue StandardError
  super
end

#revive(klass, node) ⇒ Object 



131
132
133
 
# File 'lib/delayed/psych_ext.rb', line 131

def revive(klass, node)
  klass.include?(Singleton) ? klass.instance : super
end

#visit_Psych_Nodes_Mapping(object) ⇒ Object

rubocop:disable Metrics/CyclomaticComplexity, Naming/MethodName, Metrics/PerceivedComplexity



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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
 
# File 'lib/delayed/psych_ext.rb', line 59

def visit_Psych_Nodes_Mapping(object) # rubocop:disable Metrics/CyclomaticComplexity, Naming/MethodName, Metrics/PerceivedComplexity
  klass = Psych.load_tags[object.tag]
  if klass
    # Implementation changed here https://github.com/ruby/psych/commit/2c644e184192975b261a81f486a04defa3172b3f
    # load_tags used to have class values, now the values are strings
    klass = resolve_class(klass) if klass.is_a?(String)
    return revive(klass, object)
  end

  case object.tag
    when %r{^!ruby/object}
      result = super
      if jruby_is_seriously_borked && result.is_a?(ActiveRecord::Base)
        klass = result.class
        id = result[klass.primary_key]
        begin
          klass.unscoped.find(id)
        rescue ActiveRecord::RecordNotFound => e
          raise Delayed::DeserializationError, "ActiveRecord::RecordNotFound, class: #{klass}, primary key: #{id} (#{e.message})"
        end
      else
        result
      end
    when %r{^!ruby/ActiveRecord:(.+)$}
      klass = resolve_class(Regexp.last_match[1])
      payload = Hash[*object.children.map { |c| accept c }]
      id = payload['attributes'][klass.primary_key]
      id = id.value if defined?(ActiveRecord::Attribute) && id.is_a?(ActiveRecord::Attribute)
      begin
        klass.unscoped.find(id)
      rescue ActiveRecord::RecordNotFound => e
        raise Delayed::DeserializationError, "ActiveRecord::RecordNotFound, class: #{klass}, primary key: #{id} (#{e.message})"
      end
    when %r{^!ruby/Mongoid:(.+)$}
      klass = resolve_class(Regexp.last_match[1])
      payload = Hash[*object.children.map { |c| accept c }]
      id = payload['attributes']['_id']
      begin
        klass.find(id)
      rescue Mongoid::Errors::DocumentNotFound => e
        raise Delayed::DeserializationError, "Mongoid::Errors::DocumentNotFound, class: #{klass}, primary key: #{id} (#{e.message})"
      end
    when %r{^!ruby/DataMapper:(.+)$}
      klass = resolve_class(Regexp.last_match[1])
      payload = Hash[*object.children.map { |c| accept c }]
      begin
        primary_keys = klass.properties.select(&:key?)
        key_names = primary_keys.map { |p| p.name.to_s }
        klass.get!(*key_names.map { |k| payload['attributes'][k] })
      rescue DataMapper::ObjectNotFoundError => e
        raise Delayed::DeserializationError, "DataMapper::ObjectNotFoundError, class: #{klass} (#{e.message})"
      end
    else
      super
  end
end