rabbitMQ學習筆記(三) 消息確認與公平調度消費者

從本節開始稱Sender為生產者 ， Recv為消費者

 

一、消息確認

為了確保消息一定被消費者處理，rabbitMQ提供了消息確認功能，就是在消費者處理完任務之后，就給服務器一個回饋，服務器就會將該消息刪除，如果消費者超時不回饋，那么服務器將就將該消息重新發送給其他消費者

默認是開啟的，在消費者端通過下面的方式開啟消息確認,  首先將autoAck自動確認關閉，等我們的任務執行完成之后，手動的去確認，類似JDBC的autocommit一樣

QueueingConsumer consumer = new QueueingConsumer(channel);
boolean autoAck = false;
channel.basicConsume("hello", autoAck, consumer);

在前面的例子中使用的是channel.basicConsume(channelName, true, consumer) ; 在接收到消息后，就會自動反饋一個消息給服務器。

下面這個例子來測試消息確認的功能。

Sender03.java

package com.zf.rabbitmq03;

import java.io.IOException;

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

/**
 * 發送消息
 * @author zhoufeng
 *
 */
public class Sender03 {
    
    public static void main(String[] args) throws IOException {
        
        
        ConnectionFactory connFac = new ConnectionFactory() ;
        
        //RabbitMQ-Server安裝在本機，所以直接用127.0.0.1
        connFac.setHost("127.0.0.1");
        
        //創建一個連接
        Connection conn = connFac.newConnection() ;
        
        //創建一個渠道
        Channel channel = conn.createChannel() ;
        
        //定義Queue名稱
        String queueName = "queue01" ;
        
        //為channel定義queue的屬性，queueName為Queue名稱
        channel.queueDeclare( queueName , false, false, false, null) ;
        
        String msg = "Hello World!";
        
        //發送消息
        channel.basicPublish("", queueName , null , msg.getBytes());
        
        System.out.println("send message[" + msg + "] to "+ queueName +" success!");
        
        channel.close(); 
        conn.close(); 
        
    }

}

與Sender01.java一樣，沒有什么區別。

Recv03.java

package com.zf.rabbitmq03;

import java.io.IOException;

import com.rabbitmq.client.Channel;
import com.rabbitmq.client.Connection;
import com.rabbitmq.client.ConnectionFactory;
import com.rabbitmq.client.ConsumerCancelledException;
import com.rabbitmq.client.QueueingConsumer;
import com.rabbitmq.client.QueueingConsumer.Delivery;
import com.rabbitmq.client.ShutdownSignalException;

/**
 * 接收消息
 * @author zhoufeng
 *
 */
public class Recv03 {

    public static void main(String[] args) throws IOException, ShutdownSignalException, ConsumerCancelledException, InterruptedException {
        
        ConnectionFactory connFac = new ConnectionFactory() ;
        
        connFac.setHost("127.0.0.1");
        
        Connection conn = connFac.newConnection() ;
        
        Channel channel = conn.createChannel() ;
        
        String channelName = "channel01";
        
        channel.queueDeclare(channelName, false, false, false, null) ;
        
        
        //配置好獲取消息的方式
        QueueingConsumer consumer = new QueueingConsumer(channel) ;
        

        //取消 autoAck
        boolean autoAck = false ;
        
        channel.basicConsume(channelName, autoAck, consumer) ;
        
        //循環獲取消息
        while(true){
            
            //獲取消息，如果沒有消息，這一步將會一直阻塞
            Delivery delivery = consumer.nextDelivery() ;
            
            String msg = new String(delivery.getBody()) ;  
            
            //確認消息，已經收到
            channel.basicAck(delivery.getEnvelope().getDeliveryTag()
                    , false);
            
            System.out.println("received message[" + msg + "] from " + channelName);
        }
        
    }
    
}

注意：一旦將autoAck關閉之后，一定要記得處理完消息之后，向服務器確認消息。否則服務器將會一直轉發該消息

如果將上面的channel.basicAck(delivery.getEnvelope().getDeliveryTag(), false);注釋掉， Sender03.java只需要運行一次 ， Recv03.java每次運行將都會收到HelloWorld消息

注意：

但是這樣還是不夠的，如果rabbitMQ-Server突然掛掉了，那么還沒有被讀取的消息還是會丟失 ，所以我們可以讓消息持久化。 只需要在定義Queue時，設置持久化消息就可以了，方法如下：

boolean durable = true;
channel.queueDeclare(channelName, durable, false, false, null);

這樣設置之后，服務器收到消息后就會立刻將消息寫入到硬盤，就可以防止突然服務器掛掉，而引起的數據丟失了。  但是服務器如果剛收到消息，還沒來得及寫入到硬盤，就掛掉了，這樣還是無法避免消息的丟失。

 

二、公平調度

上一個例子能夠實現發送一個Message與接收一個Message

從上一個Recv01中可以看出，必須處理完一個消息，才會去接收下一個消息。如果生產者眾多，那么一個消費者肯定是忙不過來的。此時就可以用多個消費者來對同一個Channel的消息進行處理，並且要公平的分配任務給多個消費者。不能部分很忙  部分總是空閑

實現公平調度的方式就是讓每個消費者在同一時刻會分配一個任務。 通過channel.basicQos(1);可以設置