1 /* 2 * Copyright (c) 2016, 2018, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. 8 * 9 * This code is distributed in the hope that it will be useful, but WITHOUT 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 12 * version 2 for more details (a copy is included in the LICENSE file that 13 * accompanied this code). 14 * 15 * You should have received a copy of the GNU General Public License version 16 * 2 along with this work; if not, write to the Free Software Foundation, 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 18 * 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 20 * or visit www.oracle.com if you need additional information or have any 21 * questions. 22 * 23 */ 24 25 import java.io.File; 26 import java.io.IOException; 27 import java.nio.file.Path; 28 import java.util.List; 29 30 import jdk.jfr.Event; 31 import jdk.jfr.EventType; 32 import jdk.jfr.FlightRecorder; 33 import jdk.jfr.Recording; 34 import jdk.jfr.consumer.RecordingFile; 35 import jdk.jfr.consumer.RecordedEvent; 36 37 public class GetFlightRecorder { 38 private static class TestEvent extends Event { 39 } 40 private static class SimpleEvent extends Event { 41 public int id; 42 } 43 public static void main(String args[]) throws Exception { 44 EventType type = EventType.getEventType(TestEvent.class); // This class is loaded before recording has started. 45 if (type.isEnabled()) { 46 throw new RuntimeException("Expected event to be disabled before recording start"); 47 } 48 49 // (1) make sure you can obtain the flight recorder without error. 50 System.out.println("jdk.jfr.FlightRecorder.getFlightRecorder() = " + FlightRecorder.getFlightRecorder()); 51 52 // (2) test that the event class loaded before recording can still work. 53 Recording r = new Recording(); 54 r.start(); 55 if (!type.isEnabled()) { 56 throw new RuntimeException("Expected event to be enabled during recording"); 57 } 58 TestEvent testEvent = new TestEvent(); 59 testEvent.commit(); 60 loadEventClassDuringRecording(); 61 r.stop(); 62 if (type.isEnabled()) { 63 throw new RuntimeException("Expected event to be disabled after recording stopped"); 64 } 65 System.out.println("Checking SimpleEvent"); 66 hasEvent(r, SimpleEvent.class.getName()); 67 System.out.println("OK"); 68 69 System.out.println("Checking TestEvent"); 70 hasEvent(r, TestEvent.class.getName()); 71 System.out.println("OK"); 72 } 73 74 // Classes that are loaded during a recording 75 // should get instrumentation on class load 76 private static void loadEventClassDuringRecording() { 77 SimpleEvent event = new SimpleEvent(); 78 event.commit(); 79 } 80 81 public static List<RecordedEvent> fromRecording(Recording recording) throws IOException { 82 return RecordingFile.readAllEvents(makeCopy(recording)); 83 } 84 85 private static Path makeCopy(Recording recording) throws IOException { 86 Path p = recording.getDestination(); 87 if (p == null) { 88 File directory = new File("."); 89 // FIXME: Must come up with a way to give human-readable name 90 // this will at least not clash when running parallel. 91 ProcessHandle h = ProcessHandle.current(); 92 p = new File(directory.getAbsolutePath(), "recording-" + recording.getId() + "-pid" + h.pid() + ".jfr").toPath(); 93 recording.dump(p); 94 } 95 return p; 96 } 97 98 public static void hasEvent(Recording r, String name) throws IOException { 99 List<RecordedEvent> events = fromRecording(r); 100 hasEvents(events); 101 hasEvent(events, name); 102 } 103 104 public static void hasEvents(List<RecordedEvent> events) { 105 if (events.isEmpty()) { 106 throw new RuntimeException("No events"); 107 } 108 } 109 110 public static void hasEvent(List<RecordedEvent> events, String name) throws IOException { 111 if (!containsEvent(events, name)) { 112 throw new RuntimeException("Missing event " + name + " in recording " + events.toString()); 113 } 114 } 115 116 private static boolean containsEvent(List<RecordedEvent> events, String name) { 117 for (RecordedEvent event : events) { 118 if (event.getEventType().getName().equals(name)) { 119 return true; 120 } 121 } 122 return false; 123 } 124 }