src/jdk.jfr/share/classes/jdk/jfr/internal/cmd/Command.java
author egahlin
Tue, 15 May 2018 20:24:34 +0200
changeset 50113 caf115bb98ad
permissions -rw-r--r--
8199712: Flight Recorder Reviewed-by: coleenp, ihse, erikj, dsamersoff, mseledtsov, egahlin, mgronlun Contributed-by: erik.gahlin@oracle.com, markus.gronlund@oracle.com
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
     1
/*
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
     2
 * Copyright (c) 2016, 2018, Oracle and/or its affiliates. All rights reserved.
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
     3
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
     4
 *
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
     5
 * This code is free software; you can redistribute it and/or modify it
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
     6
 * under the terms of the GNU General Public License version 2 only, as
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
     7
 * published by the Free Software Foundation.  Oracle designates this
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
     8
 * particular file as subject to the "Classpath" exception as provided
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
     9
 * by Oracle in the LICENSE file that accompanied this code.
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    10
 *
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    11
 * This code is distributed in the hope that it will be useful, but WITHOUT
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    12
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    13
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    14
 * version 2 for more details (a copy is included in the LICENSE file that
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    15
 * accompanied this code).
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    16
 *
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    17
 * You should have received a copy of the GNU General Public License version
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    18
 * 2 along with this work; if not, write to the Free Software Foundation,
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    19
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    20
 *
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    21
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    22
 * or visit www.oracle.com if you need additional information or have any
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    23
 * questions.
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    24
 */
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    25
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    26
package jdk.jfr.internal.cmd;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    27
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    28
import java.nio.file.Files;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    29
import java.nio.file.Path;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    30
import java.util.ArrayList;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    31
import java.util.Collections;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    32
import java.util.Deque;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    33
import java.util.List;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    34
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    35
abstract class Command {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    36
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    37
    private final static Command HELP = new HelpCommand();
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    38
    private final static List<Command> COMMANDS = createCommands();
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    39
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    40
    static void displayHelp() {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    41
        System.out.println("Usage: java " + Execute.class.getName() + " <command> [<options>]");
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    42
        System.out.println();
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    43
        displayAvailableCommands();
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    44
    }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    45
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    46
    static void displayAvailableCommands() {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    47
        System.out.println("Available commands are:");
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    48
        System.out.println();
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    49
        boolean first = true;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    50
        for (Command c : Command.COMMANDS) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    51
            if (!first) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    52
                System.out.println();
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    53
            }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    54
            System.out.println("  " + c.getName() + " " + c.getOptionSyntax());
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    55
            System.out.println("    " + c.getDescription());
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    56
            first = false;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    57
        }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    58
    }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    59
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    60
    public static List<Command> getCommands() {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    61
        return COMMANDS;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    62
    }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    63
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    64
    public static Command valueOf(String commandName) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    65
        for (Command command : COMMANDS) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    66
            if (command.getName().equals(commandName)) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    67
                return command;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    68
            }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    69
        }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    70
        return null;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    71
    }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    72
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    73
    abstract public String getOptionSyntax();
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    74
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    75
    abstract public String getName();
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    76
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    77
    abstract public String getDescription();
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    78
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    79
    abstract public void displayOptionUsage();
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    80
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    81
    abstract public void execute(Deque<String> options);
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    82
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    83
    final protected void userFailed(String message) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    84
        println();
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    85
        println(message);
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    86
        displayUsage();
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    87
        throw new IllegalArgumentException(message);
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    88
    }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    89
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    90
    final protected void ensureMaxArgumentCount(Deque<String> options, int maxCount) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    91
        if (options.size() > maxCount) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    92
            userFailed("Too many arguments");
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    93
        }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    94
    }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    95
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    96
    final protected void ensureMinArgumentCount(Deque<String> options, int minCount) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    97
        if (options.size() < minCount) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    98
            userFailed("Too few arguments");
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    99
        }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   100
    }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   101
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   102
    final protected void ensureFileExist(Path file) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   103
        if (!Files.exists(file)) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   104
            userFailed("Could not find file " + file);
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   105
        }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   106
    }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   107
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   108
    final protected Path ensureFileDoesNotExist(Path file) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   109
        if (Files.exists(file)) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   110
            userFailed("File " + file + " already exists");
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   111
        }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   112
        return file;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   113
    }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   114
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   115
    final protected void ensureJFRFile(Path path) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   116
        if (!path.toString().endsWith(".jfr")) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   117
            userFailed("Filename must end with .jfr");
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   118
        }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   119
    }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   120
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   121
    final protected void displayUsage() {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   122
        String javaText = "java " + Execute.class.getName();
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   123
        println();
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   124
        println("Usage: " + javaText + " " + getName() + " " + getOptionSyntax());
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   125
        println();
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   126
        displayOptionUsage();
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   127
    }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   128
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   129
    final protected void println() {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   130
        System.out.println();
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   131
    }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   132
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   133
    final protected void print(String text) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   134
        System.out.print(text);
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   135
    }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   136
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   137
    final protected void println(String text) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   138
        System.out.println(text);
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   139
    }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   140
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   141
    private static List<Command> createCommands() {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   142
        List<Command> commands = new ArrayList<>();
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   143
        commands.add(new PrintCommand());
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   144
        commands.add(new SummaryCommand());
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   145
        commands.add(new ReconstructCommand());
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   146
        commands.add(new SplitCommand());
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   147
        commands.add(HELP);
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   148
        return Collections.unmodifiableList(commands);
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   149
    }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   150
}