java 将rabbitmq多条数据组装成一条

原创

mob64ca12d5dd85 2024-10-05 06:43:05 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12d5dd85的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java 中将 RabbitMQ 多条数据组装成一条

在现代微服务架构中，消息队列作为不同服务之间通信的中介，发挥着至关重要的作用。RabbitMQ 作为一种流行的消息队列系统，常常被广泛用于消息传递和数据处理。然而，在某些场景中，我们需要将从 RabbitMQ 接收到的多条消息合并成一条。这篇文章将详细介绍这一过程。

RabbitMQ 简介

RabbitMQ 是一个开源的消息代理，它实现了高级消息队列协议（AMQP）。通过 RabbitMQ，应用程序可以以异步方式通信。RabbitMQ 支持多种消息传递模式，如发布—订阅、路由、点对点等，使得它特别适合于处理高并发的数据流。

需求背景

在一些情况下，应用程序可能需要将多条消息合并为一条，例如，在完成一个复杂的业务逻辑时，可能需要合并多条与之相关的数据。比如，一次订单的创建可能涉及到多个商品的信息。在这种情况下，我们需要将多条相关的商品信息合并为一条订单信息。

环境搭建

在开始编写代码之前，请确保你安装了以下组件：

Java JDK（版本 8 以上）
RabbitMQ 服务器
Maven（用于管理依赖）

Maven 依赖

首先，在你的 pom.xml 文件中添加 RabbitMQ 的相关依赖：

<dependency>
    <groupId>com.rabbitmq</groupId>
    <artifactId>amqp-client</artifactId>
    <version>5.16.0</version>
</dependency>

消息发布与消费

我们先创建一个简单的消息发布和消费的示例。

消息发布

以下代码示例展示了如何发布多条消息到 RabbitMQ 的队列中：

import com.rabbitmq.client.Channel;
import com.rabbitmq.client.Connection;
import com.rabbitmq.client.ConnectionFactory;

public class Publisher {

    private final static String QUEUE_NAME = "order_queue";

    public static void main(String[] argv) throws Exception {
        ConnectionFactory factory = new ConnectionFactory();
        factory.setHost("localhost");
        try (Connection connection = factory.newConnection(); 
             Channel channel = connection.createChannel()) {
            channel.queueDeclare(QUEUE_NAME, false, false, false, null);
            for (int i = 1; i <= 5; i++) {
                String message = "商品信息 " + i;
                channel.basicPublish("", QUEUE_NAME, null, message.getBytes());
                System.out.println("[x] 发送: '" + message + "'");
            }
        }
    }
}

消息消费

接下来，我们创建一个消息消费者，它将接收到来自 order_queue 队列的消息。

import com.rabbitmq.client.*;

import java.io.IOException;

public class Consumer {

    private final static String QUEUE_NAME = "order_queue";

    public static void main(String[] argv) throws Exception {
        ConnectionFactory factory = new ConnectionFactory();
        factory.setHost("localhost");
        try (Connection connection = factory.newConnection(); 
             Channel channel = connection.createChannel()) {
            channel.queueDeclare(QUEUE_NAME, false, false, false, null);
            System.out.println("[*] 等待消息...");

            DeliverCallback deliverCallback = (consumerTag, delivery) -> {
                String message = new String(delivery.getBody(), "UTF-8");
                System.out.println("[x] 接收到: '" + message + "'");
                // 在这里进行数据组装
                // ...
            };
            channel.basicConsume(QUEUE_NAME, true, deliverCallback, consumerTag -> { });
        }
    }
}

数据组装

在实际应用中，我们希望将上述消费者接收到的多条消息组装成一条完整的订单信息。我们可以使用一个集合来存储接收到的消息，并在特定条件下（如接收到最后一条消息时）触发数据组装。

import java.util.ArrayList;
import java.util.List;

public class Consumer {

    private static final List<String> messages = new ArrayList<>();

    // ...（包括队列声明和连接的代码部分）

    DeliverCallback deliverCallback = (consumerTag, delivery) -> {
        String message = new String(delivery.getBody(), "UTF-8");
        System.out.println("[x] 接收到: '" + message + "'");
        messages.add(message);

        // 假如当接收到5条消息后进行组装
        if (messages.size() == 5) {
            String assembledMessage = String.join(", ", messages);
            System.out.println("[*] 组装后的订单信息: '" + assembledMessage + "'");
            messages.clear(); // 清空列表以便下次使用
        }
    };

    //...（其余代码）
}