Make pauseBeforeReconnect overridable #41

OpenHFT · Nov 22, 2018 · 3a4980b · 3a4980b
1 parent 179e335
commit 3a4980b
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/src/main/java/net/openhft/chronicle/network/ConnectionStrategy.java b/src/main/java/net/openhft/chronicle/network/ConnectionStrategy.java
@@ -120,13 +120,17 @@ default SocketChannel openSocketChannel(@NotNull InetSocketAddress socketAddress
                 return sc;
             Thread.yield();
             // If nothing is listening, socketChannel returns pretty much immediately so we support a pause here
-            long pauseMillis = (startMs + pauseMillisBeforeReconnect()) - System.currentTimeMillis();
-            if (Jvm.isDebugEnabled(this.getClass()))
-                Jvm.debug().on(this.getClass(), "Waiting for reconnect "+pauseMillis+" ms");
-            LockSupport.parkNanos(TimeUnit.MILLISECONDS.toNanos(pauseMillis));
+            pauseBeforeReconnect(startMs);
         }
     }
 
+    default void pauseBeforeReconnect(long startMs) {
+        long pauseMillis = (startMs + pauseMillisBeforeReconnect()) - System.currentTimeMillis();
+        if (Jvm.isDebugEnabled(this.getClass()))
+            Jvm.debug().on(this.getClass(), "Waiting for reconnect "+pauseMillis+" ms");
+        LockSupport.parkNanos(TimeUnit.MILLISECONDS.toNanos(pauseMillis));
+    }
+
     /**
      * allows control of a backoff strategy
      *