diff options
Diffstat (limited to 'container-messagebus/src/main/java/com/yahoo/messagebus/jdisc/test/ReplyQueue.java')
-rw-r--r-- | container-messagebus/src/main/java/com/yahoo/messagebus/jdisc/test/ReplyQueue.java | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/container-messagebus/src/main/java/com/yahoo/messagebus/jdisc/test/ReplyQueue.java b/container-messagebus/src/main/java/com/yahoo/messagebus/jdisc/test/ReplyQueue.java new file mode 100644 index 00000000000..6c48aab5a7f --- /dev/null +++ b/container-messagebus/src/main/java/com/yahoo/messagebus/jdisc/test/ReplyQueue.java @@ -0,0 +1,26 @@ +// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. +package com.yahoo.messagebus.jdisc.test; + +import com.yahoo.messagebus.Reply; +import com.yahoo.messagebus.ReplyHandler; + +import java.util.concurrent.BlockingQueue; +import java.util.concurrent.LinkedBlockingQueue; +import java.util.concurrent.TimeUnit; + +/** + * @author Simon Thoresen Hult + */ +public class ReplyQueue implements ReplyHandler { + + private final BlockingQueue<Reply> queue = new LinkedBlockingQueue<>(); + + @Override + public void handleReply(Reply reply) { + queue.add(reply); + } + + public Reply awaitReply(int timeout, TimeUnit unit) throws InterruptedException { + return queue.poll(timeout, unit); + } +} |