Fix compile issue

neo4j · Oct 2, 2020 · 21a8f43 · 21a8f43
1 parent 20780dd
commit 21a8f43
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/driver/src/main/java/org/neo4j/driver/internal/retry/ExponentialBackoffRetryLogic.java b/driver/src/main/java/org/neo4j/driver/internal/retry/ExponentialBackoffRetryLogic.java
@@ -20,8 +20,6 @@
 
 import io.netty.util.concurrent.EventExecutor;
 import io.netty.util.concurrent.EventExecutorGroup;
-
-import org.neo4j.driver.util.Experimental;
 import org.reactivestreams.Publisher;
 import reactor.core.publisher.Flux;
 import reactor.core.publisher.Mono;
@@ -47,6 +45,7 @@
 import org.neo4j.driver.exceptions.TransientException;
 import org.neo4j.driver.internal.util.Clock;
 import org.neo4j.driver.internal.util.Futures;
+import org.neo4j.driver.util.Experimental;
 
 import static java.util.concurrent.TimeUnit.SECONDS;
 
@@ -106,7 +105,7 @@ public <T> T retry( Supplier<T> work )
             catch ( Throwable throwable )
             {
                 Throwable error = extractPossibleTerminationCause( throwable );
-                if ( isRetryable( error ) )
+                if ( canRetryOn( error ) )
                 {
                     long currentTime = clock.millis();
                     if ( startTime == -1 )
@@ -148,6 +147,11 @@ public <T> Publisher<T> retryRx( Publisher<T> work )
         return Flux.from( work ).retryWhen( retryRxCondition() );
     }
 
+    protected boolean canRetryOn( Throwable error )
+    {
+        return isRetryable( error );
+    }
+
     @Experimental
     public static boolean isRetryable( Throwable error )
     {
@@ -186,7 +190,7 @@ private Function<Flux<Throwable>,Publisher<Context>> retryRxCondition()
             long startTime = ctx.getOrDefault( "startTime", -1L );
             long nextDelayMs = ctx.getOrDefault( "nextDelayMs", initialRetryDelayMs );
 
-            if ( isRetryable( error ) )
+            if ( canRetryOn( error ) )
             {
                 long currentTime = clock.millis();
                 if ( startTime == -1 )
@@ -273,7 +277,7 @@ private <T> void retryOnError( CompletableFuture<T> resultFuture, Supplier<Compl
             List<Throwable> errors )
     {
         Throwable error = extractPossibleTerminationCause( throwable );
-        if ( isRetryable( error ) )
+        if ( canRetryOn( error ) )
         {
             long currentTime = clock.millis();
             if ( startTime == -1 )