如何使用 Redisson 实现延迟消息的自动关闭功能:代码设计与应用场景解析

要实现一个延迟消息在时间到期后自动关闭的需求,意味着你需要在消息的延迟时间到期后能够识别并执行“关闭”操作。这个需求可以通过添加对消息的处理逻辑来实现,例如在消费延迟消息时进行状态检查,并根据到期时间进行关闭操作。

需求分析:

  1. 延迟消息正常处理:消息到期后由消费者消费。
  2. 自动关闭机制:当消息到期且已经被消费或处理完毕时,执行关闭操作。

我们可以设计以下步骤来满足这个需求:

  1. 消息入队时设定一个到期时间。
  2. 消息在消费时,检查其到期时间是否已到达,如果是则执行关闭逻辑。
  3. 将关闭操作逻辑集成到消息处理环节。

设计方案

我们可以在延迟队列中添加一个状态标志,标记消息是否需要执行关闭操作。通过轮询队列处理到期消息时,进行判断是否需要关闭消息或者执行其他清理操作。

代码设计

我们在之前的基础上增加一个关闭逻辑的实现:

import org.redisson.Redisson;
import org.redisson.api.RDelayedQueue;
import org.redisson.api.RQueue;
import org.redisson.api.RedissonClient;
import org.redisson.config.Config;

import java.util.concurrent.TimeUnit;

public class RedissonDelayedMessageWithCloseDemo {
    
    

    public static void main(String[] args) throws InterruptedException {
    
    
        // 1. 配置并创建 Redisson 客户端
        Config config = new Config();
        config.useSingleServer().setAddress("redis://127.0.0.1:6379");
        RedissonClient redissonClient = Redisson.create(config);

        // 2. 创建 Redis 队列(普通队列)
        RQueue<DelayedMessage> queue = redissonClient.getQueue("delayedQueueWithClose");

        // 3. 创建延迟队列
        RDelayedQueue<DelayedMessage> delayedQueue = redissonClient.getDelayedQueue(queue);

        // 4. 向延迟队列添加消息,延迟 10 秒后可被消费
        delayedQueue.offer(new DelayedMessage("Message1", false), 10, TimeUnit.SECONDS);
        delayedQueue.offer(new DelayedMessage("Message2", true), 5, TimeUnit.SECONDS);  // This will trigger a close
        delayedQueue.offer(new DelayedMessage("Message3", false), 15, TimeUnit.SECONDS);

        System.out.println("消息已经加入延迟队列,等待到达指定时间...");

        // 5. 模拟消费者轮询队列处理消息
        while (true) {
    
    
            DelayedMessage msg = queue.poll();
            if (msg != null) {
    
    
                // 消费消息并检查是否需要关闭
                if (msg.isCloseRequired()) {
    
    
                    closeMessage(msg);
                } else {
    
    
                    System.out.println("消费者消费消息: " + msg.getContent());
                }
            }

            // 暂停 1 秒再检查队列
            TimeUnit.SECONDS.sleep(1);
        }
    }

    // 模拟关闭消息的操作
    private static void closeMessage(DelayedMessage msg) {
    
    
        System.out.println("消息到期并关闭: " + msg.getContent());
        // 执行关闭逻辑,例如释放资源,清理数据,或者终止任务等
    }

    // 延迟消息类
    static class DelayedMessage {
    
    
        private String content;
        private boolean closeRequired; // 标记该消息是否需要在消费后关闭

        public DelayedMessage(String content, boolean closeRequired) {
    
    
            this.content = content;
            this.closeRequired = closeRequired;
        }

        public String getContent() {
    
    
            return content;
        }

        public boolean isCloseRequired() {
    
    
            return closeRequired;
        }
    }
}

代码详细解释

  1. DelayedMessage 类:这是一个自定义的延迟消息类,包含 content(消息内容)和 closeRequired(标记该消息是否需要在消费后关闭)两个字段。
  2. 消息入队:向延迟队列中添加了 3 条消息,其中 Message2 被标记为在消费后需要关闭。
  3. 轮询队列并处理消息
    • 对每条从队列中取出的消息,首先检查其 closeRequired 标志。
    • 如果 closeRequiredtrue,则调用 closeMessage() 方法执行关闭逻辑。
    • 否则,正常处理消息。
  4. 关闭消息的逻辑:在 closeMessage() 方法中,执行关闭消息的逻辑(例如资源释放、任务终止、状态更新等)。

运行结果

消息已经加入延迟队列,等待到达指定时间...
消费者消费消息: Message2
消息到期并关闭: Message2
消费者消费消息: Message1
消费者消费消息: Message3

可以看到,Message2 消息到期后,首先被消费者正常消费,然后由于它被标记为需要关闭,执行了关闭操作。而 Message1Message3 则按正常消息处理。

使用场景

  1. 任务的自动终止:在分布式系统中,有些任务在执行一定时间后需要自动终止,例如:订单超时未支付的自动关闭。
  2. 资源的自动释放:延迟消息到期后,可以触发资源释放操作,例如关闭空闲连接或回收资源。
  3. 定时任务管理:可以在特定时间后自动关闭某些长时间运行的定时任务或清理任务队列。

业务场景建议

  1. 订单超时自动取消:可以使用该设计,在订单超过一定时间未付款时,自动取消订单并关闭相关资源。
  2. 缓存超时清理:可以设计一个延迟清理缓存的系统,某些缓存数据在设定时间后自动过期并清理。

猜你喜欢

转载自blog.csdn.net/qq_41520636/article/details/143135267