com.github.phantomthief.jedis.QueueConsumer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jedis-simple-queue Show documentation

A simple jedis based queue

The newest version!

/**
 * 
 */
package com.github.phantomthief.jedis;

import java.util.function.Consumer;
import java.util.function.Supplier;

/**
 * @author w.vela
 */
public class QueueConsumer {

    private final org.slf4j.Logger logger = org.slf4j.LoggerFactory.getLogger(getClass());

    private final Supplier queuePoper;

    private final Consumer queueConsumer;

    private volatile boolean stopped = false;

    /**
     * @param queuePoper
     * @param queueConsumer
     */
    public QueueConsumer(Supplier queuePoper, Consumer queueConsumer) {
        this.queuePoper = queuePoper;
        this.queueConsumer = queueConsumer;
    }

    public void startConsume() {
        while (!stopped) {
            try {
                T object = queuePoper.get();
                if (object != null) {
                    queueConsumer.accept(object);
                }
            } catch (Throwable e) {
                logger.error("fail to consumer:{}, {}, exception:{}", queuePoper, queueConsumer,
                        e.getMessage());
            }
        }
    }

    public void stop() {
        stopped = true;
    }

}