Resolve class and find or raise for Mongoid objects.

zxcvbnm4709 · Oct 24, 2011 · ef5f7e4 · ef5f7e4
1 parent fdd72bd
commit ef5f7e4
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/lib/delayed/psych_ext.rb b/lib/delayed/psych_ext.rb
@@ -49,6 +49,14 @@ def visit_Psych_Nodes_Mapping_with_class(object)
           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
         else
           visit_Psych_Nodes_Mapping_without_class(object)
         end