com.marklogic.flux.impl.VersionCommand Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of flux-api Show documentation

Flux API for data movement with MarkLogic

There is a newer version: 1.0.0.ea1

/*
 * Copyright © 2024 MarkLogic Corporation. All Rights Reserved.
 */
package com.marklogic.flux.impl;

import com.fasterxml.jackson.databind.ObjectMapper;
import org.apache.spark.sql.SparkSession;
import picocli.CommandLine;

import java.util.ListResourceBundle;
import java.util.ResourceBundle;

@CommandLine.Command(
    name = "version",
    description = "Print the version of Flux."
)
public class VersionCommand implements Command {

    private static final String BUILD_TIME = "buildTime";

    private CommandLine commandLine;

    @CommandLine.Option(names = "--verbose", description = "Print additional details about Flux.")
    private boolean verbose;

    @Override
    public void execute(SparkSession session) {
        ResourceBundle versionProperties = getResourceBundle();
        final String version = versionProperties.getString("version");
        final String javaVersion = System.getProperty("java.version");
        final String sparkVersion = session.version();
        if (verbose) {
            commandLine.getOut().println(new ObjectMapper().createObjectNode()
                .put("fluxVersion", version)
                .put(BUILD_TIME, versionProperties.getString(BUILD_TIME))
                .put("javaVersion", javaVersion)
                .put("sparkVersion", sparkVersion));
        } else {
            commandLine.getOut().println("Flux version: " + version);
            commandLine.getOut().println("Java version: " + javaVersion);
            commandLine.getOut().println("Spark version: " + sparkVersion);
        }
    }

    @Override
    public void validateCommandLineOptions(CommandLine.ParseResult parseResult) {
        this.commandLine = parseResult.commandSpec().commandLine();
    }

    private ResourceBundle getResourceBundle() {
        try {
            return ResourceBundle.getBundle("flux-version");
        } catch (Exception e) {
            // This should only occur in a development environment, and specifically only when running tests via
            // something besides Gradle.
            return new ListResourceBundle() {
                @Override
                protected Object[][] getContents() {
                    return new Object[][]{
                        {"version", "Unknown; please ensure that flux-version.properties exists on the classpath."},
                        {BUILD_TIME, "Unknown."}
                    };
                }
            };
        }
    }
}