FIX: getElement logic in BTreeStoreAndGetFuture.

naver · Sep 11, 2023 · 0398941 · 0398941
1 parent b4bec20
commit 0398941
Showing 1 changed file with 13 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,8 +17,12 @@
 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.OperationTimeoutException;
 import net.spy.memcached.collection.Element;
 import net.spy.memcached.ops.CollectionGetOpCallback;
 
@@ -41,6 +45,15 @@ public BTreeStoreAndGetFuture(CountDownLatch l, AtomicReference<T> oref, long op
   }
 
   public Element<E> getElement() {
+    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 OperationTimeoutException(e);
+    }
     CollectionGetOpCallback callback = (CollectionGetOpCallback) op.getCallback();
     callback.addResult();
     return element;