--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/test/hotspot/jtreg/runtime/cds/appcds/test-classes/GetFlightRecorder.java Mon Jul 29 10:34:20 2019 -0400
@@ -0,0 +1,124 @@
+/*
+ * Copyright (c) 2016, 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.
+ *
+ * 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.
+ *
+ */
+
+import java.io.File;
+import java.io.IOException;
+import java.nio.file.Path;
+import java.util.List;
+
+import jdk.jfr.Event;
+import jdk.jfr.EventType;
+import jdk.jfr.FlightRecorder;
+import jdk.jfr.Recording;
+import jdk.jfr.consumer.RecordingFile;
+import jdk.jfr.consumer.RecordedEvent;
+
+public class GetFlightRecorder {
+ private static class TestEvent extends Event {
+ }
+ private static class SimpleEvent extends Event {
+ public int id;
+ }
+ public static void main(String args[]) throws Exception {
+ EventType type = EventType.getEventType(TestEvent.class); // This class is loaded before recording has started.
+ if (type.isEnabled()) {
+ throw new RuntimeException("Expected event to be disabled before recording start");
+ }
+
+ // (1) make sure you can obtain the flight recorder without error.
+ System.out.println("jdk.jfr.FlightRecorder.getFlightRecorder() = " + FlightRecorder.getFlightRecorder());
+
+ // (2) test that the event class loaded before recording can still work.
+ Recording r = new Recording();
+ r.start();
+ if (!type.isEnabled()) {
+ throw new RuntimeException("Expected event to be enabled during recording");
+ }
+ TestEvent testEvent = new TestEvent();
+ testEvent.commit();
+ loadEventClassDuringRecording();
+ r.stop();
+ if (type.isEnabled()) {
+ throw new RuntimeException("Expected event to be disabled after recording stopped");
+ }
+ System.out.println("Checking SimpleEvent");
+ hasEvent(r, SimpleEvent.class.getName());
+ System.out.println("OK");
+
+ System.out.println("Checking TestEvent");
+ hasEvent(r, TestEvent.class.getName());
+ System.out.println("OK");
+ }
+
+ // Classes that are loaded during a recording
+ // should get instrumentation on class load
+ private static void loadEventClassDuringRecording() {
+ SimpleEvent event = new SimpleEvent();
+ event.commit();
+ }
+
+ public static List<RecordedEvent> fromRecording(Recording recording) throws IOException {
+ return RecordingFile.readAllEvents(makeCopy(recording));
+ }
+
+ private static Path makeCopy(Recording recording) throws IOException {
+ Path p = recording.getDestination();
+ if (p == null) {
+ File directory = new File(".");
+ // FIXME: Must come up with a way to give human-readable name
+ // this will at least not clash when running parallel.
+ ProcessHandle h = ProcessHandle.current();
+ p = new File(directory.getAbsolutePath(), "recording-" + recording.getId() + "-pid" + h.pid() + ".jfr").toPath();
+ recording.dump(p);
+ }
+ return p;
+ }
+
+ public static void hasEvent(Recording r, String name) throws IOException {
+ List<RecordedEvent> events = fromRecording(r);
+ hasEvents(events);
+ hasEvent(events, name);
+ }
+
+ public static void hasEvents(List<RecordedEvent> events) {
+ if (events.isEmpty()) {
+ throw new RuntimeException("No events");
+ }
+ }
+
+ public static void hasEvent(List<RecordedEvent> events, String name) throws IOException {
+ if (!containsEvent(events, name)) {
+ throw new RuntimeException("Missing event " + name + " in recording " + events.toString());
+ }
+ }
+
+ private static boolean containsEvent(List<RecordedEvent> events, String name) {
+ for (RecordedEvent event : events) {
+ if (event.getEventType().getName().equals(name)) {
+ return true;
+ }
+ }
+ return false;
+ }
+}