com.palantir.gradle.jdks.setup.common.CommandRunner Maven / Gradle / Ivy
/*
* (c) Copyright 2024 Palantir Technologies Inc. All rights reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License 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.
*/
package com.palantir.gradle.jdks.setup.common;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.nio.charset.StandardCharsets;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.function.Consumer;
import java.util.stream.Collectors;
import java.util.stream.Stream;
public final class CommandRunner {
public static String runWithOutputCollection(ProcessBuilder processBuilder) {
ExecutorService executorService = Executors.newFixedThreadPool(2);
try {
Process process = processBuilder.start();
CompletableFuture outputFuture =
CompletableFuture.supplyAsync(() -> readAllInput(process.getInputStream()), executorService);
CompletableFuture errorOutputFuture =
CompletableFuture.supplyAsync(() -> readAllInput(process.getErrorStream()), executorService);
String output = outputFuture.get();
String errorOutput = errorOutputFuture.get();
int exitCode = process.waitFor();
if (exitCode != 0) {
throw new RuntimeException(String.format(
"Failed to run command '%s'. "
+ "Failed with exit code %d.\nError output:\n\n%s\n\nStandard Output:\n\n%s",
String.join(" ", processBuilder.command()), exitCode, errorOutput, output));
}
return output;
} catch (IOException | InterruptedException e) {
throw new RuntimeException(
String.format("Failed to run command '%s'. ", String.join(" ", processBuilder.command())), e);
} catch (ExecutionException e) {
throw new RuntimeException(
String.format(
"Failed to get output for the command '%s'. ", String.join(" ", processBuilder.command())),
e);
} finally {
executorService.shutdown();
}
}
public static void runWithLogger(
ProcessBuilder processBuilder, Consumer stdOutputWriter, Consumer stdErrWriter) {
ExecutorService executorService = Executors.newFixedThreadPool(2);
try {
Process process = processBuilder.start();
CompletableFuture stdOutputFuture =
CompletableFuture.runAsync(() -> stdOutputWriter.accept(process.getInputStream()), executorService);
CompletableFuture stdErrFuture =
CompletableFuture.runAsync(() -> stdErrWriter.accept(process.getErrorStream()), executorService);
stdOutputFuture.get();
stdErrFuture.get();
int exitCode = process.waitFor();
if (exitCode != 0) {
throw new RuntimeException(String.format(
"Command '%s' failed with exit code %d.",
String.join(" ", processBuilder.command()), exitCode));
}
} catch (IOException | InterruptedException | ExecutionException e) {
throw new RuntimeException(
String.format("Failed to run command '%s'. ", String.join(" ", processBuilder.command())), e);
} finally {
executorService.shutdown();
}
}
public static String readAllInput(InputStream inputStream) {
try (Stream lines =
new BufferedReader(new InputStreamReader(inputStream, StandardCharsets.UTF_8)).lines()) {
return lines.collect(Collectors.joining("\n"));
}
}
public static void processStream(InputStream inputStream, Consumer logFunction) {
try (BufferedReader bufferedReader =
new BufferedReader(new InputStreamReader(inputStream, StandardCharsets.UTF_8))) {
String line;
while ((line = bufferedReader.readLine()) != null) {
logFunction.accept(line);
}
} catch (IOException e) {
throw new RuntimeException("Failed to write inputStream", e);
}
}
private CommandRunner() {}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy