--- /dev/null 2017-07-07 09:32:06.000000000 -0700 +++ new/src/jdk.internal.vm.compiler/share/classes/org.graalvm.compiler.debug/src/org/graalvm/compiler/debug/GlobalMetrics.java 2017-07-07 09:32:06.000000000 -0700 @@ -0,0 +1,133 @@ +/* + * Copyright (c) 2017, 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; + +import java.io.IOException; +import java.io.PrintStream; +import java.nio.file.Files; +import java.nio.file.Paths; +import java.util.Collections; +import java.util.List; + +import org.graalvm.compiler.options.OptionValues; +import org.graalvm.util.EconomicMap; +import org.graalvm.util.MapCursor; +import org.graalvm.util.Pair; + +/** + * Metric values that can be {@linkplain #add(DebugContext) updated} by multiple threads. + */ +public class GlobalMetrics { + long[] values; + + /** + * Adds the values in {@code debug} to the values in this object. + */ + public synchronized void add(DebugContext debug) { + values = debug.addValuesTo(values); + } + + /** + * Clears all values in this object. + */ + public void clear() { + values = null; + } + + /** + * Creates and returns a sorted map from metric names to their values in this object. + */ + public EconomicMap asKeyValueMap() { + List keys = KeyRegistry.getKeys(); + Collections.sort(keys, MetricKey.NAME_COMPARATOR); + EconomicMap res = EconomicMap.create(keys.size()); + long[] vals = values; + for (MetricKey key : keys) { + int index = ((AbstractKey) key).getIndex(); + if (vals == null || index >= vals.length) { + res.put(key, 0L); + } else { + res.put(key, vals[index]); + } + } + return res; + } + + /** + * Prints the values in the object to the file specified by + * {@link DebugOptions#AggregatedMetricsFile} if present otherwise to + * {@link DebugContext#DEFAULT_LOG_STREAM}. + */ + public void print(OptionValues options) { + long[] vals = values; + if (vals != null) { + EconomicMap map = asKeyValueMap(); + String metricsFile = DebugOptions.AggregatedMetricsFile.getValue(options); + boolean csv = metricsFile != null && (metricsFile.endsWith(".csv") || metricsFile.endsWith(".CSV")); + try (PrintStream p = metricsFile == null ? DebugContext.DEFAULT_LOG_STREAM : new PrintStream(Files.newOutputStream(Paths.get(metricsFile)))) { + if (!csv) { + if (!map.isEmpty()) { + p.println("++ Aggregated Metrics ++"); + } + } + String csvFormat = CSVUtil.buildFormatString("%s", "%s", "%s"); + MapCursor e = map.getEntries(); + while (e.advance()) { + MetricKey key = e.getKey(); + if (csv) { + Pair valueAndUnit = key.toCSVFormat(e.getValue()); + CSVUtil.Escape.println(p, csvFormat, key.getName(), valueAndUnit.getLeft(), valueAndUnit.getRight()); + } else { + p.println(key.getName() + "=" + key.toHumanReadableFormat(e.getValue())); + } + } + if (!csv) { + if (!map.isEmpty()) { + p.println("-- Aggregated Metrics --"); + } + } + } catch (IOException e) { + e.printStackTrace(); + } + } + + if (DebugOptions.ListMetrics.getValue(options)) { + PrintStream p = System.out; + p.println("++ Metric Keys ++"); + List keys = KeyRegistry.getKeys(); + Collections.sort(keys, MetricKey.NAME_COMPARATOR); + for (MetricKey key : keys) { + String name = key.getDocName(); + if (name != null) { + String doc = key.getDoc(); + if (doc != null) { + p.println(name + ": " + doc); + } else { + p.println(name); + } + } + } + p.println("-- Metric Keys --"); + } + } +}