Async driver request timeout (#387)

* async driver request timeout

* CompletableFutureUtils refactoring

* test fixes

* test fixes

Author: rashtao

src/main/java/com/arangodb/async/internal/utils/CompletableFutureUtils.java

@@ -0,0 +1,29 @@
+package com.arangodb.async.internal.utils;
+
+import java.util.concurrent.CompletableFuture;
+import java.util.concurrent.Executors;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.ScheduledFuture;
+import java.util.concurrent.TimeUnit;
+import java.util.concurrent.TimeoutException;
+
+public class CompletableFutureUtils {
+
+    private CompletableFutureUtils() {
+    }
+
+    private static final ScheduledExecutorService timeoutScheduler = Executors.newSingleThreadScheduledExecutor(r -> {
+                Thread t = Executors.defaultThreadFactory().newThread(r);
+                t.setDaemon(true);
+                return t;
+            }
+    );
+
+    public static <T> CompletableFuture<T> orTimeout(CompletableFuture<T> completableFuture, long timeout, TimeUnit unit) {
+        ScheduledFuture<?> timeoutTask = timeoutScheduler.schedule(() ->
+                completableFuture.completeExceptionally(new TimeoutException()), timeout, unit);
+        completableFuture.whenComplete((v, e) -> timeoutTask.cancel(false));
+        return completableFuture;
+    }
+
+}

src/main/java/com/arangodb/async/internal/velocystream/VstConnectionAsync.java

@@ -20,6 +20,7 @@
 
 package com.arangodb.async.internal.velocystream;
 
+import com.arangodb.async.internal.utils.CompletableFutureUtils;
 import com.arangodb.internal.net.HostDescription;
 import com.arangodb.internal.velocystream.internal.Chunk;
 import com.arangodb.internal.velocystream.internal.Message;
@@ -30,6 +31,7 @@
 import java.util.Collection;
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.FutureTask;
+import java.util.concurrent.TimeUnit;
 
 /**
  * @author Mark Vollmary
@@ -54,7 +56,11 @@ public synchronized CompletableFuture<Message> write(final Message message, fina
         });
         messageStore.storeMessage(message.getId(), task);
         super.writeIntern(message, chunks);
-        return future;
+        if (timeout == null || timeout == 0L) {
+            return future;
+        } else {
+            return CompletableFutureUtils.orTimeout(future, timeout, TimeUnit.MILLISECONDS);
+        }
     }
 
     @Override

src/test/resources/arangodb.properties

@@ -2,3 +2,4 @@ arangodb.hosts=172.28.3.1:8529
 arangodb.connections.max=1
 arangodb.acquireHostList=true
 arangodb.password=test
+arangodb.timeout=30000