SpringBoot怎麼整合RabbitMq自訂訊息監聽容器來實現訊息批次處理

SpringBoot 整合RabbitMq 自訂訊息監聽容器來實現訊息批次處理

前言

RabbitMQ是一種常用的訊息佇列，Spring Boot對其進行了深度的整合，可以快速地實現訊息的發送和接收。在RabbitMQ中，訊息的發送和接收都是非同步的，因此需要使用監聽器來監聽訊息的到來。 Spring Boot中提供了預設的監聽器容器，但有時候我們需要自訂監聽器容器，來滿足一些特殊的需求，例如大量取得資料。

在本文中，我們將使用Spring Boot來整合RabbitMQ，並自訂監聽器容器，實現大量取得資料的功能。
前置條件：
在開始之前，您需要具備以下條件：

• 已經安裝好RabbitMQ伺服器並啟動。

• 已經建立好要使用的佇列。

• 已經熟悉了Spring Boot和RabbitMQ的基本知識。

環境準備：
在開始之前，我們需要準備好以下環境：

• JDK 1.8或以上版本

• Spring Boot 2.5.0或以上版本

• RabbitMQ 3.8.0或以上版本

#新增依賴

首先，在pom.xml檔案中新增以下依賴：

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-amqp</artifactId>
</dependency>

設定檔

接下來，在application.properties檔案中新增下列配置：

spring.rabbitmq.host=localhost
spring.rabbitmq.port=5672
spring.rabbitmq.username=guest
spring.rabbitmq.password=guest
spring.rabbitmq.virtual-host=/

# 队列名称
spring.rabbitmq.listener.simple.queue-name=myQueue

# 最大并发消费者数量
spring.rabbitmq.listener.simple.concurrency=5

# 最小数量
spring.rabbitmq.listener.simple.min-concurrency=1

# 最大数量
spring.rabbitmq.listener.simple.max-concurrency=10

# 批量处理消息的大小
spring.rabbitmq.listener.simple.batch-size=50

或

spring:
  rabbitmq:
    host: localhost
    listener:
      simple:
        batch-size: 50
        concurrency: 5
        max-concurrency: 10
        min-concurrency: 1
        queue-name: myQueue
    password: guest
    port: 5672
    username: guest
    virtual-host: /

編寫監聽器

然後，我們需要建立一個監聽器類，以便處理從佇列中接收到的訊息。以下是一個簡單的範例：

@Component
public class MyListener {
    
    @RabbitListener(queues = "myQueue", containerFactory = "myFactory")
    public void handleMessage(List<MyMessage> messages, Channel channel, @Header(AmqpHeaders.DELIVERY_TAG) long deliveryTag)
            throws IOException {
        try {
            // 处理消息
            System.out.println("Received " + messages.size() + " messages");
            for (Message message : messages) {
           		// 处理消息
            	System.out.println("Received message: " + new String(message.getBody()));
        	}
        	channel.basicAck(messages.get(messages.size() - 1).getMessageProperties().getDeliveryTag(), true);
        } finally {
            // 手动确认消息
            channel.basicAck(deliveryTag, true);
        }
    }
}

在上面的程式碼中，我們使用了@RabbitListener註解來指定要監聽的佇列名稱，同時也指定了使用myFactory工廠來建立監聽容器。在這個監聽器中，我們簡單地列印了接收到的訊息。

建立SimpleRabbitListenerContainerFactory

接下來，我們需要建立一個SimpleRabbitListenerContainerFactory工廠，以便能夠自訂監聽容器的行為。以下是一個簡單的範例：

@Configuration
public class RabbitMQConfig {

//    @Bean
//    public SimpleRabbitListenerContainerFactory myFactory(ConnectionFactory connectionFactory) {
//        SimpleRabbitListenerContainerFactory factory = new SimpleRabbitListenerContainerFactory();
//        factory.setConnectionFactory(connectionFactory);
//        factory.setConcurrentConsumers(1);
//        factory.setMaxConcurrentConsumers(10);
//        factory.setBatchListener(true);
//        factory.setBatchSize(50);
//        return factory;
//    }

	@Bean
    public SimpleRabbitListenerContainerFactory myFactory(
            ConnectionFactory connectionFactory,
            PlatformTransactionManager transactionManager,
            MessageConverter messageConverter) {
        
        SimpleRabbitListenerContainerFactory factory = new SimpleRabbitListenerContainerFactory();
        
        // 并发消费者数，默认为 1
        factory.setConcurrentConsumers(5);
        
        // 最大并发消费者数，默认为 1
        factory.setMaxConcurrentConsumers(10);
        
        // 拒绝未确认的消息并重新将它们放回队列，默认为 true
        factory.setDefaultRequeueRejected(false);
        
        // 容器启动时是否自动启动，默认为 true
        factory.setAutoStartup(true);
        
        // 消息确认模式，默认为 AUTO
        factory.setAcknowledgeMode(AcknowledgeMode.MANUAL);
        
        // 每个消费者在一次请求中预获取的消息数，默认为 1
        factory.setPrefetchCount(5);
        
        // 从队列中接收消息的超时时间，默认为 0，表示没有超时限制
        factory.setReceiveTimeout(1000);
        
        // 与容器一起使用的事务管理器。默认情况下，容器不会使用事务
        factory.setTransactionManager(transactionManager);
        
        // 消息转换器，用于将接收到的消息转换为 Java 对象或将 Java 对象转换为消息
        factory.setMessageConverter(messageConverter);
        
        // 用于异步消息处理的线程池。默认情况下，容器使用一个简单的 SimpleAsyncTaskExecutor
        factory.setTaskExecutor(new SimpleAsyncTaskExecutor());
        
        // 在关闭容器时等待活动线程终止的时间，默认为 5000 毫秒
        factory.setShutdownTimeout(10000);
        
        // 重试失败的消息之前等待的时间，默认为 5000 毫秒
        factory.setRecoveryInterval(5000);
        
        // 如果消息处理器尝试监听不存在的队列，是否抛出异常。默认为 true
        factory.setMissingQueuesFatal(false);
        
        // 监听器容器连接工厂
        factory.setConnectionFactory(connectionFactory);

        return factory;
    }
}

這些屬性中的大多數都是可選的，可以根據需要進行設定。根據應用程式的需求，我們可以自由地調整這些屬性，以提高應用程式的效能和可靠性。

發送訊息

最後，我們可以寫一個簡單的發送訊息的程式碼來向佇列中發送一些訊息。以下是一個簡單的範例：

@Component
public class MySender {

    @Autowired
    private RabbitTemplate rabbitTemplate;

    public void sendMessage(String message) {
        for (int i = 0; i < 100; i++) {
            rabbitTemplate.convertAndSend("myQueue", "message：" + i);
        }
    }
}

以上是SpringBoot怎麼整合RabbitMq自訂訊息監聽容器來實現訊息批次處理的詳細內容。更多資訊請關注PHP中文網其他相關文章！