diff options
Diffstat (limited to 'container-messagebus/src/main/java/com/yahoo/messagebus/jdisc/test/MessageQueue.java')
-rw-r--r-- | container-messagebus/src/main/java/com/yahoo/messagebus/jdisc/test/MessageQueue.java | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/container-messagebus/src/main/java/com/yahoo/messagebus/jdisc/test/MessageQueue.java b/container-messagebus/src/main/java/com/yahoo/messagebus/jdisc/test/MessageQueue.java new file mode 100644 index 00000000000..c5287165e27 --- /dev/null +++ b/container-messagebus/src/main/java/com/yahoo/messagebus/jdisc/test/MessageQueue.java @@ -0,0 +1,27 @@ +// 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.Message; +import com.yahoo.messagebus.MessageHandler; + +import java.util.concurrent.BlockingQueue; +import java.util.concurrent.LinkedBlockingQueue; +import java.util.concurrent.TimeUnit; + +/** + * @author Simon Thoresen Hult + */ +public class MessageQueue implements MessageHandler { + + private final BlockingQueue<Message> queue = new LinkedBlockingQueue<>(); + + @Override + public void handleMessage(Message msg) { + queue.add(msg); + } + + public Message awaitMessage(int timeout, TimeUnit unit) throws InterruptedException { + return queue.poll(timeout, unit); + } + +} |