Ensuring instances_within does not return duplicate objects

archan937 · Jun 10, 2012 · 689ce37 · 689ce37
1 parent ea1000e
commit 689ce37
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/lib/mass.rb b/lib/mass.rb
@@ -162,14 +162,21 @@ def classes_within(mods, initial_array = nil)
   def instances_within(*mods)
     GC.start
     [].tap do |array|
+      object_ids = []
       if mods.empty?
         ObjectSpace.each_object do |object|
-          array << object
+          unless object_ids.include? object.object_id
+            array << object
+            object_ids << object.object_id
+          end
         end
       else
         classes_within(mods).each do |klass|
           ObjectSpace.each_object(klass) do |object|
-            array << object
+            unless object_ids.include? object.object_id
+              array << object
+              object_ids << object.object_id
+            end
           end
         end
       end