[LinuxCommandExecutor] #Java

LinuxCommandExecutor.java

/**
 * @author howechen
 */
@Component
public class SystemCommandExecutor {

    private static final Logger logger = LoggerFactory.getLogger(SystemCommandExecutor.class);
    private ExecuteWatchdog watchdog = new ExecuteWatchdog(60000);
    private PluginExecuteResultHandler resultHandler = new PluginExecuteResultHandler();


    public SystemCommandExecutor() {
    }

    public void run(String command) throws PluginException {
        run(command, false);
    }

    public void run(CommandLine commandLine) throws PluginException {
        run(commandLine, false);
    }

    public void run(String command, boolean oneAsExit) throws PluginException {
        CommandLine commandLine = handleCommand(command);
        run(commandLine, oneAsExit);
    }

    public void run(CommandLine commandLine, boolean oneAsExit) throws PluginException {
        DefaultExecutor executor = new DefaultExecutor();
        if (oneAsExit) {
            executor.setExitValue(1);
        }
        executor.setWatchdog(watchdog);
        try {
            executor.execute(commandLine);
        } catch (IOException e) {
            throw new PluginException(e.getMessage());
        }
    }

    public PluginExecuteResultHandler runAsync(String command) throws PluginException {
        return runAsync(command, false);
    }

    public PluginExecuteResultHandler runAsync(CommandLine commandLine) throws PluginException {
        return runAsync(commandLine, false);
    }

    public PluginExecuteResultHandler runAsync(String command, boolean oneAsExit) throws PluginException {

        CommandLine commandLine = handleCommand(command);
        return runAsync(commandLine, oneAsExit);
    }

    public PluginExecuteResultHandler runAsync(CommandLine commandLine, boolean oneAsExit) throws PluginException {

        DefaultExecutor executor = new DefaultExecutor();
        if (oneAsExit) {
            executor.setExitValue(1);
        }
        LogOutputStream stream = new LogOutputStream() {
            @Override
            protected void processLine(String s, int i) {
                System.out.println(s);
            }
        };
        executor.setWatchdog(watchdog);
        executor.setStreamHandler(new PumpStreamHandler(stream));
        try {
            executor.execute(commandLine, resultHandler);
        } catch (IOException e) {
            throw new PluginException(e.getMessage());
        }
        return resultHandler;
    }

    private CommandLine handleCommand(String command) {
        return CommandLine.parse(command);
    }

    public static class PluginExecuteResultHandler extends DefaultExecuteResultHandler {
        public PluginExecuteResultHandler() {
        }

        @Override
        public void onProcessComplete(int exitValue) {
            super.onProcessComplete(exitValue);
            logger.info("Run command success");
        }

        @Override
        public void onProcessFailed(ExecuteException e) throws PluginException {
            super.onProcessFailed(e);
            logger.error("Run command failed");
            throw new PluginException(e.getMessage());
        }
    }
}