FIX: getElement logic in BTreeStoreAndGetFuture.

naver · Sep 4, 2023 · 04ffb09 · 04ffb09
1 parent 4318361
commit 04ffb09
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/main/java/net/spy/memcached/internal/BTreeStoreAndGetFuture.java b/src/main/java/net/spy/memcached/internal/BTreeStoreAndGetFuture.java
@@ -17,6 +17,9 @@
 package net.spy.memcached.internal;
 
 import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.ExecutionException;
+import java.util.concurrent.TimeUnit;
+import java.util.concurrent.TimeoutException;
 import java.util.concurrent.atomic.AtomicReference;
 
 import net.spy.memcached.collection.Element;
@@ -42,6 +45,15 @@ public BTreeStoreAndGetFuture(CountDownLatch l, AtomicReference<T> oref, long op
 
   public Element<E> getElement() {
     CollectionGetOpCallback callback = (CollectionGetOpCallback) op.getCallback();
+    try {
+      super.get(super.timeout, TimeUnit.MILLISECONDS);
+    } catch (ExecutionException e) {
+      throw new RuntimeException(e);
+    } catch (InterruptedException e) {
+      throw new RuntimeException(e);
+    } catch (TimeoutException e) {
+      throw new RuntimeException(e);
+    }
     callback.addResult();
     return element;
   }