43 lines
2.1 KiB
Java
43 lines
2.1 KiB
Java
import java.time.Duration;
|
||
import java.util.concurrent.TimeUnit;
|
||
|
||
/** Source-launcher startup probe; uses the service's configured DOCKER_HOST. */
|
||
class WaitForDocker {
|
||
/**
|
||
* Waits for the configured Docker daemon before allowing the runner service to start.
|
||
*
|
||
* @param args optional timeout in seconds, from 1 through 300; defaults to 120
|
||
* @throws Exception if the timeout is invalid or the Docker probe cannot run
|
||
* @implNote Exits nonzero if Docker does not become ready before the deadline.
|
||
*/
|
||
public static void main(final String... args) throws Exception {
|
||
long seconds = args.length == 0 ? 120 : Long.parseLong(args[0]);
|
||
if (seconds < 1 || seconds > 300) throw new IllegalArgumentException("Timeout must be 1–300 seconds");
|
||
long deadline = System.nanoTime() + Duration.ofSeconds(seconds).toNanos();
|
||
IO.println("Waiting for the configured Docker engine...");
|
||
while (System.nanoTime() < deadline) {
|
||
Process process = new ProcessBuilder("docker", "info", "--format", "{{.ServerVersion}}")
|
||
.redirectOutput(ProcessBuilder.Redirect.DISCARD)
|
||
.redirectError(ProcessBuilder.Redirect.DISCARD)
|
||
.start();
|
||
try {
|
||
long remaining = Math.max(1, deadline - System.nanoTime());
|
||
if (process.waitFor(Math.min(remaining, TimeUnit.SECONDS.toNanos(5)), TimeUnit.NANOSECONDS)
|
||
&& process.exitValue() == 0) {
|
||
IO.println("Docker is available; starting the Gitea runner.");
|
||
return;
|
||
}
|
||
} finally {
|
||
if (process.isAlive()) {
|
||
process.destroyForcibly();
|
||
process.waitFor();
|
||
}
|
||
}
|
||
long remaining = deadline - System.nanoTime();
|
||
if (remaining > 0) TimeUnit.NANOSECONDS.sleep(Math.min(remaining, TimeUnit.SECONDS.toNanos(2)));
|
||
}
|
||
System.err.println("Docker did not become available within " + seconds + " seconds; runner not started.");
|
||
System.exit(1);
|
||
}
|
||
}
|