--- /dev/null 2017-07-07 09:32:09.000000000 -0700 +++ new/src/jdk.internal.vm.compiler/share/classes/org.graalvm.compiler.debug/src/org/graalvm/compiler/debug/UniquePathUtilities.java 2017-07-07 09:32:09.000000000 -0700 @@ -0,0 +1,123 @@ +/* + * Copyright (c) 2016, 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.nio.file.Path; +import java.nio.file.Paths; +import java.util.HashMap; +import java.util.concurrent.atomic.AtomicInteger; +import java.util.concurrent.atomic.AtomicLong; + +import org.graalvm.compiler.options.OptionKey; +import org.graalvm.compiler.options.OptionValues; + +public class UniquePathUtilities { + + private static final AtomicLong globalTimeStamp = new AtomicLong(); + /** + * This generates a per thread persistent id to aid mapping related dump files with each other. + */ + private static final ThreadLocal threadDumpId = new ThreadLocal<>(); + private static final AtomicInteger dumpId = new AtomicInteger(); + + static class PerThreadSequence { + final int threadID; + HashMap sequences = new HashMap<>(2); + + PerThreadSequence(int threadID) { + this.threadID = threadID; + } + + String generateID(String extension) { + Integer box = sequences.get(extension); + if (box == null) { + sequences.put(extension, 1); + return Integer.toString(threadID); + } else { + sequences.put(extension, box + 1); + return Integer.toString(threadID) + '-' + box; + } + } + } + + private static String getThreadDumpId(String extension) { + PerThreadSequence id = threadDumpId.get(); + if (id == null) { + id = new PerThreadSequence(dumpId.incrementAndGet()); + threadDumpId.set(id); + } + return id.generateID(extension); + } + + public static String formatExtension(String ext) { + if (ext == null || ext.length() == 0) { + return ""; + } + return "." + ext; + } + + public static long getGlobalTimeStamp() { + if (globalTimeStamp.get() == 0) { + globalTimeStamp.compareAndSet(0, System.currentTimeMillis()); + } + return globalTimeStamp.get(); + } + + /** + * Generates a {@link Path} using the format "%s-%d_%d%s" with the {@code baseNameOption}, a + * {@link #getGlobalTimeStamp() global timestamp} , {@link #getThreadDumpId a per thread unique + * id} and an optional {@code extension}. + * + * @return the output file path or null if the flag is null + */ + public static Path getPath(OptionValues options, OptionKey option, String extension) throws IOException { + return getPath(options, option, extension, true); + } + + /** + * Generate a {@link Path} using the format "%s-%d_%s" with the {@code baseNameOption}, a + * {@link #getGlobalTimeStamp() global timestamp} and an optional {@code extension} . + * + * @return the output file path or null if the flag is null + */ + public static Path getPathGlobal(OptionValues options, OptionKey baseNameOption, String extension) throws IOException { + return getPath(options, baseNameOption, extension, false); + } + + private static Path getPath(OptionValues options, OptionKey baseNameOption, String extension, boolean includeThreadId) throws IOException { + if (baseNameOption.getValue(options) == null) { + return null; + } + String ext = formatExtension(extension); + final String name = includeThreadId + ? String.format("%s-%d_%s%s", baseNameOption.getValue(options), getGlobalTimeStamp(), getThreadDumpId(ext), ext) + : String.format("%s-%d%s", baseNameOption.getValue(options), getGlobalTimeStamp(), ext); + Path result = Paths.get(name); + if (result.isAbsolute()) { + return result; + } + Path dumpDir = DebugOptions.getDumpDirectory(options); + return dumpDir.resolve(name).normalize(); + } +}