Andy Wilkinson
2 years ago
commit
a69cf002f0
@ -0,0 +1,129 @@
|
||||
/*
|
||||
* Copyright 2012-2022 the original author or authors.
|
||||
*
|
||||
* 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
|
||||
*
|
||||
* https://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 org.springframework.boot.build.docs;
|
||||
|
||||
import java.io.File;
|
||||
import java.io.IOException;
|
||||
import java.nio.file.Files;
|
||||
import java.nio.file.Path;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
import org.gradle.api.DefaultTask;
|
||||
import org.gradle.api.Task;
|
||||
import org.gradle.api.file.FileCollection;
|
||||
import org.gradle.api.file.RegularFileProperty;
|
||||
import org.gradle.api.provider.ListProperty;
|
||||
import org.gradle.api.provider.Property;
|
||||
import org.gradle.api.tasks.Classpath;
|
||||
import org.gradle.api.tasks.Input;
|
||||
import org.gradle.api.tasks.OutputFile;
|
||||
import org.gradle.api.tasks.TaskAction;
|
||||
import org.gradle.internal.jvm.Jvm;
|
||||
|
||||
/**
|
||||
* {@link Task} to run an application for the purpose of capturing its output for
|
||||
* inclusion in the reference documentation.
|
||||
*
|
||||
* @author Andy Wilkinson
|
||||
*/
|
||||
public class ApplicationRunner extends DefaultTask {
|
||||
|
||||
private final RegularFileProperty output = getProject().getObjects().fileProperty();
|
||||
|
||||
private final ListProperty<String> args = getProject().getObjects().listProperty(String.class);
|
||||
|
||||
private final Property<String> mainClass = getProject().getObjects().property(String.class);
|
||||
|
||||
private final Property<String> expectedLogging = getProject().getObjects().property(String.class);
|
||||
|
||||
private FileCollection classpath;
|
||||
|
||||
@OutputFile
|
||||
public RegularFileProperty getOutput() {
|
||||
return this.output;
|
||||
}
|
||||
|
||||
@Classpath
|
||||
public FileCollection getClasspath() {
|
||||
return this.classpath;
|
||||
}
|
||||
|
||||
public void setClasspath(FileCollection classpath) {
|
||||
this.classpath = classpath;
|
||||
}
|
||||
|
||||
@Input
|
||||
public ListProperty<String> getArgs() {
|
||||
return this.args;
|
||||
}
|
||||
|
||||
@Input
|
||||
public Property<String> getMainClass() {
|
||||
return this.mainClass;
|
||||
}
|
||||
|
||||
@Input
|
||||
public Property<String> getExpectedLogging() {
|
||||
return this.expectedLogging;
|
||||
}
|
||||
|
||||
@TaskAction
|
||||
void runApplication() throws IOException {
|
||||
List<String> command = new ArrayList<>();
|
||||
File executable = Jvm.current().getExecutable("java");
|
||||
command.add(executable.getAbsolutePath());
|
||||
command.add("-cp");
|
||||
command.add(this.classpath.getFiles().stream().map(File::getAbsolutePath)
|
||||
.collect(Collectors.joining(File.pathSeparator)));
|
||||
command.add(this.mainClass.get());
|
||||
command.addAll(this.args.get());
|
||||
File outputFile = this.output.getAsFile().get();
|
||||
Process process = new ProcessBuilder().redirectOutput(outputFile).redirectError(outputFile).command(command)
|
||||
.start();
|
||||
awaitLogging(process);
|
||||
process.destroy();
|
||||
}
|
||||
|
||||
private void awaitLogging(Process process) {
|
||||
long end = System.currentTimeMillis() + 30000;
|
||||
String expectedLogging = this.expectedLogging.get();
|
||||
while (System.currentTimeMillis() < end) {
|
||||
for (String line : outputLines()) {
|
||||
if (line.contains(expectedLogging)) {
|
||||
return;
|
||||
}
|
||||
}
|
||||
if (!process.isAlive()) {
|
||||
throw new IllegalStateException("Process exited before '" + expectedLogging + "' was logged");
|
||||
}
|
||||
}
|
||||
throw new IllegalStateException("'" + expectedLogging + "' was not logged within 30 seconds");
|
||||
}
|
||||
|
||||
private List<String> outputLines() {
|
||||
Path outputPath = this.output.get().getAsFile().toPath();
|
||||
try {
|
||||
return Files.readAllLines(outputPath);
|
||||
}
|
||||
catch (IOException ex) {
|
||||
throw new RuntimeException("Failed to read lines of output from '" + outputPath + "'", ex);
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
Loading…
Reference in New Issue