io.cloudslang.content.utilities.util.ProcessExecutor.java Source code

Introduction

Here is the source code for io.cloudslang.content.utilities.util.ProcessExecutor.java

Source

/*
 * (c) Copyright 2017 EntIT Software LLC, a Micro Focus company, L.P.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Apache License v2.0 which accompany this distribution.
 *
 * The Apache License is available 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.
 */
/*
 * (c) Copyright 2017 Hewlett-Packard Enterprise Development Company, L.P.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Apache License v2.0 which accompany this distribution.
 *
 * The Apache License is available at
 * http://www.apache.org/licenses/LICENSE-2.0
 *
*/
package io.cloudslang.content.utilities.util;

import com.google.common.collect.ImmutableList;
import io.cloudslang.content.utilities.entities.ProcessResponseEntity;
import org.apache.commons.lang3.text.StrTokenizer;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;
import java.util.concurrent.TimeoutException;

import static java.util.concurrent.TimeUnit.MILLISECONDS;
import static org.apache.commons.io.IOUtils.closeQuietly;
import static org.apache.commons.lang3.StringUtils.isEmpty;
import static org.apache.commons.lang3.SystemUtils.IS_OS_WINDOWS;
import static org.apache.commons.lang3.SystemUtils.IS_OS_WINDOWS_95;
import static org.apache.commons.lang3.SystemUtils.IS_OS_WINDOWS_98;
import static org.apache.commons.lang3.SystemUtils.IS_OS_WINDOWS_ME;

public class ProcessExecutor {
    private final List<String> arguments;

    public ProcessExecutor() {
        if (IS_OS_WINDOWS_95 || IS_OS_WINDOWS_98 || IS_OS_WINDOWS_ME) {
            arguments = ImmutableList.of("command.exe", "/C");
        } else if (IS_OS_WINDOWS) {
            arguments = ImmutableList.of("cmd.exe", "/C");
        } else {
            arguments = ImmutableList.of();
        }
    }

    private List<String> processCommand(String commandLine) {
        List<String> command = new ArrayList<>(arguments);
        if (!isEmpty(commandLine)) {
            command.addAll(new StrTokenizer(commandLine, ',', '"').getTokenList());
        }

        return command;
    }

    public ProcessResponseEntity execute(String commandLine, int timeout)
            throws IOException, ExecutionException, InterruptedException, TimeoutException {
        Process process = new ProcessBuilder().command(processCommand(commandLine)).start();

        ProcessStreamConsumer processStreamConsumer = new ProcessStreamConsumer(process);
        ExecutorService executor = Executors.newFixedThreadPool(1);
        Future<ProcessResponseEntity> futureResult = executor.submit(processStreamConsumer);

        try {
            return futureResult.get(timeout, MILLISECONDS);
        } catch (InterruptedException | ExecutionException | TimeoutException e) {
            stopProcess(process, futureResult);
            throw e;
        } finally {
            executor.shutdownNow();
        }
    }

    private void stopProcess(Process process, Future<ProcessResponseEntity> futureResult) {
        futureResult.cancel(true);
        process.destroy();

        closeQuietly(process.getInputStream());
        closeQuietly(process.getErrorStream());
    }

}