SpringBoot:詳解RabbitMQ 延遲隊列

SpringBoot 是爲了簡化 Spring 應用的建立、運行、調試、部署等一系列問題而誕生的產物,自動裝配的特性讓咱們能夠更好的關注業務自己而不是外部的XML配置,咱們只需遵循規範,引入相關的依賴就能夠輕易的搭建出一個 WEB 工程html

初探RabbitMQ消息隊列中介紹了RabbitMQ的簡單用法,順帶說起了下延遲隊列的做用。所謂延時消息就是指當消息被髮送之後,並不想讓消費者當即拿到消息,而是等待指定時間後,消費者纔拿到這個消息進行消費。java

延遲隊列

延遲隊列能作什麼?程序員

  • 訂單業務: 在電商/點餐中,都有下單後 30 分鐘內沒有付款,就自動取消訂單。
  • 短信通知: 下單成功後 60s 以後給用戶發送短信通知。
  • 失敗重試: 業務操做失敗後,間隔必定的時間進行失敗重試。

這類業務的特色就是:非實時的,須要延遲處理,須要進行失敗重試。一種比較笨的方式是採用定時任務,輪訓數據庫,方法簡單好用,但性能底下,在高併發狀況下容易弄死數據庫,間隔時間很差設置,時間過大,影響精度,太小影響性能,並且作不到按超時的時間順序處理。另外一種就是用Java中的DelayQueue 位於java.util.concurrent包下,本質是由PriorityQueue和BlockingQueue實現的阻塞優先級隊列。,這玩意最大的問題就是不支持分佈式與持久化web

RabbitMQ 實現思路

RabbitMQ隊列自己是沒有直接實現支持延遲隊列的功能,但能夠經過它的Time-To-Live Extensions 與 Dead Letter Exchange 的特性模擬出延遲隊列的功能。spring

Time-To-Live Extensions數據庫

RabbitMQ支持爲隊列或者消息設置TTL(time to live 存活時間)。TTL代表了一條消息可在隊列中存活的最大時間。當某條消息被設置了TTL或者當某條消息進入了設置了TTL的隊列時,這條消息會在TTL時間後死亡成爲Dead Letter。若是既配置了消息的TTL,又配置了隊列的TTL,那麼較小的那個值會被取用。json

Dead Letter Exchange併發

死信交換機,上文中提到設置了 TTL 的消息或隊列最終會成爲Dead Letter。若是爲隊列設置了Dead Letter Exchange(DLX),那麼這些Dead Letter就會被從新發送到Dead Letter Exchange中,而後經過Dead Letter Exchange路由到其餘隊列,便可實現延遲隊列的功能。app

導入依賴

在 pom.xml 中添加 spring-boot-starter-amqp的依賴分佈式

<dependencies>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-amqp</artifactId>
    </dependency>
    <dependency>
        <groupId>com.alibaba</groupId>
        <artifactId>fastjson</artifactId>
        <version>1.2.46</version>
    </dependency>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-web</artifactId>
    </dependency>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-test</artifactId>
        <scope>test</scope>
    </dependency>
</dependencies>

屬性配置

在 application.properties 文件中配置rabbitmq相關內容,值得注意的是這裏配置了手動ACK的開關

spring.rabbitmq.username=battcn
spring.rabbitmq.password=battcn
spring.rabbitmq.host=192.168.0.133
spring.rabbitmq.port=5672
spring.rabbitmq.virtual-host=/
# 手動ACK 不開啓自動ACK模式,目的是防止報錯後未正確處理消息丟失 默認 爲 none
spring.rabbitmq.listener.simple.acknowledge-mode=manual

具體編碼

定義隊列

若是手動建立過或者RabbitMQ中已經存在該隊列那麼也能夠省略下述代碼…

package com.battcn.config;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.amqp.core.*;
import org.springframework.amqp.rabbit.connection.CachingConnectionFactory;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.util.HashMap;
import java.util.Map;

/**
 * RabbitMQ配置
 *
 * @author Levin
 * @since 2018/4/11 0011
 */
@Configuration
public class RabbitConfig {

    private static final Logger log = LoggerFactory.getLogger(RabbitConfig.class);

    @Bean
    public RabbitTemplate rabbitTemplate(CachingConnectionFactory connectionFactory) {
        connectionFactory.setPublisherConfirms(true);
        connectionFactory.setPublisherReturns(true);
        RabbitTemplate rabbitTemplate = new RabbitTemplate(connectionFactory);
        rabbitTemplate.setMandatory(true);
        rabbitTemplate.setConfirmCallback((correlationData, ack, cause) -> log.info("消息發送成功:correlationData({}),ack({}),cause({})", correlationData, ack, cause));
        rabbitTemplate.setReturnCallback((message, replyCode, replyText, exchange, routingKey) -> log.info("消息丟失:exchange({}),route({}),replyCode({}),replyText({}),message:{}", exchange, routingKey, replyCode, replyText, message));
        return rabbitTemplate;
    }

    /**
     * 延遲隊列 TTL 名稱
     */
    private static final String REGISTER_DELAY_QUEUE = "dev.book.register.delay.queue";
    /**
     * DLX,dead letter發送到的 exchange
     * TODO 此處的 exchange 很重要,具體消息就是發送到該交換機的
     */
    public static final String REGISTER_DELAY_EXCHANGE = "dev.book.register.delay.exchange";
    /**
     * routing key 名稱
     * TODO 此處的 routingKey 很重要要,具體消息發送在該 routingKey 的
     */
    public static final String DELAY_ROUTING_KEY = "";


    public static final String REGISTER_QUEUE_NAME = "dev.book.register.queue";
    public static final String REGISTER_EXCHANGE_NAME = "dev.book.register.exchange";
    public static final String ROUTING_KEY = "all";

    /**
     * 延遲隊列配置
     * <p>
     * 一、params.put("x-message-ttl", 5 * 1000);
     * TODO 第一種方式是直接設置 Queue 延遲時間 但若是直接給隊列設置過時時間,這種作法不是很靈活,(固然兩者是兼容的,默認是時間小的優先)
     * 二、rabbitTemplate.convertAndSend(book, message -> {
     * message.getMessageProperties().setExpiration(2 * 1000 + "");
     * return message;
     * });
     * TODO 第二種就是每次發送消息動態設置延遲時間,這樣咱們能夠靈活控制
     **/
    @Bean
    public Queue delayProcessQueue() {
        Map<String, Object> params = new HashMap<>();
        // x-dead-letter-exchange 聲明瞭隊列裏的死信轉發到的DLX名稱,
        params.put("x-dead-letter-exchange", REGISTER_EXCHANGE_NAME);
        // x-dead-letter-routing-key 聲明瞭這些死信在轉發時攜帶的 routing-key 名稱。
        params.put("x-dead-letter-routing-key", ROUTING_KEY);
        return new Queue(REGISTER_DELAY_QUEUE, true, false, false, params);
    }

    /**
     * 須要將一個隊列綁定到交換機上,要求該消息與一個特定的路由鍵徹底匹配。
     * 這是一個完整的匹配。若是一個隊列綁定到該交換機上要求路由鍵 「dog」,則只有被標記爲「dog」的消息才被轉發,不會轉發dog.puppy,也不會轉發dog.guard,只會轉發dog。
     * TODO 它不像 TopicExchange 那樣可使用通配符適配多個
     *
     * @return DirectExchange
     */
    @Bean
    public DirectExchange delayExchange() {
        return new DirectExchange(REGISTER_DELAY_EXCHANGE);
    }

    @Bean
    public Binding dlxBinding() {
        return BindingBuilder.bind(delayProcessQueue()).to(delayExchange()).with(DELAY_ROUTING_KEY);
    }


    @Bean
    public Queue registerBookQueue() {
        return new Queue(REGISTER_QUEUE_NAME, true);
    }

    /**
     * 將路由鍵和某模式進行匹配。此時隊列須要綁定要一個模式上。
     * 符號「#」匹配一個或多個詞,符號「*」匹配很少很多一個詞。所以「audit.#」可以匹配到「audit.irs.corporate」,可是「audit.*」 只會匹配到「audit.irs」。
     **/
    @Bean
    public TopicExchange registerBookTopicExchange() {
        return new TopicExchange(REGISTER_EXCHANGE_NAME);
    }

    @Bean
    public Binding registerBookBinding() {
        // TODO 若是要讓延遲隊列之間有關聯,這裏的 routingKey 和 綁定的交換機很關鍵
        return BindingBuilder.bind(registerBookQueue()).to(registerBookTopicExchange()).with(ROUTING_KEY);
    }

}

實體類

建立一個Book

public class Book implements java.io.Serializable {

    private static final long serialVersionUID = -2164058270260403154L;

    private String id;
    private String name;
	// 省略get set ...
}

控制器

編寫一個Controller類,用於消息發送工做,同時爲了看到測試效果,添加日誌輸出,將發送消息的時間記錄下來..

package com.battcn.controller;

import com.battcn.config.RabbitConfig;
import com.battcn.entity.Book;
import com.battcn.handler.BookHandler;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.support.converter.AbstractJavaTypeMapper;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.time.LocalDateTime;

/**
 * @author Levin
 * @since 2018/4/2 0002
 */
@RestController
@RequestMapping(value = "/books")
public class BookController {

    private static final Logger log = LoggerFactory.getLogger(BookController.class);

    private final RabbitTemplate rabbitTemplate;

    @Autowired
    public BookController(RabbitTemplate rabbitTemplate) {
        this.rabbitTemplate = rabbitTemplate;
    }

    /**
     * this.rabbitTemplate.convertAndSend(RabbitConfig.REGISTER_DELAY_EXCHANGE, RabbitConfig.DELAY_ROUTING_KEY, book); 對應 {@link BookHandler#listenerDelayQueue}
     */
    @GetMapping
    public void defaultMessage() {
        Book book = new Book();
        book.setId("1");
        book.setName("一塊兒來學Spring Boot");
        // 添加延時隊列
        this.rabbitTemplate.convertAndSend(RabbitConfig.REGISTER_DELAY_EXCHANGE, RabbitConfig.DELAY_ROUTING_KEY, book, message -> {
            // TODO 第一句是可要可不要,根據本身須要自行處理
            message.getMessageProperties().setHeader(AbstractJavaTypeMapper.DEFAULT_CONTENT_CLASSID_FIELD_NAME, Book.class.getName());
            // TODO 若是配置了 params.put("x-message-ttl", 5 * 1000); 那麼這一句也能夠省略,具體根據業務須要是聲明 Queue 的時候就指定好延遲時間仍是在發送本身控制時間
            message.getMessageProperties().setExpiration(5 * 1000 + "");
            return message;
        });
        log.info("[發送時間] - [{}]", LocalDateTime.now());
    }
}

消息消費者

默認狀況下 spring-boot-data-amqp 是自動ACK機制,就意味着 MQ 會在消息消費完畢後自動幫咱們去ACK,這樣依賴就存在這樣一個問題:若是報錯了,消息不會丟失,會無限循環消費,很容易就吧磁盤空間耗完,雖然能夠配置消費的次數但這種作法也有失優雅。目前比較推薦的就是咱們手動ACK而後將消費錯誤的消息轉移到其它的消息隊列中,作補償處理。 因爲咱們須要手動控制ACK,所以下面監聽完消息後須要調用basicAck通知rabbitmq消息已被正確消費,能夠將遠程隊列中的消息刪除

package com.battcn.handler;

import com.battcn.config.RabbitConfig;
import com.battcn.entity.Book;
import com.rabbitmq.client.Channel;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import org.springframework.stereotype.Component;

import java.io.IOException;
import java.time.LocalDateTime;

/**
 * BOOK_QUEUE 消費者
 *
 * @author Levin
 * @since 2018/4/11 0011
 */
@Component
public class BookHandler {

    private static final Logger log = LoggerFactory.getLogger(BookHandler.class);

    @RabbitListener(queues = {RabbitConfig.REGISTER_QUEUE_NAME})
    public void listenerDelayQueue(Book book, Message message, Channel channel) {
        log.info("[listenerDelayQueue 監聽的消息] - [消費時間] - [{}] - [{}]", LocalDateTime.now(), book.toString());
        try {
            // TODO 通知 MQ 消息已被成功消費,能夠ACK了
            channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);
        } catch (IOException e) {
            // TODO 若是報錯了,那麼咱們能夠進行容錯處理,好比轉移當前消息進入其它隊列
        }
    }
}

主函數

package com.battcn;

import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;

/**
 * @author Levin
 */
@SpringBootApplication
public class Chapter12Application {

    public static void main(String[] args) {
        SpringApplication.run(Chapter12Application.class, args);
    }
}

測試

完成準備事項後,啓動Chapter12Application 訪問 http://localhost:8080/books 將會看到以下內容,就表明一切正常….

2018-05-23 19:56:36.248  INFO 29048 --- [nio-8080-exec-1] com.battcn.controller.BookController     : [發送時間] - [2018-05-23T19:56:36.248]
2018-05-23 19:56:41.256  INFO 29048 --- [cTaskExecutor-1] com.battcn.handler.BookHandler           : [listenerDelayQueue 監聽的消息] - [消費時間] - [2018-05-23T19:56:41.256] - [Book{id='1', name='一塊兒來學Spring Boot'}]

點關注,不迷路,這是一個程序員都想關注的公衆號

相關文章
相關標籤/搜索