java的spring框架中使用依赖是spring-rabbit使用注解@RabbitListener接收rabbitmq的消息如何批量获取固定的条数？(亲测)

迷南。 2023-10-15 17:44 9阅读 0赞

在 Spring Framework 中使用 spring-rabbit 依赖，并使用 @RabbitListener 注解接收 RabbitMQ 消息时，你可以通过配置 SimpleRabbitListenerContainerFactory 来实现批量获取固定数量的消息。以下是一种常见的设置方式：

首先，在 Spring 配置类中创建一个 SimpleRabbitListenerContainerFactory 的 bean，并设置 prefetchCount 参数来指定每次批量获取的消息数量。  
java  
Copy code  
import org.springframework.amqp.rabbit.config.SimpleRabbitListenerContainerFactory;  
import org.springframework.context.annotation.Bean;  
import org.springframework.context.annotation.Configuration;

@Configuration  
public class RabbitMQConfig \{

@Bean  
public SimpleRabbitListenerContainerFactory rabbitListenerContainerFactory() \{  
SimpleRabbitListenerContainerFactory factory = new SimpleRabbitListenerContainerFactory();  
factory.setConnectionFactory(connectionFactory());  
factory.setConcurrentConsumers(5); // 设置并发消费者数量  
factory.setPrefetchCount(10); // 设置每次批量获取的消息数量为 10 条  
return factory;  
\}

// 创建 RabbitMQ 连接工厂  
@Bean  
public ConnectionFactory connectionFactory() \{  
CachingConnectionFactory connectionFactory = new CachingConnectionFactory();  
// 设置连接参数...  
return connectionFactory;  
\}  
\}  
在以上示例中，我们创建了一个名为 rabbitListenerContainerFactory 的 SimpleRabbitListenerContainerFactory 实例，并将 prefetchCount 设置为 10，表示每次从 RabbitMQ 中批量获取 10 条消息。

我们还设置了并发消费者数量和连接工厂，你需要根据实际需求进行适当的配置。

接下来，使用 @RabbitListener 注解标记要接收消息的方法，并指定使用刚刚创建的 rabbitListenerContainerFactory。  
java  
Copy code  
import org.springframework.amqp.rabbit.annotation.RabbitListener;  
import org.springframework.stereotype.Component;

@Component  
public class RabbitMQReceiver \{

@RabbitListener(queues = "your\_queue\_name", containerFactory = "rabbitListenerContainerFactory")  
public void receiveMessages(List<String> messages) \{  
for (String message : messages) \{  
// 处理接收到的消息逻辑  
\}  
\}  
\}  
在上述示例中，我们使用 @RabbitListener 注解标记了 receiveMessages 方法，并通过 containerFactory 属性指定了使用的容器工厂为 rabbitListenerContainerFactory。

这样，当消息到达时，框架将使用我们自定义的容器工厂来创建 RabbitMQ 监听器容器，从而实现批量获取固定数量的消息。

请确保已经正确配置了 RabbitMQ 连接信息和队列信息，并且在 Spring Boot 应用程序的配置类中包含了上述的配置类。

以上就是使用依赖 spring-rabbit 和注解 @RabbitListener 批量获取固定数量消息的一种方式。根据你的实际需求，可能需要对其他属性进行适当配置，例如消息确认模式、错误处理等。