--- old/src/hotspot/share/include/jmm.h 2019-09-13 11:38:47.000000000 -0700 +++ new/src/hotspot/share/include/jmm.h 2019-09-13 11:38:46.000000000 -0700 @@ -1,5 +1,5 @@ /* - * Copyright (c) 2003, 2017, Oracle and/or its affiliates. All rights reserved. + * Copyright (c) 2003, 2019, Oracle and/or its affiliates. All rights reserved. * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. * * This code is free software; you can redistribute it and/or modify it @@ -239,6 +239,9 @@ jobject (JNICALL *GetMemoryPoolUsage) (JNIEnv* env, jobject pool); jobject (JNICALL *GetPeakMemoryPoolUsage) (JNIEnv* env, jobject pool); + jlong (JNICALL *GetOneThreadAllocatedMemory) + (JNIEnv *env, + jlong thread_id); void (JNICALL *GetThreadAllocatedMemory) (JNIEnv *env, jlongArray ids, --- old/src/hotspot/share/services/management.cpp 2019-09-13 11:38:50.000000000 -0700 +++ new/src/hotspot/share/services/management.cpp 2019-09-13 11:38:50.000000000 -0700 @@ -2068,6 +2068,31 @@ } #endif // INCLUDE_MANAGEMENT +// Gets the amount of memory allocated on the Java heap for a single thread. +// Returns -1 if the thread does not exist or has terminated. +JVM_ENTRY(jlong, jmm_GetOneThreadAllocatedMemory(JNIEnv *env, jlong thread_id)) + if (thread_id < 0) { + THROW_MSG_(vmSymbols::java_lang_IllegalArgumentException(), + "Invalid thread ID", -1); + } + + if (thread_id == 0) { + // current thread + if (THREAD->is_Java_thread()) { + return ((JavaThread*)THREAD)->cooked_allocated_bytes(); + } + return -1; + } + + ThreadsListHandle tlh; + JavaThread* java_thread = tlh.list()->find_JavaThread_from_java_tid(thread_id); + + if (java_thread != NULL) { + return java_thread->cooked_allocated_bytes(); + } + return -1; +JVM_END + // Gets an array containing the amount of memory allocated on the Java // heap for a set of threads (in bytes). Each element of the array is // the amount of memory allocated for the thread ID specified in the @@ -2192,6 +2217,7 @@ jmm_GetMemoryManagers, jmm_GetMemoryPoolUsage, jmm_GetPeakMemoryPoolUsage, + jmm_GetOneThreadAllocatedMemory, jmm_GetThreadAllocatedMemory, jmm_GetMemoryUsage, jmm_GetLongAttribute, --- old/src/java.management/share/classes/java/lang/management/ThreadMXBean.java 2019-09-13 11:38:53.000000000 -0700 +++ new/src/java.management/share/classes/java/lang/management/ThreadMXBean.java 2019-09-13 11:38:52.000000000 -0700 @@ -160,7 +160,7 @@ * * @return an array of {@code long}, each is a thread ID. * - * @throws java.lang.SecurityException if a security manager + * @throws SecurityException if a security manager * exists and the caller does not have * ManagementPermission("monitor"). */ @@ -199,7 +199,7 @@ * it does not exist. * * @throws IllegalArgumentException if {@code id <= 0}. - * @throws java.lang.SecurityException if a security manager + * @throws SecurityException if a security manager * exists and the caller does not have * ManagementPermission("monitor"). */ @@ -237,7 +237,7 @@ * * @throws IllegalArgumentException if any element in the input array * {@code ids} is {@code <= 0}. - * @throws java.lang.SecurityException if a security manager + * @throws SecurityException if a security manager * exists and the caller does not have * ManagementPermission("monitor"). */ @@ -284,7 +284,7 @@ * * @throws IllegalArgumentException if {@code id <= 0}. * @throws IllegalArgumentException if {@code maxDepth is negative}. - * @throws java.lang.SecurityException if a security manager + * @throws SecurityException if a security manager * exists and the caller does not have * ManagementPermission("monitor"). * @@ -337,7 +337,7 @@ * @throws IllegalArgumentException if {@code maxDepth is negative}. * @throws IllegalArgumentException if any element in the input array * {@code ids} is {@code <= 0}. - * @throws java.lang.SecurityException if a security manager + * @throws SecurityException if a security manager * exists and the caller does not have * ManagementPermission("monitor"). * @@ -360,7 +360,7 @@ * @return {@code true} if thread contention monitoring is enabled; * {@code false} otherwise. * - * @throws java.lang.UnsupportedOperationException if the Java virtual + * @throws UnsupportedOperationException if the Java virtual * machine does not support thread contention monitoring. * * @see #isThreadContentionMonitoringSupported @@ -374,10 +374,10 @@ * @param enable {@code true} to enable; * {@code false} to disable. * - * @throws java.lang.UnsupportedOperationException if the Java + * @throws UnsupportedOperationException if the Java * virtual machine does not support thread contention monitoring. * - * @throws java.lang.SecurityException if a security manager + * @throws SecurityException if a security manager * exists and the caller does not have * ManagementPermission("control"). * @@ -394,7 +394,7 @@ * the current thread has executed in user mode or system mode. * *

- * This is a convenient method for local management use and is + * This is a convenience method for local management use and is * equivalent to calling: *

      *   {@link #getThreadCpuTime getThreadCpuTime}(Thread.currentThread().getId());
@@ -403,7 +403,7 @@
      * @return the total CPU time for the current thread if CPU time
      * measurement is enabled; {@code -1} otherwise.
      *
-     * @throws java.lang.UnsupportedOperationException if the Java
+     * @throws UnsupportedOperationException if the Java
      * virtual machine does not support CPU time measurement for
      * the current thread.
      *
@@ -421,7 +421,7 @@
      * not necessarily nanoseconds accuracy.
      *
      * 

- * This is a convenient method for local management use and is + * This is a convenience method for local management use and is * equivalent to calling: *

      *   {@link #getThreadUserTime getThreadUserTime}(Thread.currentThread().getId());
@@ -430,7 +430,7 @@
      * @return the user-level CPU time for the current thread if CPU time
      * measurement is enabled; {@code -1} otherwise.
      *
-     * @throws java.lang.UnsupportedOperationException if the Java
+     * @throws UnsupportedOperationException if the Java
      * virtual machine does not support CPU time measurement for
      * the current thread.
      *
@@ -467,7 +467,7 @@
      * {@code -1} otherwise.
      *
      * @throws IllegalArgumentException if {@code id <= 0}.
-     * @throws java.lang.UnsupportedOperationException if the Java
+     * @throws UnsupportedOperationException if the Java
      * virtual machine does not support CPU time measurement for
      * other threads.
      *
@@ -502,7 +502,7 @@
      * {@code -1} otherwise.
      *
      * @throws IllegalArgumentException if {@code id <= 0}.
-     * @throws java.lang.UnsupportedOperationException if the Java
+     * @throws UnsupportedOperationException if the Java
      * virtual machine does not support CPU time measurement for
      * other threads.
      *
@@ -548,7 +548,7 @@
      * @return {@code true} if thread CPU time measurement is enabled;
      *         {@code false} otherwise.
      *
-     * @throws java.lang.UnsupportedOperationException if the Java virtual
+     * @throws UnsupportedOperationException if the Java virtual
      * machine does not support CPU time measurement for other threads
      * nor for the current thread.
      *
@@ -564,11 +564,11 @@
      * @param enable {@code true} to enable;
      *               {@code false} to disable.
      *
-     * @throws java.lang.UnsupportedOperationException if the Java
+     * @throws UnsupportedOperationException if the Java
      * virtual machine does not support CPU time measurement for
      * any threads nor for the current thread.
      *
-     * @throws java.lang.SecurityException if a security manager
+     * @throws SecurityException if a security manager
      *         exists and the caller does not have
      *         ManagementPermission("control").
      *
@@ -604,7 +604,7 @@
      * @return an array of IDs of the threads that are monitor
      * deadlocked, if any; {@code null} otherwise.
      *
-     * @throws java.lang.SecurityException if a security manager
+     * @throws SecurityException if a security manager
      *         exists and the caller does not have
      *         ManagementPermission("monitor").
      *
@@ -616,7 +616,7 @@
      * Resets the peak thread count to the current number of
      * live threads.
      *
-     * @throws java.lang.SecurityException if a security manager
+     * @throws SecurityException if a security manager
      *         exists and the caller does not have
      *         ManagementPermission("control").
      *
@@ -642,10 +642,10 @@
      * deadlocked waiting for object monitors or ownable synchronizers, if any;
      * {@code null} otherwise.
      *
-     * @throws java.lang.SecurityException if a security manager
+     * @throws SecurityException if a security manager
      *         exists and the caller does not have
      *         ManagementPermission("monitor").
-     * @throws java.lang.UnsupportedOperationException if the Java virtual
+     * @throws UnsupportedOperationException if the Java virtual
      * machine does not support monitoring of ownable synchronizer usage.
      *
      * @see #isSynchronizerUsageSupported
@@ -704,10 +704,10 @@
      * information about a thread whose ID is in the corresponding
      * element of the input array of IDs.
      *
-     * @throws java.lang.SecurityException if a security manager
+     * @throws SecurityException if a security manager
      *         exists and the caller does not have
      *         ManagementPermission("monitor").
-     * @throws java.lang.UnsupportedOperationException
+     * @throws UnsupportedOperationException
      *         
    *
  • if {@code lockedMonitors} is {@code true} but * the Java virtual machine does not support monitoring @@ -794,10 +794,10 @@ * element of the input array of IDs. * * @throws IllegalArgumentException if {@code maxDepth} is negative. - * @throws java.lang.SecurityException if a security manager + * @throws SecurityException if a security manager * exists and the caller does not have * ManagementPermission("monitor"). - * @throws java.lang.UnsupportedOperationException + * @throws UnsupportedOperationException *
      *
    • if {@code lockedMonitors} is {@code true} but * the Java virtual machine does not support monitoring @@ -835,10 +835,10 @@ * * @return an array of {@link ThreadInfo} for all live threads. * - * @throws java.lang.SecurityException if a security manager + * @throws SecurityException if a security manager * exists and the caller does not have * ManagementPermission("monitor"). - * @throws java.lang.UnsupportedOperationException + * @throws UnsupportedOperationException *
        *
      • if {@code lockedMonitors} is {@code true} but * the Java virtual machine does not support monitoring @@ -884,10 +884,10 @@ * @return an array of {@link ThreadInfo} for all live threads. * * @throws IllegalArgumentException if {@code maxDepth} is negative. - * @throws java.lang.SecurityException if a security manager + * @throws SecurityException if a security manager * exists and the caller does not have * ManagementPermission("monitor"). - * @throws java.lang.UnsupportedOperationException + * @throws UnsupportedOperationException *
          *
        • if {@code lockedMonitors} is {@code true} but * the Java virtual machine does not support monitoring --- old/src/java.management/share/classes/sun/management/ThreadImpl.java 2019-09-13 11:38:56.000000000 -0700 +++ new/src/java.management/share/classes/sun/management/ThreadImpl.java 2019-09-13 11:38:55.000000000 -0700 @@ -29,6 +29,7 @@ import java.lang.management.ThreadInfo; import java.lang.management.ThreadMXBean; import javax.management.ObjectName; +import java.util.Objects; /** * Implementation for java.lang.management.ThreadMXBean as well as providing the @@ -112,11 +113,15 @@ return cpuTimeEnabled; } - protected boolean isThreadAllocatedMemoryEnabled() { + private void ensureThreadAllocatedMemorySupported() { if (!isThreadAllocatedMemorySupported()) { throw new UnsupportedOperationException( - "Thread allocated memory measurement is not supported"); + "Thread allocated memory measurement is not supported."); } + } + + protected boolean isThreadAllocatedMemoryEnabled() { + ensureThreadAllocatedMemorySupported(); return allocatedMemoryEnabled; } @@ -155,16 +160,18 @@ return getThreadInfo(ids, 0); } - private void verifyThreadIds(long[] ids) { - if (ids == null) { - throw new NullPointerException("Null ids parameter."); + private void verifyThreadId(long id) { + if (id <= 0) { + throw new IllegalArgumentException( + "Invalid thread ID parameter: " + id); } + } + + private void verifyThreadIds(long[] ids) { + Objects.requireNonNull(ids); for (int i = 0; i < ids.length; i++) { - if (ids[i] <= 0) { - throw new IllegalArgumentException( - "Invalid thread ID parameter: " + ids[i]); - } + verifyThreadId(ids[i]); } } @@ -342,26 +349,41 @@ } } - protected long getThreadAllocatedBytes(long id) { - long[] ids = new long[1]; - ids[0] = id; - final long[] sizes = getThreadAllocatedBytes(ids); - return sizes[0]; + protected long getCurrentThreadAllocatedBytes() { + if (isThreadAllocatedMemoryEnabled()) { + return getThreadAllocatedMemory0(0); + } + return -1; } - private boolean verifyThreadAllocatedMemory(long[] ids) { - verifyThreadIds(ids); + private boolean verifyThreadAllocatedMemory(long id) { + verifyThreadId(id); + return isThreadAllocatedMemoryEnabled(); + } - // check if Thread allocated memory measurement is supported. - if (!isThreadAllocatedMemorySupported()) { - throw new UnsupportedOperationException( - "Thread allocated memory measurement is not supported."); + protected long getThreadAllocatedBytes(long id) { + boolean verified = verifyThreadAllocatedMemory(id); + + if (verified) { + return getThreadAllocatedMemory0( + Thread.currentThread().getId() == id ? 0 : id); } + return -1; + } + private boolean verifyThreadAllocatedMemory(long[] ids) { + verifyThreadIds(ids); return isThreadAllocatedMemoryEnabled(); } protected long[] getThreadAllocatedBytes(long[] ids) { + Objects.requireNonNull(ids); + + if (ids.length == 1) { + long size = getThreadAllocatedBytes(ids[0]); + return new long[] { size }; + } + boolean verified = verifyThreadAllocatedMemory(ids); long[] sizes = new long[ids.length]; @@ -374,10 +396,7 @@ } protected void setThreadAllocatedMemoryEnabled(boolean enable) { - if (!isThreadAllocatedMemorySupported()) { - throw new UnsupportedOperationException( - "Thread allocated memory measurement is not supported."); - } + ensureThreadAllocatedMemorySupported(); Util.checkControlAccess(); synchronized (this) { @@ -511,6 +530,7 @@ private static native void getThreadTotalCpuTime1(long[] ids, long[] result); private static native long getThreadUserCpuTime0(long id); private static native void getThreadUserCpuTime1(long[] ids, long[] result); + private static native long getThreadAllocatedMemory0(long id); private static native void getThreadAllocatedMemory1(long[] ids, long[] result); private static native void setThreadCpuTimeEnabled0(boolean enable); private static native void setThreadAllocatedMemoryEnabled0(boolean enable); --- old/src/java.management/share/native/libmanagement/ThreadImpl.c 2019-09-13 11:38:58.000000000 -0700 +++ new/src/java.management/share/native/libmanagement/ThreadImpl.c 2019-09-13 11:38:58.000000000 -0700 @@ -1,5 +1,5 @@ /* - * Copyright (c) 2003, 2017, Oracle and/or its affiliates. All rights reserved. + * Copyright (c) 2003, 2019, Oracle and/or its affiliates. All rights reserved. * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. * * This code is free software; you can redistribute it and/or modify it @@ -94,6 +94,13 @@ JNI_FALSE /* user */); } +JNIEXPORT jlong JNICALL +Java_sun_management_ThreadImpl_getThreadAllocatedMemory0 + (JNIEnv *env, jclass cls, jlong tid) +{ + return jmm_interface->GetOneThreadAllocatedMemory(env, tid); +} + JNIEXPORT void JNICALL Java_sun_management_ThreadImpl_getThreadAllocatedMemory1 (JNIEnv *env, jclass cls, jlongArray ids, jlongArray sizeArray) --- old/src/jdk.management/share/classes/com/sun/management/ThreadMXBean.java 2019-09-13 11:39:00.000000000 -0700 +++ new/src/jdk.management/share/classes/com/sun/management/ThreadMXBean.java 2019-09-13 11:39:00.000000000 -0700 @@ -1,5 +1,5 @@ /* - * Copyright (c) 2011, 2013, Oracle and/or its affiliates. All rights reserved. + * Copyright (c) 2011, 2019, Oracle and/or its affiliates. All rights reserved. * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. * * This code is free software; you can redistribute it and/or modify it @@ -61,7 +61,7 @@ * @throws NullPointerException if {@code ids} is {@code null} * @throws IllegalArgumentException if any element in the input array * {@code ids} is {@code <=} {@code 0}. - * @throws java.lang.UnsupportedOperationException if the Java + * @throws UnsupportedOperationException if the Java * virtual machine implementation does not support CPU time * measurement. * @@ -95,7 +95,7 @@ * @throws NullPointerException if {@code ids} is {@code null} * @throws IllegalArgumentException if any element in the input array * {@code ids} is {@code <=} {@code 0}. - * @throws java.lang.UnsupportedOperationException if the Java + * @throws UnsupportedOperationException if the Java * virtual machine implementation does not support CPU time * measurement. * @@ -109,13 +109,44 @@ /** * Returns an approximation of the total amount of memory, in bytes, - * allocated in heap memory for the thread of the specified ID. + * allocated in heap memory for the current thread. + * The returned value is an approximation because some Java virtual machine + * implementations may use object allocation mechanisms that result in a + * delay between the time an object is allocated and the time its size is + * recorded. + * + *

          + * This is a convenience method for local management use and is + * equivalent to calling: + *

          +     *   {@link #getThreadAllocatedBytes getThreadAllocatedBytes}(Thread.currentThread().getId());
          +     * 
          + * + * @return an approximation of the total memory allocated, in bytes, in + * heap memory for the current thread + * if thread memory allocation measurement is enabled; + * {@code -1} otherwise. + * + * @throws UnsupportedOperationException if the Java virtual + * machine implementation does not support thread memory allocation + * measurement. + * + * @see #isThreadAllocatedMemorySupported + * @see #isThreadAllocatedMemoryEnabled + * @see #setThreadAllocatedMemoryEnabled + * @since 14 + */ + public long getCurrentThreadAllocatedBytes(); + + /** + * Returns an approximation of the total amount of memory, in bytes, + * allocated in heap memory for the thread with the specified ID. * The returned value is an approximation because some Java virtual machine * implementations may use object allocation mechanisms that result in a * delay between the time an object is allocated and the time its size is * recorded. *

          - * If the thread of the specified ID is not alive or does not exist, + * If the thread with the specified ID is not alive or does not exist, * this method returns {@code -1}. If thread memory allocation measurement * is disabled, this method returns {@code -1}. * A thread is alive if it has been started and has not yet died. @@ -127,13 +158,13 @@ * * @param id the thread ID of a thread * @return an approximation of the total memory allocated, in bytes, in - * heap memory for a thread of the specified ID - * if the thread of the specified ID exists, the thread is alive, + * heap memory for the thread with the specified ID + * if the thread with the specified ID exists, the thread is alive, * and thread memory allocation measurement is enabled; * {@code -1} otherwise. * * @throws IllegalArgumentException if {@code id} {@code <=} {@code 0}. - * @throws java.lang.UnsupportedOperationException if the Java virtual + * @throws UnsupportedOperationException if the Java virtual * machine implementation does not support thread memory allocation * measurement. * @@ -165,7 +196,7 @@ * @throws NullPointerException if {@code ids} is {@code null} * @throws IllegalArgumentException if any element in the input array * {@code ids} is {@code <=} {@code 0}. - * @throws java.lang.UnsupportedOperationException if the Java virtual + * @throws UnsupportedOperationException if the Java virtual * machine implementation does not support thread memory allocation * measurement. * @@ -194,7 +225,7 @@ * @return {@code true} if thread memory allocation measurement is enabled; * {@code false} otherwise. * - * @throws java.lang.UnsupportedOperationException if the Java virtual + * @throws UnsupportedOperationException if the Java virtual * machine does not support thread memory allocation measurement. * * @see #isThreadAllocatedMemorySupported @@ -208,10 +239,10 @@ * @param enable {@code true} to enable; * {@code false} to disable. * - * @throws java.lang.UnsupportedOperationException if the Java virtual + * @throws UnsupportedOperationException if the Java virtual * machine does not support thread memory allocation measurement. * - * @throws java.lang.SecurityException if a security manager + * @throws SecurityException if a security manager * exists and the caller does not have * ManagementPermission("control"). * --- old/src/jdk.management/share/classes/com/sun/management/internal/HotSpotThreadImpl.java 2019-09-13 11:39:03.000000000 -0700 +++ new/src/jdk.management/share/classes/com/sun/management/internal/HotSpotThreadImpl.java 2019-09-13 11:39:02.000000000 -0700 @@ -58,6 +58,11 @@ } @Override + public long getCurrentThreadAllocatedBytes() { + return super.getCurrentThreadAllocatedBytes(); + } + + @Override public long getThreadAllocatedBytes(long id) { return super.getThreadAllocatedBytes(id); } --- old/test/jdk/com/sun/management/ThreadMXBean/ThreadAllocatedMemory.java 2019-09-13 11:39:05.000000000 -0700 +++ new/test/jdk/com/sun/management/ThreadMXBean/ThreadAllocatedMemory.java 2019-09-13 11:39:05.000000000 -0700 @@ -1,5 +1,5 @@ /* - * Copyright (c) 2011, 2015, Oracle and/or its affiliates. All rights reserved. + * Copyright (c) 2011, 2019, Oracle and/or its affiliates. All rights reserved. * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. * * This code is free software; you can redistribute it and/or modify it @@ -34,8 +34,8 @@ private static com.sun.management.ThreadMXBean mbean = (com.sun.management.ThreadMXBean)ManagementFactory.getThreadMXBean(); private static boolean testFailed = false; - private static boolean done = false; - private static boolean done1 = false; + private static volatile boolean done = false; + private static volatile boolean done1 = false; private static Object obj = new Object(); private static final int NUM_THREADS = 10; private static Thread[] threads = new Thread[NUM_THREADS]; @@ -78,7 +78,13 @@ "ThreadAllocatedMemory is expected to be enabled"); } - long size = mbean.getThreadAllocatedBytes(id); + + // Test current thread two ways + + + // First way, getCurrentThreadAllocatedBytes + + long size = mbean.getCurrentThreadAllocatedBytes(); // implementation could have started measurement when // measurement was enabled, in which case size can be 0 if (size < 0) { @@ -95,18 +101,92 @@ " expected >= " + size); } System.out.println(curThread.getName() + - " Current thread allocated bytes = " + size + - " allocated bytes = " + size1); + " Previous allocated bytes = " + size + + " Current allocated bytes = " + size1); + + + // Second way, getThreadAllocatedBytes + + size = mbean.getThreadAllocatedBytes(id); + // implementation could have started measurement when + // measurement was enabled, in which case size can be 0 + if (size < 0) { + throw new RuntimeException( + "Invalid allocated bytes returned = " + size); + } + + doit(); + // Expected to be size1 >= size + size1 = mbean.getThreadAllocatedBytes(id); + if (size1 < size) { + throw new RuntimeException("Allocated bytes " + size1 + + " expected >= " + size); + } + System.out.println(curThread.getName() + + " Previous allocated bytes = " + size + + " Current allocated bytes = " + size1); + + + // Test a single thread that isn't ourself + + // Start one thread, wait for it to block + // after doing some allocation + done = false; done1 = false; + curThread = new MyThread("MyThread"); + curThread.start(); + id = curThread.getId(); + waitUntilThreadBlocked(curThread); + size = mbean.getThreadAllocatedBytes(id); + + // let thread go to do some more allocation + synchronized (obj) { + done = true; + obj.notifyAll(); + } + + // wait for thread to get going again and sample it + goSleep(400); + size1 = mbean.getThreadAllocatedBytes(id); + if (size > size1) { + throw new RuntimeException( + curThread.getName() + + " previous allocated bytes = " + size + + " > current allocated bytes = " + size1); + } + System.out.println(curThread.getName() + + " Previous allocated bytes = " + size + + " Current allocated bytes = " + size1); + + // let thread exit + synchronized (obj) { + done1 = true; + obj.notifyAll(); + } + + try { + curThread.join(); + } catch (InterruptedException e) { + System.out.println("Unexpected exception is thrown."); + e.printStackTrace(System.out); + testFailed = true; + } + if (testFailed) { + throw new RuntimeException("TEST FAILED"); + } + + + // Test many threads // start threads, wait for them to block + done = false; done1 = false; for (int i = 0; i < NUM_THREADS; i++) { threads[i] = new MyThread("MyThread-" + i); threads[i].start(); } // threads block after doing some allocation - waitUntilThreadBlocked(); + waitUntilThreadsBlocked(); for (int i = 0; i < NUM_THREADS; i++) { sizes[i] = mbean.getThreadAllocatedBytes(threads[i].getId()); @@ -126,7 +206,7 @@ for (int i = 0; i < NUM_THREADS; i++) { long newSize = mbean.getThreadAllocatedBytes(threads[i].getId()); if (sizes[i] > newSize) { - throw new RuntimeException("TEST FAILED: " + + throw new RuntimeException( threads[i].getName() + " previous allocated bytes = " + sizes[i] + " > current allocated bytes = " + newSize); @@ -156,6 +236,7 @@ throw new RuntimeException("TEST FAILED"); } + System.out.println("Test passed"); } @@ -169,7 +250,18 @@ } } - private static void waitUntilThreadBlocked() + private static void waitUntilThreadBlocked(Thread thread) + throws Exception { + while (true) { + goSleep(100); + ThreadInfo info = mbean.getThreadInfo(thread.getId()); + if (info.getThreadState() == Thread.State.WAITING) { + break; + } + } + } + + private static void waitUntilThreadsBlocked() throws Exception { int count = 0; while (count != NUM_THREADS) { @@ -225,7 +317,7 @@ " ThreadAllocatedBytes = " + size2); if (size1 > size2) { - throw new RuntimeException("TEST FAILED: " + getName() + + throw new RuntimeException(getName() + " ThreadAllocatedBytes = " + size1 + " > ThreadAllocatedBytes = " + size2); }