test/jdk/jdk/jfr/api/consumer/streaming/TestOutOfProcessMigration.java
author egahlin
Mon, 25 Nov 2019 14:06:13 +0100
changeset 59246 fcad92f425c5
parent 59226 a0f39cc47387
permissions -rw-r--r--
8234703: JFR TestOutOfProcessMigration.java should clean up files Reviewed-by: mgronlun
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
59226
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
     1
/*
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
     2
 * Copyright (c) 2019, Oracle and/or its affiliates. All rights reserved.
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
     3
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
     4
 *
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
     5
 * This code is free software; you can redistribute it and/or modify it
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
     6
 * under the terms of the GNU General Public License version 2 only, as
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
     7
 * published by the Free Software Foundation.  Oracle designates this
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
     8
 * particular file as subject to the "Classpath" exception as provided
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
     9
 * by Oracle in the LICENSE file that accompanied this code.
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    10
 *
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    11
 * This code is distributed in the hope that it will be useful, but WITHOUT
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    12
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    13
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    14
 * version 2 for more details (a copy is included in the LICENSE file that
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    15
 * accompanied this code).
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    16
 *
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    17
 * You should have received a copy of the GNU General Public License version
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    18
 * 2 along with this work; if not, write to the Free Software Foundation,
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    19
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    20
 *
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    21
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    22
 * or visit www.oracle.com if you need additional information or have any
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    23
 * questions.
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    24
 */
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    25
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    26
package jdk.jfr.api.consumer.streaming;
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    27
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    28
import java.nio.file.Path;
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    29
import java.nio.file.Paths;
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    30
import java.time.Instant;
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    31
import java.util.concurrent.atomic.AtomicInteger;
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    32
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    33
import jdk.jfr.consumer.EventStream;
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    34
import jdk.test.lib.dcmd.CommandExecutor;
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    35
import jdk.test.lib.dcmd.PidJcmdExecutor;
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    36
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    37
/**
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    38
 * @test
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    39
 * @summary Verifies that a out-of-process stream is closed when the repository
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    40
 *          is changed.
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    41
 * @key jfr
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    42
 * @requires vm.hasJFR
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    43
 * @library /test/lib /test/jdk
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    44
 * @modules jdk.jfr jdk.attach java.base/jdk.internal.misc
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    45
 * @run main/othervm jdk.jfr.api.consumer.streaming.TestOutOfProcessMigration
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    46
 */
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    47
public class TestOutOfProcessMigration {
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    48
    public static void main(String... args) throws Exception {
59246
fcad92f425c5 8234703: JFR TestOutOfProcessMigration.java should clean up files
egahlin
parents: 59226
diff changeset
    49
        try (TestProcess process = new TestProcess("application"))  {
fcad92f425c5 8234703: JFR TestOutOfProcessMigration.java should clean up files
egahlin
parents: 59226
diff changeset
    50
            AtomicInteger eventCounter = new AtomicInteger();
fcad92f425c5 8234703: JFR TestOutOfProcessMigration.java should clean up files
egahlin
parents: 59226
diff changeset
    51
            Path newRepo = Paths.get("new-repository").toAbsolutePath();
fcad92f425c5 8234703: JFR TestOutOfProcessMigration.java should clean up files
egahlin
parents: 59226
diff changeset
    52
            try (EventStream es = EventStream.openRepository(process.getRepository())) {
fcad92f425c5 8234703: JFR TestOutOfProcessMigration.java should clean up files
egahlin
parents: 59226
diff changeset
    53
                // Start from first event in repository
fcad92f425c5 8234703: JFR TestOutOfProcessMigration.java should clean up files
egahlin
parents: 59226
diff changeset
    54
                es.setStartTime(Instant.EPOCH);
fcad92f425c5 8234703: JFR TestOutOfProcessMigration.java should clean up files
egahlin
parents: 59226
diff changeset
    55
                es.onEvent(e -> {
fcad92f425c5 8234703: JFR TestOutOfProcessMigration.java should clean up files
egahlin
parents: 59226
diff changeset
    56
                    if (eventCounter.incrementAndGet() == TestProcess.NUMBER_OF_EVENTS) {
fcad92f425c5 8234703: JFR TestOutOfProcessMigration.java should clean up files
egahlin
parents: 59226
diff changeset
    57
                        System.out.println("Changing repository to " + newRepo + " ...");
fcad92f425c5 8234703: JFR TestOutOfProcessMigration.java should clean up files
egahlin
parents: 59226
diff changeset
    58
                        CommandExecutor executor = new PidJcmdExecutor(String.valueOf(process.pid()));
fcad92f425c5 8234703: JFR TestOutOfProcessMigration.java should clean up files
egahlin
parents: 59226
diff changeset
    59
                        // This should close stream
fcad92f425c5 8234703: JFR TestOutOfProcessMigration.java should clean up files
egahlin
parents: 59226
diff changeset
    60
                        executor.execute("JFR.configure repositorypath=" + newRepo);
fcad92f425c5 8234703: JFR TestOutOfProcessMigration.java should clean up files
egahlin
parents: 59226
diff changeset
    61
                    }
fcad92f425c5 8234703: JFR TestOutOfProcessMigration.java should clean up files
egahlin
parents: 59226
diff changeset
    62
                });
fcad92f425c5 8234703: JFR TestOutOfProcessMigration.java should clean up files
egahlin
parents: 59226
diff changeset
    63
                es.start();
fcad92f425c5 8234703: JFR TestOutOfProcessMigration.java should clean up files
egahlin
parents: 59226
diff changeset
    64
                process.exit();
fcad92f425c5 8234703: JFR TestOutOfProcessMigration.java should clean up files
egahlin
parents: 59226
diff changeset
    65
                // Wait for process to die, so files are cleaned up
fcad92f425c5 8234703: JFR TestOutOfProcessMigration.java should clean up files
egahlin
parents: 59226
diff changeset
    66
                process.awaitDeath();
fcad92f425c5 8234703: JFR TestOutOfProcessMigration.java should clean up files
egahlin
parents: 59226
diff changeset
    67
            }
59226
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    68
        }
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    69
    }
a0f39cc47387 8233700: EventStream not closed
egahlin
parents:
diff changeset
    70
}