track mongoid models

Signed-off-by: David A. Cuadrado <krawek@gmail.com>
dcu · Oct 5, 2011 · 5052d72 · 5052d72
1 parent 9b9a134
commit 5052d72
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/lib/mongoid_ext/patches.rb b/lib/mongoid_ext/patches.rb
@@ -14,4 +14,17 @@ def nolock_eval(code, *args)
       doc['retval']
     end
   end
-end
+end
+
+module Mongoid
+  module Document
+    def self.included(base)
+      models << base.to_s
+      super
+    end
+
+    def self.models
+      @models ||= []
+    end
+  end
+end