如何實現Java中的延遲隊列

小編給大家分享一下如何實現Java中的延遲隊列，希望大家閱讀完這篇文章之后都有所收獲，下面讓我們一起去探討吧！

常見的實現方法主要有：定時任務掃描、RocketMQ延遲隊列、Java自動的延遲隊列、監聽Redis Key過期等等

1.  DelayQueue

首先，定義一個延遲任務

package com.cjs.example;

import lombok.Data;

import java.util.concurrent.Delayed;
import java.util.concurrent.TimeUnit;

/**
 * @author ChengJianSheng
 * @since 2021/3/18
 */
@Data
public class DelayTask implements Delayed {

 private Long orderId;

 private long expireTime;

 public DelayTask(Long orderId, long expireTime) {
  this.orderId = orderId;
  this.expireTime = expireTime;
 }

 @Override
 public long getDelay(TimeUnit unit) {
  return expireTime - System.currentTimeMillis();
 }

 @Override
 public int compareTo(Delayed o) {
  return (int) (getDelay(TimeUnit.MILLISECONDS) - o.getDelay(TimeUnit.MILLISECONDS));
 }

}

然后，定義一個管理類

package com.cjs.example;

import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.CommandLineRunner;
import org.springframework.stereotype.Component;

import java.util.concurrent.DelayQueue;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

/**
 * @author ChengJianSheng
 * @since 2021/3/19
 */
@Slf4j
@Component
public class DelayQueueManager implements CommandLineRunner {

 private DelayQueue<DelayTask> queue = new DelayQueue<>();

 @Autowired
 private ParkOrderQueryHandler handler;

 @Override
 public void run(String... strings) throws Exception {
  ExecutorService executorService = Executors.newSingleThreadExecutor();
  executorService.execute(new Runnable() {
   @Override
   public void run() {
    while (true) {
     try {
      DelayTask task = queue.take();
      handler.handle(task);
     } catch (InterruptedException e) {
      e.printStackTrace();
     }
    }
   }
  });
 }

 public void put(DelayTask task) {
  queue.put(task);
 }
}

插入任務

@Slf4j
@Service
public class PayServiceImpl implements PayService {

 @Autowired
 private DelayQueueManager delayQueueManager;

 @Override
 public void pay() {
  
  delayQueueManager.put(new DelayTask(1, 15));
  delayQueueManager.put(new DelayTask(2, 30));
  delayQueueManager.put(new DelayTask(3, 60));

 }
}

2.  Redis Key過期回調

修改redis.conf文件

# bind 127.0.0.1 -::1
protected-mode no
notify-keyspace-events Ex

[root@localhost redis-6.2.1]$ src/redis-server redis.conf 
<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
   xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
 <modelVersion>4.0.0</modelVersion>
 <parent>
  <groupId>org.springframework.boot</groupId>
  <artifactId>spring-boot-starter-parent</artifactId>
  <version>2.4.4</version>
  <relativePath/> <!-- lookup parent from repository -->
 </parent>
 <groupId>com.example</groupId>
 <artifactId>demo0401</artifactId>
 <version>0.0.1-SNAPSHOT</version>
 <name>demo0401</name>
 <description>Demo project for Spring Boot</description>
 <properties>
  <java.version>1.8</java.version>
 </properties>
 <dependencies>
  <dependency>
   <groupId>org.springframework.boot</groupId>
   <artifactId>spring-boot-starter-data-redis</artifactId>
  </dependency>
 </dependencies>

 <build>
  <plugins>
   <plugin>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-maven-plugin</artifactId>
   </plugin>
  </plugins>
 </build>

</project>

RedisConfig.java

package com.example.config;

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.redis.connection.RedisConnectionFactory;
import org.springframework.data.redis.listener.RedisMessageListenerContainer;

/**
 * @author ChengJianSheng
 * @since 2021/4/2
 */
@Configuration
public class RedisConfig {

 @Bean
 public RedisMessageListenerContainer container(RedisConnectionFactory connectionFactory) {
  RedisMessageListenerContainer container = new RedisMessageListenerContainer();
  container.setConnectionFactory(connectionFactory);
  return container;
 }
}

創建一個監聽類

package com.example.listener;

import org.springframework.data.redis.connection.Message;
import org.springframework.data.redis.listener.KeyExpirationEventMessageListener;
import org.springframework.data.redis.listener.RedisMessageListenerContainer;
import org.springframework.stereotype.Component;

/**
 * @author ChengJianSheng
 * @since 2021/4/2
 */
@Component
public class MyRedisKeyExpirationListener extends KeyExpirationEventMessageListener {

 public MyRedisKeyExpirationListener(RedisMessageListenerContainer listenerContainer) {
  super(listenerContainer);
 }

 @Override
 public void onMessage(Message message, byte[] pattern) {
  String expiredKey = message.toString();
  System.out.println("監聽到Key: " + expiredKey + " 已過期");
 }
}

看完了這篇文章，相信你對“如何實現Java中的延遲隊列”有了一定的了解，如果想了解更多相關知識，歡迎關注億速云行業資訊頻道，感謝各位的閱讀！