in.hocg.boot.cache.autoconfiguration.queue.RedisDelayedQueue Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of cache-spring-boot-autoconfigure Show documentation

The newest version!

package in.hocg.boot.cache.autoconfiguration.queue;

import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.redisson.api.RBlockingQueue;
import org.redisson.api.RDelayedQueue;
import org.redisson.api.RedissonClient;

import java.util.concurrent.TimeUnit;
import java.util.function.Consumer;

/**
 * 延迟队列
 *
 * @author hocgin
 */
@Slf4j
@RequiredArgsConstructor
public class RedisDelayedQueue {

    private final RedissonClient redissonClient;

    /**
     * 添加到队列
     *
     * @param data     数据对象
     * @param delay    时间数量
     * @param timeUnit 时间单位
     * @param       泛型
     */
    public  void addQueue(T data, long delay, TimeUnit timeUnit) {
        RBlockingQueue blockingFairQueue = redissonClient.getBlockingQueue(data.getClass().getName());
        RDelayedQueue delayedQueue = redissonClient.getDelayedQueue(blockingFairQueue);
        delayedQueue.offer(data, delay, timeUnit);
        delayedQueue.destroy();
    }

    /**
     * 获取队列
     *
     * @param zClass            数据对象类型，用于获取队列
     * @param taskEventListener 任务回调监听
     * @param                泛型
     */
    public  void getQueue(Class zClass, Consumer taskEventListener) {
        RBlockingQueue blockingFairQueue = redissonClient.getBlockingQueue(zClass.getName());
        // 由于此线程需要常驻，可以新建线程，不用交给线程池管理
        ((Runnable) () -> {
            while (true) {
                try {
                    taskEventListener.accept(blockingFairQueue.take());
                } catch (InterruptedException e) {
                    log.error("程序中断", e);
                }
            }
        }).run();
    }


}