1. 程式人生 > >Rabbitmq延遲佇列實現定時任務

Rabbitmq延遲佇列實現定時任務

場景

開發中經常需要用到定時任務,對於商城來說,定時任務尤其多,比如優惠券定時過期、訂單定時關閉、微信支付2小時未支付關閉訂單等等,都需要用到定時任務,但是定時任務本身有一個問題,一般來說我們都是通過定時輪詢查詢資料庫來判斷是否有任務需要執行,也就是說不管怎麼樣,我們需要先查詢資料庫,而且有些任務對時間準確要求比較高的,需要每秒查詢一次,對於系統小倒是無所謂,如果系統本身就大而且資料也多的情況下,這就不大現實了,所以需要其他方式的,當然實現的方式有多種多樣的,比如Redis實現定時佇列、基於優先順序佇列的JDK延遲佇列、時間輪等。因為我們專案中本身就使用到了Rabbitmq,所以基於方便開發和維護的原則,我們使用了Rabbitmq延遲佇列來實現定時任務,不知道rabbitmq是什麼的和不知道springboot怎麼整合Rabbitmq的可以檢視我之前的文章

Spring boot整合RabbitMQ

Rabbitmq延遲佇列

Rabbitmq本身是沒有延遲佇列的,只能通過Rabbitmq本身佇列的特性來實現,想要Rabbitmq實現延遲佇列,需要使用Rabbitmq的死信交換機(Exchange)和訊息的存活時間TTL(Time To Live)

死信交換機

一個訊息在滿足如下條件下,會進死信交換機,記住這裡是交換機而不是佇列,一個交換機可以對應很多佇列。

  1. 一個訊息被Consumer拒收了,並且reject方法的引數裡requeue是false。也就是說不會被再次放在佇列裡,被其他消費者使用。

  2. 上面的訊息的TTL到了,訊息過期了。

  3. 佇列的長度限制滿了。排在前面的訊息會被丟棄或者扔到死信路由上。

==死信交換機就是普通的交換機==,只是因為我們把過期的訊息扔進去,所以叫死信交換機,並不是說死信交換機是某種特定的交換機

訊息TTL(訊息存活時間)

訊息的TTL就是訊息的存活時間。RabbitMQ可以對佇列和訊息分別設定TTL。對佇列設定就是佇列沒有消費者連著的保留時間,也可以對每一個單獨的訊息做單獨的設定。超過了這個時間,我們認為這個訊息就死了,稱之為死信。如果佇列設定了,訊息也設定了,那麼會取小的。所以一個訊息如果被路由到不同的佇列中,這個訊息死亡的時間有可能不一樣(不同的佇列設定)。這裡單講單個訊息的TTL,因為它才是實現延遲任務的關鍵。

byte[] messageBodyBytes = "Hello, world!".getBytes();
AMQP.BasicProperties properties = new AMQP.BasicProperties();
properties.setExpiration("60000");
channel.basicPublish("my-exchange", "queue-key", properties, messageBodyBytes);

可以通過設定訊息的expiration欄位或者x-message-ttl屬性來設定時間,兩者是一樣的效果。只是expiration欄位是字串引數,所以要寫個int型別的字串:
當上面的訊息扔到佇列中後,過了60秒,如果沒有被消費,它就死了。不會被消費者消費到。這個訊息後面的,沒有“死掉”的訊息對頂上來,被消費者消費。死信在佇列中並不會被刪除和釋放,它會被統計到佇列的訊息數中去

處理流程圖

流程圖

建立交換機(Exchanges)和佇列(Queues)

建立死信交換機

建立死信交換機
如圖所示,就是建立一個普通的交換機,這裡為了方便區分,把交換機的名字取為:delay

建立自動過期訊息佇列

這個佇列的主要作用是讓訊息定時過期的,比如我們需要2小時候關閉訂單,我們就需要把訊息放進這個佇列裡面,把訊息過期時間設定為2小時
建立自動過期訊息佇列
建立一個一個名為delay_queue1的自動過期的佇列,當然圖片上面的引數並不會讓訊息自動過期,因為我們並沒有設定x-message-ttl引數,如果整個佇列的訊息有訊息都是相同的,可以設定,這裡為了靈活,所以並沒有設定,另外兩個引數x-dead-letter-exchange代表訊息過期後,訊息要進入的交換機,這裡配置的是delay,也就是死信交換機,x-dead-letter-routing-key是配置訊息過期後,進入死信交換機的routing-key,跟傳送訊息的routing-key一個道理,根據這個key將訊息放入不同的佇列

建立訊息處理佇列

這個佇列才是真正處理訊息的佇列,所有進入這個佇列的訊息都會被處理
建立訊息處理佇列
訊息佇列的名字為delay_queue2

訊息佇列繫結到交換機

進入交換機詳情頁面,將建立的2個佇列(delay_queue1和delay_queue2)繫結到交換機上面
繫結自動過期訊息佇列
自動過期訊息佇列的routing key 設定為delay

繫結delay_queue2
繫結delay_queue2
delay_queue2 的key要設定為建立自動過期的佇列的x-dead-letter-routing-key引數,這樣當訊息過期的時候就可以自動把訊息放入delay_queue2這個佇列中了

繫結後的管理頁面如下圖:
佇列繫結後

當然這個繫結也可以使用程式碼來實現,只是為了直觀表現,所以本文使用的管理平臺來操作

傳送訊息

String msg = "hello word";
MessageProperties messageProperties = new MessageProperties();
        messageProperties.setExpiration("6000");
        messageProperties.setCorrelationId(UUID.randomUUID().toString().getBytes());
        Message message = new Message(msg.getBytes(), messageProperties);
        rabbitTemplate.convertAndSend("delay", "delay",message);

主要的程式碼就是

messageProperties.setExpiration("6000");

設定了讓訊息6秒後過期

==注意:==因為要讓訊息自動過期,所以==一定不能設定delay_queue1的監聽,不能讓這個佇列裡面的訊息被接受到,否則訊息一旦被消費,就不存在過期了==

接收訊息

接收訊息配置好delay_queue2的監聽就好了

package wang.raye.rabbitmq.demo1;

import org.springframework.amqp.core.AcknowledgeMode;
import org.springframework.amqp.core.Binding;
import org.springframework.amqp.core.BindingBuilder;
import org.springframework.amqp.core.DirectExchange;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.core.Queue;
import org.springframework.amqp.rabbit.connection.CachingConnectionFactory;
import org.springframework.amqp.rabbit.connection.ConnectionFactory;
import org.springframework.amqp.rabbit.core.ChannelAwareMessageListener;
import org.springframework.amqp.rabbit.listener.SimpleMessageListenerContainer;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class DelayQueue {
    /** 訊息交換機的名字*/
    public static final String EXCHANGE = "delay";
    /** 佇列key1*/
    public static final String ROUTINGKEY1 = "delay";
    /** 佇列key2*/
    public static final String ROUTINGKEY2 = "delay_key";

    /**
     * 配置連結資訊
     * @return
     */
    @Bean
    public ConnectionFactory connectionFactory() {
        CachingConnectionFactory connectionFactory = new CachingConnectionFactory("120.76.237.8",5672);

        connectionFactory.setUsername("kberp");
        connectionFactory.setPassword("kberp");
        connectionFactory.setVirtualHost("/");
        connectionFactory.setPublisherConfirms(true); // 必須要設定
        return connectionFactory;
    }

    /**  
     * 配置訊息交換機
     * 針對消費者配置  
        FanoutExchange: 將訊息分發到所有的繫結佇列,無routingkey的概念  
        HeadersExchange :通過新增屬性key-value匹配  
        DirectExchange:按照routingkey分發到指定佇列  
        TopicExchange:多關鍵字匹配  
     */  
    @Bean  
    public DirectExchange defaultExchange() {  
        return new DirectExchange(EXCHANGE, true, false);
    } 

    /**
     * 配置訊息佇列2
     * 針對消費者配置  
     * @return
     */
    @Bean
    public Queue queue() {  
       return new Queue("delay_queue2", true); //佇列持久  

    }
    /**
     * 將訊息佇列2與交換機繫結
     * 針對消費者配置  
     * @return
     */
    @Bean  
    @Autowired
    public Binding binding() {  
        return BindingBuilder.bind(queue()).to(defaultExchange()).with(DelayQueue.ROUTINGKEY2);  
    } 

    /**
     * 接受訊息的監聽,這個監聽會接受訊息佇列1的訊息
     * 針對消費者配置  
     * @return
     */
    @Bean  
    @Autowired
    public SimpleMessageListenerContainer messageContainer2(ConnectionFactory connectionFactory) {  
        SimpleMessageListenerContainer container = new SimpleMessageListenerContainer(connectionFactory());  
        container.setQueues(queue());  
        container.setExposeListenerChannel(true);  
        container.setMaxConcurrentConsumers(1);  
        container.setConcurrentConsumers(1);  
        container.setAcknowledgeMode(AcknowledgeMode.MANUAL); //設定確認模式手工確認  
        container.setMessageListener(new ChannelAwareMessageListener() {

            public void onMessage(Message message, com.rabbitmq.client.Channel channel) throws Exception {
                byte[] body = message.getBody();  
                System.out.println("delay_queue2 收到訊息 : " + new String(body));  
                channel.basicAck(message.getMessageProperties().getDeliveryTag(), false); //確認訊息成功消費  

            }  

        });  
        return container;  
    }  


}

在訊息監聽中處理需要定時處理的任務就好了,因為Rabbitmq能傳送訊息,所以可以把任務特徵碼發過來,比如關閉訂單就把訂單id發過來,這樣就避免了需要查詢一下那些訂單需要關閉而加重MySQL負擔了,畢竟一旦訂單量大的話,查詢本身也是一件很費IO的事情

總結

基於Rabbitmq實現定時任務,就是將訊息設定一個過期時間,放入一個沒有讀取的佇列中,讓訊息過期後自動轉入另外一個佇列中,監控這個佇列訊息的監聽處來處理定時任務具體的操作