Redis 延时队列

dependency

        <dependency>
            <groupId>redis.clients</groupId>
            <artifactId>jedis</artifactId>
            <version>2.6.0</version>
        </dependency>

class Constant

public class Constant {
    public static final int REDIS_PORT = 6379;
    public static final String REDIS_IP = "localhost";// 需要自行修改
    public static final String REDIS_TEST_ZSET_KEY = "testZsetKey";
}

class DelayQueue

import java.util.Set;

import com.tbryant.test.domain.Constant;
import redis.clients.jedis.Jedis;

/**
 * 使用有序集合实现延时队列
 */
public class DelayQueue {

    private static final Jedis JEDIS = new Jedis(Constant.REDIS_IP, Constant.REDIS_PORT);

    /**
     * 推送消息
     * @param delaySecond 延迟时间（秒）
     */
    static Long setMessage(String queueKey, String message, int delaySecond) {
        return JEDIS.zadd(queueKey, System.currentTimeMillis() + delaySecond * 1000, message);
    }

    /**
     * 拉取一条消息
     * 暂时没有消息返回null
     */
    static String getMessage(String queueKey) {
        Set<String> values = JEDIS.zrangeByScore(queueKey, 0, System.currentTimeMillis(), 0, 1);
        if (!values.isEmpty()) {
            String message = values.iterator().next();
            System.out.println(message);
            // 竞争这条数据
            if (JEDIS.zrem(queueKey, message) > 0) {
                return message;
            }
        }
        return null;
    }
}

class DelayQueueTestProducer

/**
 * 消息生产者
 * @auther TBryant
 * @date 2021/5/26 16:40
 */
public class DelayQueueTestProducer {
    public static void main(String[] args) {
        for (int i = 0; i < 100; i++) {
            System.out.println(DelayQueue.setMessage(Constant.REDIS_TEST_ZSET_KEY, "TBryant-" + i,5));
        }
    }
}

class DelayQueueTestConsumer

/**
 * 消息消费者
 * @auther TBryant
 * @date 2021/5/26 16:40
 */
public class DelayQueueTestConsumer {
    public static void main(String[] args) {
        while (true) {
            String message = DelayQueue.getMessage(Constant.REDIS_TEST_ZSET_KEY);
            if (null == message) {
                // 如果暂时没有消息，等一秒再取
                try {
                    Thread.sleep(1000);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            } else {
                System.out.println("process:" + message);
            }
        }
    }
}