1 /* 2 * Copyright (c) 2020, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. Oracle designates this 8 * particular file as subject to the "Classpath" exception as provided 9 * by Oracle in the LICENSE file that accompanied this code. 10 * 11 * This code is distributed in the hope that it will be useful, but WITHOUT 12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 14 * version 2 for more details (a copy is included in the LICENSE file that 15 * accompanied this code). 16 * 17 * You should have received a copy of the GNU General Public License version 18 * 2 along with this work; if not, write to the Free Software Foundation, 19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 20 * 21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 22 * or visit www.oracle.com if you need additional information or have any 23 * questions. 24 */ 25 package jdk.incubator.jpackage.internal; 26 27 import java.io.IOException; 28 import java.util.function.Consumer; 29 import java.util.function.Supplier; 30 31 public final class RetryExecutor { 32 RetryExecutor() { 33 setMaxAttemptsCount(5); 34 setAttemptTimeoutMillis(2 * 1000); 35 } 36 37 RetryExecutor setMaxAttemptsCount(int v) { 38 attempts = v; 39 return this; 40 } 41 42 RetryExecutor setAttemptTimeoutMillis(int v) { 43 timeoutMillis = v; 44 return this; 45 } 46 47 RetryExecutor setExecutorInitializer(Consumer<Executor> v) { 48 executorInitializer = v; 49 return this; 50 } 51 52 void abort() { 53 aborted = true; 54 } 55 56 static RetryExecutor retryOnKnownErrorMessage(String v) { 57 RetryExecutor result = new RetryExecutor(); 58 return result.setExecutorInitializer(exec -> { 59 exec.setOutputConsumer(output -> { 60 if (!output.anyMatch(v::equals)) { 61 result.abort(); 62 } 63 }); 64 }); 65 } 66 67 void execute(String cmdline[]) throws IOException { 68 executeLoop(() -> Executor.of(cmdline)); 69 } 70 71 void execute(ProcessBuilder pb) throws IOException { 72 executeLoop(() -> Executor.of(pb)); 73 } 74 75 private void executeLoop(Supplier<Executor> execSupplier) throws IOException { 76 aborted = false; 77 for (;;) { 78 try { 79 Executor exec = execSupplier.get(); 80 if (executorInitializer != null) { 81 executorInitializer.accept(exec); 82 } 83 exec.executeExpectSuccess(); 84 break; 85 } catch (IOException ex) { 86 if (aborted || (--attempts) <= 0) { 87 throw ex; 88 } 89 } 90 91 try { 92 Thread.sleep(timeoutMillis); 93 } catch (InterruptedException ex) { 94 Log.verbose(ex); 95 throw new RuntimeException(ex); 96 } 97 } 98 } 99 100 private Consumer<Executor> executorInitializer; 101 private boolean aborted; 102 private int attempts; 103 private int timeoutMillis; 104 }