hu.bme.mit.trainbenchmark.benchmark.fourstore.driver.UnixUtils.java Source code

Introduction

Here is the source code for hu.bme.mit.trainbenchmark.benchmark.fourstore.driver.UnixUtils.java

Source

package hu.bme.mit.trainbenchmark.benchmark.fourstore.driver;

/*******************************************************************************
 * Copyright (c) 2010-2014, Gabor Szarnyas, Istvan Rath and Daniel Varro
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 * Gabor Szarnyas - initial API and implementation
 *******************************************************************************/

import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.util.Map;
import java.util.Map.Entry;

import org.apache.commons.exec.CommandLine;
import org.apache.commons.exec.DefaultExecutor;
import org.apache.commons.exec.ExecuteException;
import org.apache.commons.exec.PumpStreamHandler;
import org.apache.commons.exec.environment.EnvironmentUtils;
import org.apache.commons.io.IOUtils;
import org.apache.commons.io.output.NullOutputStream;

import com.google.common.collect.ImmutableMap;

public class UnixUtils {

    public static String createTempFileFromScript(final String script) throws IOException {
        return createTempFileFromResource(script, ".sh", true);
    }

    public static String createTempFileFromResource(final String script, final String extension,
            final boolean executable) throws IOException {
        final InputStream scriptInputStream = UnixUtils.class.getResourceAsStream("/" + script);

        // create a temporary file
        final File scriptTempFile = File.createTempFile("unix-utils-", extension);
        scriptTempFile.deleteOnExit();
        try (FileOutputStream out = new FileOutputStream(scriptTempFile)) {
            IOUtils.copy(scriptInputStream, out);
        }
        scriptTempFile.setExecutable(executable);

        final String command = scriptTempFile.getAbsolutePath();
        return command;
    }

    public static void execResourceScript(final String command, final Map<String, String> environmentVariables,
            final boolean showOutput) throws IOException {
        execResourceScript(command, "", environmentVariables, showOutput);
    }

    public static void execResourceScript(final String command, final String arguments,
            final Map<String, String> environmentVariables, final boolean showOutput) throws IOException {
        final String tempScript = UnixUtils.createTempFileFromScript(command);

        final String scriptCommand = tempScript + " " + arguments;
        if (showOutput) {
            System.out.println("Command: " + scriptCommand);
        }
        exec(scriptCommand, environmentVariables, showOutput ? System.out : new NullOutputStream());
    }

    public static BufferedReader exec(final String command, final Map<String, String> environmentVariables)
            throws IOException {
        final InputStream is = execToStream(command, environmentVariables);
        return new BufferedReader(new InputStreamReader(is));
    }

    public static InputStream execToStream(final String command, final Map<String, String> environmentVariables)
            throws IOException, ExecuteException {
        final ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
        exec(command, environmentVariables, byteArrayOutputStream);
        final ByteArrayInputStream is = new ByteArrayInputStream(byteArrayOutputStream.toByteArray());
        return is;
    }

    public static BufferedReader exec(final String command) throws IOException {
        return exec(command, ImmutableMap.<String, String>of());
    }

    public static void exec(final String command, final Map<String, String> environmentVariables,
            final OutputStream outputStream) throws IOException, ExecuteException {
        final Map<?, ?> executionEnvironment = EnvironmentUtils.getProcEnvironment();
        for (final Entry<String, String> environmentVariable : environmentVariables.entrySet()) {
            final String keyAndValue = environmentVariable.getKey() + "=" + environmentVariable.getValue();
            EnvironmentUtils.addVariableToEnvironment(executionEnvironment, keyAndValue);
        }

        final PumpStreamHandler streamHandler = new PumpStreamHandler(outputStream);

        final CommandLine commandLine = new CommandLine("/bin/bash");
        commandLine.addArguments(new String[] { "-c", command }, false);

        final DefaultExecutor executor = new DefaultExecutor();
        executor.setStreamHandler(streamHandler);
        executor.execute(commandLine, executionEnvironment);
    }

}