1 /*

     2  * Copyright (c) 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 

    26 package jdk.jfr.api.event;

    27 

    28 import java.util.ArrayList;

    29 import java.util.Iterator;

    30 import java.util.List;

    31 

    32 import jdk.jfr.Event;

    33 import jdk.jfr.EventType;

    34 import jdk.jfr.Recording;

    35 import jdk.jfr.consumer.RecordedEvent;

    36 import jdk.test.lib.Asserts;

    37 import jdk.test.lib.jfr.Events;

    38 

    39 /*

    40  * @test

    41  * @summary Test enable/disable event and verify recording has expected events.

    42  * @key jfr

    43  * @library /test/lib

    44  * @run main/othervm jdk.jfr.api.event.TestEnableDisable

    45  */

    46 

    47 public class TestEnableDisable {

    48 

    49     public static void main(String[] args) throws Exception {

    50         List<MyEvent> expectedEvents = new ArrayList<>();

    51         Recording r = new Recording();

    52 

    53         r.start();

    54         createEvent(expectedEvents, true); // id=0 Custom event classes are enabled by default.

    55 

    56         r.disable(MyEvent.class);

    57         createEvent(expectedEvents, false); // id=1

    58         r.enable(MyEvent.class);

    59         createEvent(expectedEvents, true);  // id=2

    60 

    61         // enable/disable by event setting name

    62         String eventSettingName = String.valueOf(EventType.getEventType(MyEvent.class).getId());

    63         System.out.println("eventSettingName=" + eventSettingName);

    64 

    65         r.disable(eventSettingName);

    66         createEvent(expectedEvents, false); // id=3

    67         r.enable(eventSettingName);

    68         createEvent(expectedEvents, true);

    69 

    70         r.stop();

    71         createEvent(expectedEvents, false);

    72 

    73         Iterator<MyEvent> expectedIterator = expectedEvents.iterator();

    74         for (RecordedEvent event : Events.fromRecording(r)) {

    75             System.out.println("event.id=" + Events.assertField(event, "id").getValue());

    76             Asserts.assertTrue(expectedIterator.hasNext(), "Found more events than expected");

    77             Events.assertField(event, "id").equal(expectedIterator.next().id);

    78         }

    79         Asserts.assertFalse(expectedIterator.hasNext(), "Did not find all expected events.");

    80 

    81         r.close();

    82     }

    83 

    84     private static int eventId = 0;

    85     private static void createEvent(List<MyEvent> expectedEvents, boolean isExpected) {

    86         MyEvent event = new MyEvent();

    87         event.begin();

    88         event.id = eventId;

    89         event.commit();

    90 

    91         if (isExpected) {

    92             expectedEvents.add(event);

    93         }

    94         eventId++;

    95     }

    96 

    97     private static class MyEvent extends Event {

    98         private int id;

    99     }

   100 

   101 }