From 5e7fe7ae76790a775bc8dac88d6fc0b1ebc937a6 Mon Sep 17 00:00:00 2001
From: Ryan McCaffrey <ryan.mccaffrey.42@gmail.com>
Date: Mon, 7 Aug 2017 17:04:09 -0400
Subject: [PATCH] Update OperationFuture.java

Adds a method for `getStatus()` that takes expiration duration and TimeUnit parameters to be used with the the parametrized version of `get()`.
---
 .../memcached/internal/OperationFuture.java   | 26 +++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/src/main/java/net/spy/memcached/internal/OperationFuture.java b/src/main/java/net/spy/memcached/internal/OperationFuture.java
index 3a7f8cca8..88ab12052 100644
--- a/src/main/java/net/spy/memcached/internal/OperationFuture.java
+++ b/src/main/java/net/spy/memcached/internal/OperationFuture.java
@@ -254,6 +254,32 @@ public OperationStatus getStatus() {
     }
     return status;
   }
+  
+  /**
+   * Get the current status of this operation.
+   *
+   * Note that the operation status may change as the operation is tried and
+   * potentially retried against the servers specified by the NodeLocator.
+   *
+   * The interrupted status of the current thread is cleared by this method.
+   * Inspect the returned OperationStatus to check whether an interruption has taken place.
+   *
+   * @param duration amount of time to wait
+   * @param units unit of time to wait
+   * @return OperationStatus
+   */
+  public OperationStatus getStatus(long duration, TimeUnit units) {
+    if (status == null) {
+      try {
+        get(duration, units);
+      } catch (InterruptedException e) {
+        status = new OperationStatus(false, "Interrupted", StatusCode.INTERRUPTED);
+      } catch (ExecutionException e) {
+        getLogger().warn("Error getting status of operation", e);
+      }
+    }
+    return status;
+  }
 
   /**
    * Set the Operation associated with this OperationFuture.