diff options
Diffstat (limited to 'vespaclient-core/src/main/java/com/yahoo/feedapi/SingleSender.java')
-rwxr-xr-x | vespaclient-core/src/main/java/com/yahoo/feedapi/SingleSender.java | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/vespaclient-core/src/main/java/com/yahoo/feedapi/SingleSender.java b/vespaclient-core/src/main/java/com/yahoo/feedapi/SingleSender.java index e0e12b26ae6..6f9d539855e 100755 --- a/vespaclient-core/src/main/java/com/yahoo/feedapi/SingleSender.java +++ b/vespaclient-core/src/main/java/com/yahoo/feedapi/SingleSender.java @@ -19,16 +19,18 @@ public class SingleSender implements SimpleFeedAccess { private final SharedSender.ResultCallback owner; private final SharedSender sender; private final List<MessageProcessor> messageProcessors = new ArrayList<>(); + private final long timeofDoom; private boolean blockingQueue; - public SingleSender(SharedSender.ResultCallback owner, SharedSender sender, boolean blockingQueue) { + public SingleSender(SharedSender.ResultCallback owner, SharedSender sender, long timeoutMS, boolean blockingQueue) { this.owner = owner; this.sender = sender; + timeofDoom = System.currentTimeMillis() + timeoutMS; this.blockingQueue = blockingQueue; } - public SingleSender(SharedSender.ResultCallback owner, SharedSender sender) { - this(owner, sender, true); + public SingleSender(SharedSender.ResultCallback owner, int timeoutMS, SharedSender sender) { + this(owner, sender, timeoutMS, true); } @Override @@ -103,4 +105,9 @@ public class SingleSender implements SimpleFeedAccess { @Override public void close() { } + + @Override + boolean hasTimedOut() { + return System.currentTimeMillis() > timeofDoom; + } } |