From 87badeacebd9da11bbf463815258273e44d10292 Mon Sep 17 00:00:00 2001
From: Cedric Champeau <cedric.champeau@oracle.com>
Date: Tue, 21 May 2024 17:52:26 +0200
Subject: [PATCH] Add exponential backoff to Maven plugin too

---
 docs/src/docs/asciidoc/index.adoc                  |  4 ++++
 .../buildtools/maven/AbstractNativeMojo.java       | 14 +++++++++++++-
 2 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/docs/src/docs/asciidoc/index.adoc b/docs/src/docs/asciidoc/index.adoc
index 5a1ee3912..3465c591e 100644
--- a/docs/src/docs/asciidoc/index.adoc
+++ b/docs/src/docs/asciidoc/index.adoc
@@ -25,6 +25,10 @@ If you are using alternative build systems, see <<alternative-build-systems.adoc
 
 - Add retries when downloading the metadata repository when using a URL directly
 
+==== Maven plugin
+
+- Add retries when downloading the metadata repository when using a URL directly
+
 === Release 0.10.1
 
 - Mark additional JUnit 5 types for build-time initialization for compatibility with Native Image's `--strict-image-heap` option.
diff --git a/native-maven-plugin/src/main/java/org/graalvm/buildtools/maven/AbstractNativeMojo.java b/native-maven-plugin/src/main/java/org/graalvm/buildtools/maven/AbstractNativeMojo.java
index bfe70a5d5..6ceb4736c 100644
--- a/native-maven-plugin/src/main/java/org/graalvm/buildtools/maven/AbstractNativeMojo.java
+++ b/native-maven-plugin/src/main/java/org/graalvm/buildtools/maven/AbstractNativeMojo.java
@@ -59,6 +59,7 @@
 import org.eclipse.aether.resolution.DependencyResult;
 import org.graalvm.buildtools.VersionInfo;
 import org.graalvm.buildtools.maven.config.MetadataRepositoryConfiguration;
+import org.graalvm.buildtools.utils.ExponentialBackoff;
 import org.graalvm.buildtools.utils.FileUtils;
 import org.graalvm.reachability.DirectoryConfiguration;
 import org.graalvm.reachability.GraalVMReachabilityMetadataRepository;
@@ -73,6 +74,7 @@
 import java.net.URL;
 import java.nio.file.Files;
 import java.nio.file.Path;
+import java.time.Duration;
 import java.util.HashSet;
 import java.util.Optional;
 import java.util.Set;
@@ -104,6 +106,12 @@ public abstract class AbstractNativeMojo extends AbstractMojo {
     @Parameter(alias = "metadataRepository")
     protected MetadataRepositoryConfiguration metadataRepositoryConfiguration;
 
+    @Parameter(defaultValue = "3")
+    protected int metadataRepositoryMaxRetries;
+
+    @Parameter(defaultValue = "100")
+    protected int metadataRepositoryInitialBackoffMillis;
+
     protected final Set<DirectoryConfiguration> metadataRepositoryConfigurations;
 
     protected GraalVMReachabilityMetadataRepository metadataRepository;
@@ -207,7 +215,11 @@ private Path getRepo(Path destinationRoot) {
                 }
             }
 
-            return downloadMetadataRepo(destinationRoot, targetUrl);
+            URL finalTargetUrl = targetUrl;
+            return ExponentialBackoff.get()
+                .withMaxRetries(metadataRepositoryMaxRetries)
+                .withInitialWaitPeriod(Duration.ofMillis(metadataRepositoryInitialBackoffMillis))
+                .supply(() -> downloadMetadataRepo(destinationRoot, finalTargetUrl));
         }
     }