com.tvd12.ezydata.elasticsearch.concurrent.EzyEsActionQueue Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ezydata-elasticsearch Show documentation

The newest version!

package com.tvd12.ezydata.elasticsearch.concurrent;

import com.tvd12.ezydata.elasticsearch.action.EzyEsActionWrapper;
import com.tvd12.ezydata.elasticsearch.exception.EzyEsActionQueueFullException;

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

public class EzyEsActionQueue {

    protected final int capacity;
    protected final BlockingQueue queue;

    public EzyEsActionQueue(int capacity) {
        this.capacity = capacity;
        this.queue = new LinkedBlockingQueue<>(capacity);
    }

    public void addAction(EzyEsActionWrapper wrapper) {
        boolean success = this.queue.offer(wrapper);
        if (!success) {
            throw new EzyEsActionQueueFullException(capacity, queue.size());
        }
    }

    public EzyEsActionWrapper takeAction() throws Exception {
        return this.queue.take();
    }
}