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.  Oracle designates this

     8  * particular file as subject to the "Classpath" exception as provided

     9  * by Oracle in the LICENSE file that accompanied this code.

    10  *

    11  * This code is distributed in the hope that it will be useful, but WITHOUT

    12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

    13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

    14  * version 2 for more details (a copy is included in the LICENSE file that

    15  * accompanied this code).

    16  *

    17  * You should have received a copy of the GNU General Public License version

    18  * 2 along with this work; if not, write to the Free Software Foundation,

    19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

    20  *

    21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

    22  * or visit www.oracle.com if you need additional information or have any

    23  * questions.

    24  */

    25 package jdk.jfr.api.recorder;

    26 

    27 import java.util.concurrent.CountDownLatch;

    28 

    29 import jdk.jfr.FlightRecorder;

    30 import jdk.jfr.FlightRecorderListener;

    31 import jdk.jfr.Recording;

    32 import jdk.jfr.RecordingState;

    33 

    34 /*

    35  * @test TestRecorderListener

    36  *

    37  * @key jfr

    38  * @run main/othervm jdk.jfr.api.recorder.TestRecorderListener

    39  */

    40 public class TestRecorderListener {

    41 

    42     static class Listener implements FlightRecorderListener {

    43 

    44         private final CountDownLatch latch = new CountDownLatch(1);

    45         private final RecordingState waitFor;

    46 

    47         public Listener(RecordingState state) {

    48             waitFor = state;

    49         }

    50 

    51         @Override

    52         public void recordingStateChanged(Recording recording) {

    53             System.out.println("Listener: recording=" + recording.getName() + " state=" + recording.getState());

    54             RecordingState rs = recording.getState();

    55             if (rs == waitFor) {

    56                 latch.countDown();

    57             }

    58         }

    59 

    60         public void waitFor() throws InterruptedException {

    61             latch.await();

    62         }

    63     }

    64 

    65     public static void main(String... args) throws Exception {

    66         Listener recordingListener = new Listener(RecordingState.RUNNING);

    67         FlightRecorder.addListener(recordingListener);

    68 

    69         Listener stoppedListener = new Listener(RecordingState.STOPPED);

    70         FlightRecorder.addListener(stoppedListener);

    71 

    72         Listener finishedListener = new Listener(RecordingState.CLOSED);

    73         FlightRecorder.addListener(finishedListener);

    74 

    75         Recording recording = new Recording();

    76         if (recording.getState() != RecordingState.NEW) {

    77             recording.close();

    78             throw new Exception("New recording should be in NEW state");

    79         }

    80 

    81         recording.start();

    82         recordingListener.waitFor();

    83 

    84         recording.stop();

    85         stoppedListener.waitFor();

    86 

    87         recording.close();

    88         finishedListener.waitFor();

    89 

    90         testDefaultrecordingStateChangedListener();

    91 

    92     }

    93 

    94     private static class DummyListener implements FlightRecorderListener {

    95 

    96     }

    97 

    98     private static void testDefaultrecordingStateChangedListener() {

    99         FlightRecorder.addListener(new DummyListener());

   100         Recording recording = new Recording();

   101         recording.start();

   102         recording.stop();

   103         recording.close();

   104     }

   105 }