| /* |
| * Copyright (c) 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. |
| */ |
| |
| package jdk.jfr.javaagent; |
| |
| import java.lang.instrument.Instrumentation; |
| import java.nio.file.Path; |
| import java.nio.file.Paths; |
| |
| import jdk.jfr.Configuration; |
| import jdk.jfr.Event; |
| import jdk.jfr.Name; |
| import jdk.jfr.Recording; |
| import jdk.jfr.consumer.RecordingFile; |
| import jdk.test.lib.Asserts; |
| import jdk.test.lib.jfr.EventNames; |
| |
| // Java agent that emits in multiple threads |
| public class EventEmitterAgent { |
| |
| private static final int THREADS = 5; |
| private static final int EVENTS_PER_THREAD = 150_000; |
| private static final int EXPECTED_COUNT = THREADS * EVENTS_PER_THREAD; |
| private static final Path DUMP_PATH = Paths.get("dump.jfr").toAbsolutePath(); |
| |
| // Called when agent is loaded from command line |
| public static void agentmain(String agentArgs, Instrumentation inst) throws Exception { |
| agentWork(); |
| } |
| |
| // Called when agent is dynamically loaded |
| public static void premain(String agentArgs, Instrumentation inst) throws Exception { |
| agentWork(); |
| } |
| |
| private static void agentWork() throws Exception { |
| try (Recording r = new Recording(Configuration.getConfiguration("default"))) { |
| r.enable(EventNames.JavaExceptionThrow); |
| r.setDestination(DUMP_PATH); |
| r.start(); |
| Thread[] threads = new Thread[THREADS]; |
| for (int i = 0; i < THREADS; i++) { |
| threads[i] = new Thread(EventEmitterAgent::emitEvents); |
| threads[i].start(); |
| } |
| for (int i = 0; i < THREADS; i++) { |
| threads[i].join(); |
| } |
| r.stop(); |
| } |
| } |
| |
| public static void emitEvents() { |
| for (int i = 0; i < EVENTS_PER_THREAD; i++) { |
| TestEvent e = new TestEvent(); |
| e.msg = "Long message that puts pressure on the string pool " + i % 100; |
| e.count = i; |
| e.thread = Thread.currentThread(); |
| e.clazz = String.class; |
| e.commit(); |
| if (i % 10000 == 0) { |
| try { |
| Thread.sleep(1); |
| } catch (InterruptedException ie) { |
| // ignore |
| } |
| } |
| } |
| } |
| |
| @Name("Test") |
| static class TestEvent extends Event { |
| String msg; |
| int count; |
| Thread thread; |
| Class<?> clazz; |
| } |
| |
| public static void validateRecording() throws Exception { |
| long testEventCount = RecordingFile.readAllEvents(DUMP_PATH) |
| .stream() |
| .filter(e -> e.getEventType().getName().equals("Test")) |
| .count(); |
| Asserts.assertTrue(testEventCount == EXPECTED_COUNT, "Mismatch in TestEvent count"); |
| } |
| } |