使用 Redis 的 zset + 线程池，实现延时队列

一个靠谱的延时队列，通常需要保证以下几个特性：

1. 持久化，不能因为某个应用的重启而信息丢失；
2. 确认和重试机制，发送成功了再删除，发送失败了可以重试；
3. 低延时，发送时间尽量精准，误差不能太大；

前几天写的两个demo，一个不能满足持久化，应用一重启信息就丢失了。另一个使用了轮询，太费性能了。于是想再次优化一下，先来看看代码吧。

一、延时任务线程池配置

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;

/**
 * 延迟任务线程池配置类
 * @author 程就人生
 * @Date
 */
@Configuration
public class TheadPoolConfig {
  
  @Bean("threadPoolTaskScheduler")
  public ThreadPoolTaskScheduler getThreadPoolTaskScheduler(){
    // 定时任务线程池
    ThreadPoolTaskScheduler executor = new ThreadPoolTaskScheduler();
    // 线程池大小
    executor.setPoolSize(10);
    // 线程执行前缀
    executor.setThreadNamePrefix("ThreadPoolTaskScheduler-");
    // executor.setWaitForTasksToCompleteOnShutdown(true);
    // executor.setAwaitTerminationSeconds(60);
    executor.initialize();
    return executor;
  }  
}

二、延时任务类

import java.text.SimpleDateFormat;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.data.redis.core.ZSetOperations;

/**
 * 延时任务执行的线程
 * @author
 * @Date
 */
public class DelayTaskExec implements Runnable{

  SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
  
  private static final Logger log = LoggerFactory.getLogger(DelayTaskExec.class);
  
  ZSetOperations zset;
  
  String uid;
  
  public DelayTaskExec(ZSetOperations zset, String uid){
    this.zset = zset;
    this.uid = uid;
  }
  
  @Override
  public void run() {
    // TODO 执行具体的业务逻辑
    Long now = System.currentTimeMillis();    
    log.info("计划执行时间：{}， 实际执行时间：{},set={}", format.format(zset.score("AA", uid).longValue()), format.format(now), uid);
    // 从redis缓存中移除
    zset.remove("AA", uid);
  }
}

这是任务类，可以把具体延时任务的业务逻辑写在这里。

三、项目重启时，redis缓存数据的处理

import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Set;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.CommandLineRunner;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.ZSetOperations;
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;

import com.example.demo.controller.DelayTaskExec;

/**
 * 启动类
 * @author 程就人生
 * @Date
 */
@EnableScheduling
@SpringBootApplication
public class SrpingRedisMqDemoApplication implements CommandLineRunner{
  // redis资源
  @Autowired
  private RedisTemplate redisTemplate;
  // 线程池
  @Autowired
  private ThreadPoolTaskScheduler threadPoolTaskScheduler;
  
  private static final Logger log = LoggerFactory.getLogger(SrpingRedisMqDemoApplication.class);
  
  SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");

  public static void main(String[] args) {
    SpringApplication.run(SrpingRedisMqDemoApplication.class, args);
  }

  @Override
  public void run(String... args) throws Exception {
    ZSetOperations zset = redisTemplate.opsForZSet();
    Long scheduleTime = System.currentTimeMillis();    
    Set