src/jdk.jfr/share/classes/jdk/jfr/internal/cmd/PrintCommand.java

/*
 * Copyright (c) 2016, 2018, 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.  Oracle designates this
 * particular file as subject to the "Classpath" exception as provided
 * by Oracle in the LICENSE file that accompanied this code.
 *
 * 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.internal.cmd;

import java.io.IOException;
import java.io.PrintWriter;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.Deque;

final class PrintCommand extends Command {
    @Override
    public String getName() {
        return "print";
    }

    @Override
    public String getOptionSyntax() {
        return "[--xml|--json] <file>";
    }

    @Override
    public String getDescription() {
        return "Print contents of a recording file (.jfr)";
    }

    @Override
    public void displayOptionUsage() {
        println("  --xml    Print a recording in XML format");
        println();
        println("  --json   Print a recording in JSON format");
        println();
        println("  <file>   Location of the recording file (.jfr) to print");
    }

    @Override
    public void execute(Deque<String> options) {
        if (options.isEmpty()) {
            userFailed("Missing file");
        }
        ensureMaxArgumentCount(options, 2);

        Path file = Paths.get(options.removeLast());

        ensureFileExist(file);
        ensureJFRFile(file);
        ensureMaxArgumentCount(options, 1);

        String format = "--pretty";
        if (!options.isEmpty()) {
            format = options.remove();
        }
        try (PrintWriter pw = new PrintWriter(System.out)) {
            try {
                switch (format) {
                case "--pretty":
                    PrettyWriter prettyWriter = new PrettyWriter(pw);
                    prettyWriter.print(file);
                    break;
                case "--xml":
                    XMLWriter xmlPrinter = new XMLWriter(pw);
                    xmlPrinter.print(file);
                    break;
                case "--json":
                    JSONWriter jsonWriter = new JSONWriter(pw);
                    jsonWriter.print(file);
                    break;
                default:
                    userFailed("Unknown option " + format);
                    break;
                }
            } catch (IOException ioe) {
                userFailed("Could not read recording at " + file.toAbsolutePath() + ". " + ioe.getMessage());
            }
        }
    }
}