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