[Truffle] Do not visit singleton classes with ObjectSpace.each_object.

eregon · eregon · commit b3bfc9d38833 · 2015-01-29T14:54:50.000+01:00
diff --git a/truffle/src/main/java/org/jruby/truffle/runtime/core/RubyClass.java b/truffle/src/main/java/org/jruby/truffle/runtime/core/RubyClass.java
@@ -12,9 +12,11 @@
 import com.oracle.truffle.api.CompilerAsserts;
 import com.oracle.truffle.api.CompilerDirectives.CompilationFinal;
 import com.oracle.truffle.api.nodes.Node;
+
 import org.jruby.truffle.nodes.RubyNode;
 import org.jruby.truffle.nodes.objects.Allocator;
 import org.jruby.truffle.runtime.RubyContext;
+import org.jruby.truffle.runtime.subsystems.ObjectSpaceManager.ObjectGraphVisitor;
 
 import java.util.Collections;
 import java.util.Set;
@@ -149,6 +151,14 @@ public Allocator getAllocator() {
         return allocator;
     }
 
+    @Override
+    public void visitObjectGraph(ObjectGraphVisitor visitor) {
+        // MRI consider all singleton classes as internal objects.
+        if (!isSingleton) {
+            super.visitObjectGraph(visitor);
+        }
+    }
+
     public static class ClassAllocator implements Allocator {
 
         @Override
