[GR-19220] Added Truffle::System.allocated_bytes_of_current_thread (#2708)

PullRequest: truffleruby/3465

Author: eregon

mx.truffleruby/suite.py

@@ -201,6 +201,7 @@
             "requires": [
                 "java.logging",
                 "java.management",
+                "jdk.management",
                 "jdk.unsupported", # sun.misc.Signal
             ],
             "dependencies": [

src/main/java/org/truffleruby/core/TruffleSystemNodes.java

@@ -39,11 +39,13 @@
 package org.truffleruby.core;
 
 import java.io.IOException;
+import java.lang.management.ManagementFactory;
 import java.nio.file.NoSuchFileException;
 import java.util.Set;
 import java.util.logging.Level;
 
 import com.oracle.truffle.api.strings.TruffleString;
+import com.sun.management.ThreadMXBean;
 import org.truffleruby.RubyLanguage;
 import org.truffleruby.builtins.CoreMethod;
 import org.truffleruby.builtins.CoreMethodArrayArgumentsNode;
@@ -273,4 +275,19 @@ protected int availableProcessors() {
 
     }
 
+    @CoreMethod(names = "allocated_bytes_of_current_thread", onSingleton = true)
+    public abstract static class AllocatedBytesNode extends CoreMethodArrayArgumentsNode {
+        private static ThreadMXBean bean;
+
+        @TruffleBoundary
+        @Specialization
+        protected static long allocatedBytes() {
+            if (bean == null) {
+                bean = (ThreadMXBean) ManagementFactory.getThreadMXBean();
+                bean.setThreadAllocatedMemoryEnabled(true);
+            }
+
+            return bean.getCurrentThreadAllocatedBytes();
+        }
+    }
 }