org.opencb.commons.containers.DataRW Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of commons-lib Show documentation
There is a newer version: 6.0.0
package org.opencb.commons.containers;

import java.util.concurrent.BlockingQueue;
import java.util.concurrent.PriorityBlockingQueue;
import java.util.concurrent.TimeUnit;

/**
 * Created with IntelliJ IDEA.
 * User: aaleman
 * Date: 10/7/13
 * Time: 9:52 AM
 * To change this template use File | Settings | File Templates.
 */
public class DataRW {

    private BlockingQueue> read;
    private BlockingQueue> write;
    private int numProducersConsumers;
    private boolean continueProducing = true;
    private int maxCapacity = 10;
    private int currentElement = 0;

    public DataRW() {
        read = new PriorityBlockingQueue<>(maxCapacity);
        write = new PriorityBlockingQueue<>(maxCapacity);
        numProducersConsumers = 0;
    }

    public void putRead(int i, R batch) {

        try {

            while (read.size() >= maxCapacity) {
                Thread.sleep(10);
            }
            DataItem dr = new DataItem<>(i, batch);
            this.read.put(dr);


        } catch (InterruptedException e) {
            e.printStackTrace();
        }

    }

    public DataItem getRead() {
        try {
            DataItem dr = this.read.poll(1, TimeUnit.SECONDS);
            return dr;

        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        return null;
    }

    public void putWrite(int i, W batch) {


        DataItem dw = new DataItem<>(i, batch);
        try {
            this.write.put(dw);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

    }

    public DataItem getWrite() {
        try {
            while (numProducersConsumers > 0) {
                DataItem dw = null;
                synchronized (write) {
                    dw = write.peek();
                    if (dw != null && dw.getTokenId() == currentElement) {
                        dw = write.take();
                        currentElement++;
                        return dw;
                    }
                }

                if (dw == null) {     // No element retrieved
                    Thread.sleep(10);
                }
            }
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        return null;
    }

    public int writeSize() {
        return this.write.size();
    }

    public void incConsumer() {
        this.numProducersConsumers++;
    }

    public int getNumProducersConsumers() {
        return numProducersConsumers;
    }

    public void decConsumer() {
        this.numProducersConsumers--;

    }

    public String toString() {

        return read.size() + " - " + write.size() + " --- " + numProducersConsumers;
    }

    public boolean isContinueProducing() {
        return continueProducing;
    }

    public void setContinueProducing(boolean continueProducing) {
        this.continueProducing = continueProducing;
    }
}