Spring Boot 整合：Redis 延时队列的实现方案（基于有赞的设计）

private String NAME = “job.pool”;

private BoundHashOperations getPool () {

BoundHashOperations ops = redisTemplate.boundHashOps(NAME);

return ops;

}

/**

* 添加任务

* @param job

*/

public void addJob (Job job) {

log.info(“任务池添加任务：{}”, JSON.toJSONString(job));

getPool().put(job.getId(),job);

return ;

}

/**

* 获得任务

* @param jobId

* @return

*/

public Job getJob(Long jobId) {

Object o = getPool().get(jobId);

if (o instanceof Job) {

return (Job) o;

}

return null;

}

/**

* 移除任务

* @param jobId

*/

public void removeDelayJob (Long jobId) {

log.info(“任务池移除任务：{}”,jobId);

// 移除任务

getPool().delete(jobId);

}

}

延迟任务，使用可排序的ZSet保存数据，提供取出最小值等操作

@Slf4j

@Component

public class DelayBucket {

@Autowired

private RedisTemplate redisTemplate;

private static AtomicInteger index = new AtomicInteger(0);

@Value(“${thread.size}”)

private int bucketsSize;

private List  bucketNames = new ArrayList <>();

@Bean

public List  createBuckets() {

for (int i = 0; i < bucketsSize; i++) {

bucketNames.add(“bucket” + i);

}

return bucketNames;

}

/**

* 获得桶的名称

* @return

*/

private String getThisBucketName() {

int thisIndex = index.addAndGet(1);

int i1 = thisIndex % bucketsSize;

return bucketNames.get(i1);

}

/**

* 获得桶集合

* @param bucketName

* @return

*/

private BoundZSetOperations getBucket(String bucketName) {

return redisTemplate.boundZSetOps(bucketName);

}

/**

* 放入延时任务

* @param job

*/

public void addDelayJob(DelayJob job) {

log.info(“添加延迟任务:{}”, JSON.toJSONString(job));

String thisBucketName = getThisBucketName();

BoundZSetOperations bucket = getBucket(thisBucketName);

bucket.add(job,job.getDelayDate());

}

/**

* 获得最新的延期任务

* @return

*/

public DelayJob getFirstDelayTime(Integer index) {

String name = bucketNames.get(ind