Propagate TimeoutExceptions from OperationFutures as OperationTimeoutExceptions

fixes awslabs#48

Author: Jovan Attisha 🥃

src/main/java/net/spy/memcached/MemcachedClient.java

@@ -923,8 +923,10 @@ public <T> CASResponse cas(String key, long casId, int exp, T value,
     } catch (InterruptedException e) {
       throw new RuntimeException("Interrupted waiting for value", e);
     } catch (ExecutionException e) {
-      if(e.getCause() instanceof CancellationException) {
+      if (e.getCause() instanceof CancellationException) {
         throw (CancellationException) e.getCause();
+      } else if (e.getCause() instanceof TimeoutException) {
+        throw new OperationTimeoutException("Timeout waiting for value", e);
       } else {
         throw new RuntimeException("Exception waiting for value", e);
       }
@@ -1386,8 +1388,10 @@ public <T> CASValue<T> gets(String key, Transcoder<T> tc) {
     } catch (InterruptedException e) {
       throw new RuntimeException("Interrupted waiting for value", e);
     } catch (ExecutionException e) {
-      if(e.getCause() instanceof CancellationException) {
+      if (e.getCause() instanceof CancellationException) {
         throw (CancellationException) e.getCause();
+      } else if (e.getCause() instanceof TimeoutException) {
+        throw new OperationTimeoutException("Timeout waiting for value", e);
       } else {
         throw new RuntimeException("Exception waiting for value", e);
       }
@@ -1479,8 +1483,10 @@ public <T> T get(String key, Transcoder<T> tc) {
     } catch (InterruptedException e) {
       throw new RuntimeException("Interrupted waiting for value", e);
     } catch (ExecutionException e) {
-      if(e.getCause() instanceof CancellationException) {
+      if (e.getCause() instanceof CancellationException) {
         throw (CancellationException) e.getCause();
+      } else if (e.getCause() instanceof TimeoutException) {
+          throw new OperationTimeoutException("Timeout waiting for value", e);
       } else {
         throw new RuntimeException("Exception waiting for value", e);
       }
@@ -1807,8 +1813,10 @@ public <T> Map<String, T> getBulk(Iterator<String> keyIter,
     } catch (InterruptedException e) {
       throw new RuntimeException("Interrupted getting bulk values", e);
     } catch (ExecutionException e) {
-      if(e.getCause() instanceof CancellationException) {
+      if (e.getCause() instanceof CancellationException) {
         throw (CancellationException) e.getCause();
+      } else if (e.getCause() instanceof TimeoutException) {
+        throw new OperationTimeoutException("Timeout waiting for bulk values", e);
       } else {
         throw new RuntimeException("Exception waiting for bulk values", e);
       }
@@ -2389,8 +2397,10 @@ private long mutateWithDefault(Mutator t, String key, long by, long def,
       } catch (InterruptedException e) {
         throw new RuntimeException("Interrupted waiting for store", e);
       } catch (ExecutionException e) {
-        if(e.getCause() instanceof CancellationException) {
+        if (e.getCause() instanceof CancellationException) {
           throw (CancellationException) e.getCause();
+        } else if (e.getCause() instanceof TimeoutException) {
+          throw new OperationTimeoutException("Timeout waiting for store", e);
         } else {
           throw new RuntimeException("Failed waiting for store", e);
         }