--- /dev/null 2018-06-17 23:18:20.806999507 +0800 +++ new/test/jdk/jfr/startupargs/StartupHelper.java 2019-01-29 11:05:37.270118613 +0800 @@ -0,0 +1,106 @@ +/* + * Copyright (c) 2013, 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 + * under the terms of the GNU General Public License version 2 only, as + * published by the Free Software Foundation. Oracle designates this + * particular file as subject to the "Classpath" exception as provided + * by Oracle in the LICENSE file that accompanied this code. + * + * 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 jdk.jfr.startupargs; + +import java.io.IOException; +import java.nio.file.DirectoryStream; +import java.nio.file.Files; +import java.nio.file.Path; +import java.util.ArrayList; +import java.util.Arrays; +import java.util.List; +import java.util.stream.Collectors; + +import jdk.jfr.FlightRecorder; +import jdk.jfr.Recording; +import jdk.testlibrary.Asserts; +import jdk.testlibrary.dcmd.CommandExecutor; +import jdk.testlibrary.dcmd.PidJcmdExecutor; +import jdk.testlibrary.OutputAnalyzer; + + + +public class StartupHelper { + + public static Recording getRecording(String name) { + for (Recording r : FlightRecorder.getFlightRecorder().getRecordings()) { + System.out.println("Recording=" + r.getName()); + if (name.equals(r.getName())) { + return r; + } + } + Asserts.fail("No recording with name " + name); + return null; + } + + public static List listFilesInDir(Path root) throws IOException { + List paths = new ArrayList<>(); + List searchPaths = new ArrayList<>(); + searchPaths.add(root); + + while (!searchPaths.isEmpty()) { + Path currRoot = searchPaths.remove(searchPaths.size() - 1); + DirectoryStream contents = Files.newDirectoryStream(currRoot); + for (Path path : contents) { + paths.add(path); + if (Files.isDirectory(path)) { + searchPaths.add(path); + } + } + } + + System.out.println("Files in '" + root + "':"); + for (Path path : paths) { + System.out.println(" path: " + path); + } + return paths; + } + + public static Path findRecordingFileInRepo(Path baseLocation) throws IOException { + // Check that repo contains a file ending in "jfr" or "part" + for (Path path : listFilesInDir(baseLocation)) { + if (Files.isRegularFile(path)) { + String filename = path.toString(); + if (filename.endsWith("jfr") || filename.endsWith("part")) { + return path; + } + } + } + return null; + } + + public static OutputAnalyzer jcmd(String... args) { + String argsString = Arrays.stream(args).collect(Collectors.joining(" ")); + CommandExecutor executor = new PidJcmdExecutor(); + OutputAnalyzer oa = executor.execute(argsString); + oa.shouldHaveExitValue(0); + return oa; + } + + public static OutputAnalyzer jcmdCheck(String recordingName) { + return jcmd("JFR.check", "name=" + recordingName, "verbose=true"); + } +}