org.apache.drill.yarn.client.CommandLineOptions Maven / Gradle / Ivy
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.drill.yarn.client;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
/**
* Drill YARN client command line options.
*
* DrillYarnClient -h|--help |
* start |
* stop |
* status |
* resize [+|-]n
*
*
* - help: Prints command line usage
* - start: starts the defined cluster
* - stop: stops the defined cluster
* - resize: adds (+n), removes (-n) or resizes (n) the cluster
* - status: prints status about the cluster
*
*
* This is a do-it-yourself parser because the command line parser
* used by Drill does not accept parameters (arguments) without a dash,
* and does not accept arguments (such as resize -3) with a dash.
*/
public class CommandLineOptions {
public enum Command {
HELP( "help", "Provide description of usage."),
/**
* Primary command to upload the application archive and start the Drill cluster.
*/
START( "start", "Start the cluster."),
// Removed at QA request. QA wants a "real" restart. Also, upload of the
// archive is fast enough that a "start without upload" option is not really
// needed.
// /**
// * Convenience method when debugging, testing. Restarts the cluster without the
// * archive upload; assumes the upload was already done.
// */
//
// RESTART( "restart", "Restart the cluster (without archive upload)."),
/**
* Primary command to stop a running cluster.
*/
STOP( "stop", "Stop the cluster."),
/**
* Primary command to get the status of a running cluster.
*/
STATUS( "status", "Provide the status of the cluster."),
RESIZE( "resize", "Resize the cluster +n: add nodes, -n: remove nodes, n resize to given size."),
TEST( null, null ),
/**
* Convenience command to display the effective configuration settings to
* diagnose problems.
*/
DESCRIBE( "describe", "Display and validate configuration." ),
/**
* Convenience command to upload the application archive to test the DFS
* settings without launching the Drill cluster.
*/
UPLOAD( "upload", "Upload archives to validate DFS." ),
/**
* Convenience command to remove the Drill-on-YARN archive(s) from DFS.
* Note: doing this while a Drill cluster is running will cause subsequent
* Drillbit launches to fail.
*/
CLEAN( "clean", "Remove archives stored in DFS." );
private String cmd;
private String descrip;
private Command(String cmd, String descrip) {
this.cmd = cmd;
this.descrip = descrip;
}
public boolean isMatch(String arg) {
String key = (cmd == null) ? toString() : cmd;
return key.equalsIgnoreCase(arg);
}
public boolean isHidden() {
return descrip == null;
}
public String getCommand( ) { return cmd; }
public String getDescription( ) { return descrip; }
}
Command command;
public String appId;
public boolean dryRun;
public String resizePrefix;
public int resizeValue;
public boolean verbose = false;
public boolean force = false;
/**
* Parse the command line. Invalid option combinations result in the
* error option being set.
*/
public boolean parse(String args[]) {
for (int i = 0; i < args.length; i++) {
String arg = args[i];
if (arg.equals("-h") || arg.equals("-?")) {
command = Command.HELP;
break;
}
if (arg.equals("-v") || arg.equals("--verbose")) {
verbose = true;
continue;
}
if (arg.equals("-f") || arg.equals("--force")) {
force = true;
continue;
}
if (arg.equals("-d") || arg.equals("--dryrun")) {
dryRun = true;
continue;
}
if (arg.equals("-a") || arg.equals("--appid")) {
if (i + 1 == args.length) {
return false;
}
appId = args[++i];
continue;
}
if (command != null) {
command = null;
return false;
}
// Check if a command line word matches this command. Be nice,
// allow -foo and --foo in addition to the "proper" foo.
String cmdStr = arg;
if (cmdStr.startsWith("--")) {
cmdStr = arg.substring(2);
} else if (cmdStr.startsWith("-")) {
cmdStr = cmdStr.substring(1);
}
for (Command cmd : Command.values()) {
if (cmd.isMatch(cmdStr)) {
command = cmd;
if (command == Command.RESIZE) {
if (i + 1 == args.length) {
command = null;
break;
}
parseResizeOption(args[++i]);
}
break;
}
}
}
return true;
}
private void parseResizeOption(String resize) {
Pattern p = Pattern.compile("([+-]?)(\\d+)");
Matcher m = p.matcher(resize);
if (m.matches()) {
resizePrefix = m.group(1);
resizeValue = Integer.parseInt(m.group(2));
} else {
command = null;
return;
}
}
public Command getCommand() {
return command;
}
public String getResizePrefix() {
return resizePrefix;
}
public int getResizeValue() {
return resizeValue;
}
public void usage() {
ClientContext.out.println(
"Usage: drill-on-yarn.sh [--site site-dir] command [-v|--verbose][-a app-id]");
ClientContext.out.println("Where command is one of:");
for (Command cmd : Command.values()) {
if (cmd.isHidden()) {
continue;
}
ClientContext.out
.println(" " + cmd.getCommand() + " - " + cmd.getDescription());
}
}
}