add timeout

aerospike · Jul 21, 2024 · ee5edf6 · ee5edf6
1 parent 8e944ad
commit ee5edf6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/src/main/java/org/springframework/data/aerospike/util/InfoCommandUtils.java b/src/main/java/org/springframework/data/aerospike/util/InfoCommandUtils.java
@@ -11,6 +11,7 @@
 import java.util.Map;
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.CompletionException;
+import java.util.concurrent.TimeUnit;
 
 @Slf4j
 @UtilityClass
@@ -48,7 +49,7 @@ public void onFailure(AerospikeException ae) {
 
         String value = null;
         try {
-            value = listener.getValueFuture().join();
+            value = listener.getValueFuture().orTimeout(infoPolicy.timeout, TimeUnit.MILLISECONDS).join();
         } catch (CompletionException ce) {
             fail(command, ce.getCause());
         }

diff --git a/src/test/java/org/springframework/data/aerospike/util/IndexUtils.java b/src/test/java/org/springframework/data/aerospike/util/IndexUtils.java
@@ -44,8 +44,7 @@ static void createIndex(IAerospikeClient client, ServerVersionSupport serverVers
 
     public static List<Index> getIndexes(IAerospikeClient client, String namespace, IndexInfoParser indexInfoParser) {
         Node node = client.getCluster().getRandomNode();
-        String response = InfoCommandUtils.request(client, node, "sindex-list:ns=" + namespace + ";b64=true"
-        );
+        String response = InfoCommandUtils.request(client, node, "sindex-list:ns=" + namespace + ";b64=true");
         return Arrays.stream(response.split(";"))
             .map(indexInfoParser::parse)
             .collect(Collectors.toList());
@@ -57,8 +56,7 @@ public static List<Index> getIndexes(IAerospikeClient client, String namespace,
      */
     public static boolean indexExists(IAerospikeClient client, String namespace, String indexName) {
         Node node = client.getCluster().getRandomNode();
-        String response = InfoCommandUtils.request(client, node, "sindex/" + namespace + '/' + indexName
-        );
+        String response = InfoCommandUtils.request(client, node, "sindex/" + namespace + '/' + indexName);
         return !response.startsWith("FAIL:201");
     }