Merge pull request #991: [proxima-beam-core] configurable finish bundle timeout for ProximaIO

Author: je-ik

beam/core/src/main/java/cz/o2/proxima/beam/io/ProximaIO.java

@@ -122,7 +122,8 @@ public void startBundle() {
     public void finishBundle() {
       long startTime = System.currentTimeMillis();
       while (missingResponses.get() > 0) {
-        if (System.currentTimeMillis() - startTime > bundleFinalizeTimeoutMs) {
+        long elapsed = System.currentTimeMillis() - startTime;
+        if (elapsed >= bundleFinalizeTimeoutMs) {
           throw new IllegalStateException("Failed to flush bundle within timeout of 5s");
         }
         // clone to avoid ConcurrentModificationException
@@ -136,7 +137,7 @@ public void finishBundle() {
                 .map(
                     f ->
                         ExceptionUtils.uncheckedFactory(
-                            () -> f.get(bundleFinalizeTimeoutMs, TimeUnit.MILLISECONDS)))
+                            () -> f.get(bundleFinalizeTimeoutMs - elapsed, TimeUnit.MILLISECONDS)))
                 .filter(p -> !p.getFirst())
                 // this will be retried
                 .filter(p -> !(p.getSecond() instanceof TransactionRejectedException))