/* * Copyright (c) 2012, 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 * under the terms of the GNU General Public License version 2 only, as * published by the Free Software Foundation. * * This code is distributed in the hope that it will be useful, but WITHOUT * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License * version 2 for more details (a copy is included in the LICENSE file that * accompanied this code). * * You should have received a copy of the GNU General Public License version * 2 along with this work; if not, write to the Free Software Foundation, * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. * * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA * or visit www.oracle.com if you need additional information or have any * questions. */ package org.graalvm.compiler.debug.internal; import static org.graalvm.compiler.debug.DebugCloseable.VOID_CLOSEABLE; import org.graalvm.compiler.debug.Debug; import org.graalvm.compiler.debug.DebugCloseable; import org.graalvm.compiler.debug.DebugMemUseTracker; import org.graalvm.compiler.debug.Management; import org.graalvm.compiler.debug.internal.method.MethodMetricsImpl; public class MemUseTrackerImpl extends AccumulatedDebugValue implements DebugMemUseTracker { private final boolean intercepting; public static long getCurrentThreadAllocatedBytes() { return Management.getCurrentThreadAllocatedBytes(); } /** * Records the most recent active tracker. */ private static final ThreadLocal currentTracker = new ThreadLocal<>(); public MemUseTrackerImpl(String name, boolean conditional, boolean intercepting) { super(name, conditional, new DebugValue(name + "_Flat", conditional) { @Override public String toString(long value) { return valueToString(value); } @Override public String rawUnit() { return "B"; } @Override public String toRawString(long value) { return Long.toString(value); } }); this.intercepting = intercepting; } @Override public DebugCloseable start() { if (!isConditional() || Debug.isMemUseTrackingEnabled()) { CloseableCounterImpl result = intercepting ? new MemUseInterceptingCloseableCounterImpl(this) : new MemUseCloseableCounterImpl(this); currentTracker.set(result); return result; } else { return VOID_CLOSEABLE; } } public static String valueToString(long value) { return String.format("%d bytes", value); } @Override public String toString(long value) { return valueToString(value); } private static final class MemUseCloseableCounterImpl extends CloseableCounterImpl implements DebugCloseable { private MemUseCloseableCounterImpl(AccumulatedDebugValue counter) { super(currentTracker.get(), counter); } @Override long getCounterValue() { return getCurrentThreadAllocatedBytes(); } @Override public void close() { super.close(); currentTracker.set(parent); } } private static final class MemUseInterceptingCloseableCounterImpl extends CloseableCounterImpl implements DebugCloseable { private MemUseInterceptingCloseableCounterImpl(AccumulatedDebugValue counter) { super(currentTracker.get(), counter); } @Override long getCounterValue() { return getCurrentThreadAllocatedBytes(); } @Override public void close() { super.close(); currentTracker.set(parent); } @Override protected void interceptDifferenceAccm(long difference) { if (Debug.isMethodMeterEnabled()) { MethodMetricsImpl.addToCurrentScopeMethodMetrics(counter.getName(), difference); } } @Override protected void interceptDifferenceFlat(long difference) { if (Debug.isMethodMeterEnabled()) { MethodMetricsImpl.addToCurrentScopeMethodMetrics(counter.flat.getName(), difference); } } } @Override public String rawUnit() { return "B"; } @Override public String toRawString(long value) { return Long.toString(value); } }