com.shinesolutions.aemstackmanager.service.CommandExecutor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of aem-stack-manager Show documentation
Java application for managing AEM infrastructure created using aem-aws-stack-builder
The newest version!
package com.shinesolutions.aemstackmanager.service;

import org.apache.commons.exec.*;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

import java.io.IOException;
import java.util.List;
import java.util.stream.Collectors;

@Component
public class CommandExecutor {

    private final Logger logger = LoggerFactory.getLogger(this.getClass());

    @Value("${command.shell}")
    private String shell;

    @Value("${command.timeout}")
    private long timeout;

    public boolean executeReturnBoolean(String command) {

        try {

            execute(command);

        } catch (ExecuteException e) {

            logger.error("Error executing command: " + command + ", check logs for errors", e);

            //remove from queue
            return true;

        } catch (IOException | InterruptedException e) {

            logger.error("Error executing command: " + command + ", check logs for errors", e);

            //leave in queue for retry.
            return false;
        }

        return true;

    }

    public void execute(String command) throws IOException, InterruptedException {

        execute(command, new CollectingLogOutputStream());

    }

    private void execute(String command, LogOutputStream outputStream) throws IOException, InterruptedException {

        logger.debug("Executing command: " + command);

        CommandLine commandLine = new CommandLine(shell);

        commandLine.addArguments(new String[]{
                "-c",
                command
        }, false);

        PumpStreamHandler streamHandler = new PumpStreamHandler(outputStream);

        Executor executor = new DefaultExecutor();
        executor.setStreamHandler(streamHandler);
        executor.setExitValue(1);

        ExecuteWatchdog watchdog = new ExecuteWatchdog(timeout);
        executor.setWatchdog(watchdog);

        try {

            DefaultExecuteResultHandler resultHandler = new DefaultExecuteResultHandler();
            executor.execute(commandLine, resultHandler);

            // some time later the result handler callback was invoked so we
            // can safely request the exit value
            resultHandler.waitFor();
            int exitValue = resultHandler.getExitValue();

            logger.debug("Command execution complete. exitValue: " + exitValue);

            if(outputStream instanceof CollectingLogOutputStream) {

                //check lines for Exitcode: != 0
                List matches = ((CollectingLogOutputStream)outputStream).getLines().stream()
                        .filter(it -> it.contains("Exitcode:"))
                        .filter(it -> !it.trim().endsWith(" 0"))
                        .collect(Collectors.toList());

                if(!matches.isEmpty()){
                    throw new ExecuteException("Execution of: " + commandLine.toString() + " Returned output containing ExitCode != 0", 1);
                }

            }

        } catch (IOException | InterruptedException e) {

            logger.error("Error executing command: " + command + ", check logs for errors", e);

            throw e;
        }

    }


    public String executeReturnOutput(String command) throws IOException, InterruptedException {

        StringBuilder stringBuilder = new StringBuilder();

        LogOutputStream outputStream = new CollectingLogOutputStream() {
            @Override
            protected void processLine(String line, int level) {
                super.processLine(line, level);
                stringBuilder.append(line);
            }
        };

        execute(command, outputStream);

        return stringBuilder.toString();

    }

    public List executeReturnOutputAsList(String command) throws IOException, InterruptedException {

        CollectingLogOutputStream outputStream = new CollectingLogOutputStream();

        execute(command, outputStream);

        return outputStream.getLines();

    }

}