FifoMap.java

package kg.apc.jmeter.modifiers;

import org.apache.jmeter.util.JMeterUtils;

import java.util.concurrent.BlockingQueue;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.TimeUnit;

public class FifoMap extends ConcurrentHashMap<String, BlockingQueue<Object>> {

    public static final String TIMEOUT_PROP = "kg.apc.jmeter.functions.FifoTimeout";
    public static final String CAPACITY_PROP = "kg.apc.jmeter.functions.FifoCapacity";
    private static FifoMap instance = new FifoMap();

    private FifoMap() {
    }

    public static FifoMap getInstance() {
        return instance;
    }

    private BlockingQueue<Object> getFifo(String fifoName) {
        if (super.containsKey(fifoName)) {
            return super.get(fifoName);
        } else {
            BlockingQueue<Object> fifo = new LinkedBlockingQueue<Object>(JMeterUtils.getPropDefault(FifoMap.CAPACITY_PROP, Integer.MAX_VALUE));
            super.put(fifoName, fifo);
            return fifo;
        }
    }

    public Object get(String fifoName) {
        BlockingQueue<Object> fifo = getFifo(fifoName);
        Object value;
        value = fifo.peek();
        return value == null ? "" : value;
    }

    public Object pop(String fifoName, long timeout) throws InterruptedException {
        BlockingQueue<Object> fifo = getFifo(fifoName);
        return fifo.poll(timeout, TimeUnit.SECONDS);
    }

    public Object pop(String fifoName) throws InterruptedException {
        return pop(fifoName, Long.MAX_VALUE);
    }

    public int length(String fifoName) {
        BlockingQueue<Object> fifo = getFifo(fifoName);
        return fifo.size();
    }

    public void put(String fifoName, Object v) throws InterruptedException {
        getFifo(fifoName).offer(v, Long.MAX_VALUE, TimeUnit.SECONDS);
    }
}