Java Command类代码示例

import org.apache.hadoop.fs.shell.Command; //导入依赖的package包/类
@Test
public void testExceptionNullMessage() throws Exception {
  final String cmdName = "-cmdExNullMsg";
  final Command cmd = Mockito.mock(Command.class);
  Mockito.when(cmd.run(Mockito.<String>anyVararg())).thenThrow(
      new IllegalArgumentException());
  Mockito.when(cmd.getUsage()).thenReturn(cmdName);

  final CommandFactory cmdFactory = Mockito.mock(CommandFactory.class);
  final String[] names = {cmdName};
  Mockito.when(cmdFactory.getNames()).thenReturn(names);
  Mockito.when(cmdFactory.getInstance(cmdName)).thenReturn(cmd);

  FsShell shell = new FsShell(new Configuration());
  shell.commandFactory = cmdFactory;
  try (GenericTestUtils.SystemErrCapturer capture =
           new GenericTestUtils.SystemErrCapturer()) {
    ToolRunner.run(shell, new String[]{cmdName});
    Assert.assertThat(capture.getOutput(),
        StringContains.containsString(cmdName
            + ": Null exception message"));
  }
}
 

import org.apache.hadoop.fs.shell.Command; //导入依赖的package包/类
private void displayError(String cmd, String message) {
  for (String line : message.split("\n")) {
    System.err.println(cmd + ": " + line);
    if (cmd.charAt(0) != '-') {
      Command instance = null;
      instance = commandFactory.getInstance("-" + cmd);
      if (instance != null) {
        System.err.println("Did you mean -" + cmd + "?  This command " +
            "begins with a dash.");
      }
    }
  }
}
 

import org.apache.hadoop.fs.shell.Command; //导入依赖的package包/类
private void printInstanceHelp(PrintStream out, Command instance) {
  boolean firstLine = true;
  for (String line : instance.getDescription().split("\n")) {
    String prefix;
    if (firstLine) {
      prefix = instance.getUsage() + ":\t";
      firstLine = false;
    } else {
      prefix = "\t\t";
    }
    System.out.println(prefix + line);
  }    
}
 

import org.apache.hadoop.fs.shell.Command; //导入依赖的package包/类
@Override
public Command getInstance(String cmd) {
    Command command = commandFactory.getInstance(cmd);
    command.err = this.err;
    command.out = this.out;
    return command;
}
 

import org.apache.hadoop.fs.shell.Command; //导入依赖的package包/类
private void printInstanceUsage(PrintStream out, Command instance) {
  out.println(getUsagePrefix() + " " + instance.getUsage());
}
 

import org.apache.hadoop.fs.shell.Command; //导入依赖的package包/类
String runCommand(String cmdName, String[] arguments)  {
        final Configuration conf = contextCommands.getConfiguration();

        cmdName = "-" + cmdName;
        final Command command = getCommandInstance(cmdName, conf);
        if (command == null) {
            return "Unknown command " + cmdName;
        }

        if (arguments.length == 0 && !NO_PARAMS_COMMANDS.contains(cmdName)) {
            return command.getDescription();
        }


        contextCommands.updateCurrentWorkingDirectory();

        arguments = replaceHdfsPath(arguments);
        final PrintStream printStream = ClientConnection.context.get();
        if (printStream != null) {
            command.err = printStream;
            command.out = printStream;
        }
//        final ByteArrayOutputStream out = new ByteArrayOutputStream();
//        final PrintStream printStream = new PrintStream(out);
//        command.err = printStream;
//        command.out = printStream;

        final int result = command.run(arguments);
        if (contextCommands.isShowResultCode()) {
            if (result == 0) {
                command.out.println("Exit code = " + result);
            } else {
                command.err.println("Exit code = " + result);
            }
        }

        //printStream.close();
        if (result != 0 && contextCommands.isFailOnError()) {
            throw new RuntimeException("HDFS Command finished with result code " + result);
        }
        return "";
    }
 

import org.apache.hadoop.fs.shell.Command; //导入依赖的package包/类
private static Command getCommandInstance(String cmdName, Configuration conf) {
    final CommandFactory commandFactory = new CommandFactory(conf);
    FsCommand.registerCommands(commandFactory);
    return commandFactory.getInstance(cmdName, conf);
}
 

import org.apache.hadoop.fs.shell.Command; //导入依赖的package包/类
private void printInstanceUsage(PrintStream out, Command instance) {
  out.println(usagePrefix + " " + instance.getUsage());
}
 

import org.apache.hadoop.fs.shell.Command; //导入依赖的package包/类
@Override
public void addClass(Class<? extends Command> cmdClass, String... names) {
    commandFactory.addClass(cmdClass, names);
}
 

import org.apache.hadoop.fs.shell.Command; //导入依赖的package包/类
@Override
public void addObject(Command cmdObject, String... names) {
    commandFactory.addObject(cmdObject, names);
}