diff --git a/tchannel-core/src/main/java/com/uber/tchannel/api/TFuture.java b/tchannel-core/src/main/java/com/uber/tchannel/api/TFuture.java index dcbb893..2713b99 100644 --- a/tchannel-core/src/main/java/com/uber/tchannel/api/TFuture.java +++ b/tchannel-core/src/main/java/com/uber/tchannel/api/TFuture.java @@ -41,6 +41,7 @@ import java.util.concurrent.ExecutionException; import java.util.concurrent.Executor; import java.util.concurrent.TimeUnit; +import java.util.concurrent.TimeoutException; import java.util.concurrent.atomic.AtomicInteger; /** @@ -162,8 +163,9 @@ public V get() throws InterruptedException, ExecutionException { } @Override - public V get(long timeout, TimeUnit unit) { - throw new UnsupportedOperationException("Get timeout is unsupported. Use request timeout instead."); + public V get(long timeout, TimeUnit unit) throws InterruptedException, ExecutionException, TimeoutException { + listenerCount.incrementAndGet(); + return super.get(timeout, unit); } @Override