Spring集成RabbitMQ测试

1、发送消息

发送消息首先要获取与rabbitmq-server的连接，然后从渠道(chann)中指定的queue发送消息 ， 不能定义两个queue名字相同，但属性不同

示例：

Sender01.java

package com.zf.rabbitmq01;

import java.io.IOException;

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

/**
 * 发送消息
 * @author zhoufeng
 *
 */
public class Sender01 {
    
    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();
        
    }

}

Recv01.java

package com.zf.rabbitmq01;

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 Recv01 {

    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 queueName = "queue01";
        
        channel.queueDeclare(queueName, false, false, false, null) ;
        
        //上面的部分，与Sender01是一样的
        
        //配置好获取消息的方式
        QueueingConsumer consumer = new QueueingConsumer(channel) ;
        channel.basicConsume(queueName, true, consumer) ;
        
        //循环获取消息
        while(true){
            
            //获取消息，如果没有消息，这一步将会一直阻塞
            Delivery delivery = consumer.nextDelivery() ;
            
            String msg = new String(delivery.getBody()) ;  
            
            System.out.println("received message[" + msg + "] from " + queueName);
        }
        
    }
    
}

此时，无论先后启动哪个类  ，都没有关系 ，如果执行Sender01时  Recv01还没有启动 ， 那么消息将被保存在RabbitMQ-Server上面，直到Recv01启动后获取，才会被移除

从本节开始称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);可以设置.

配置实例:

<rabbit:template exchange="my-mq-exchange" id="amqpTemplate"  connection-factory="connectionFactory"  message-converter="jsonMessageConverter" />

   <!-- 消息对象json转换类 -->
   <bean id="jsonMessageConverter" class="org.springframework.amqp.support.converter.Jackson2JsonMessageConverter" />  
   
   <!-- queue 队列声明-->
   <rabbit:queue id="my_queue" name="my_queue" durable="true" auto-delete="false" exclusive="false" />

   <!-- queue 队列声明-->
   <rabbit:queue id="order_queue" name="order_queue" durable="true" auto-delete="false" exclusive="false" />
   
<!--     交换机定义 -->
   <rabbit:direct-exchange name="my-mq-exchange" durable="true" auto-delete="false" id="my-mq-exchange">
    <rabbit:bindings>
           <rabbit:binding queue="order_queue" key="order_queue_key"/>
    </rabbit:bindings>
</rabbit:direct-exchange>

   <task:executor id="orderExecutor" pool-size="20" queue-capacity="50" rejection-policy="CALLER_RUNS"/>

   <rabbit:listener-container connection-factory="connectionFactory" prefetch="1" acknowledge="manual"  concurrency="2" task-executor="orderExecutor">
       <rabbit:listener queues="order_queue" ref="orderQueueListener"/>
   </rabbit:listener-container>

   <bean id="orderQueueListener" class="com.zwh.core.mq.OrderMQConsumer" />

•Concurrency  指定并发消费线程数

•task-executor  线程执行器创建消费线程负责具体执行，可创建线程池，可缓存本地缓存队列

•Prefetch     每个消费者一次性从broker里面取的待消费消息的个数

•Exclusive    默认为false,如果设置为true，concurrency就必须设置为1，即只能单个消费者消费队列里的消息，适用于必须严格执行消息队列的消费顺序（先进先出）。

•rejection-policy 线程执行器达到最大线程执行的拒绝策略

     四个选择：

    ThreadPoolExecutor.AbortPolicy()             ---   抛出java.util.concurrent.RejectedExecutionException异常

   ThreadPoolExecutor.CallerRunsPolicy()      ---   重试添加当前的任务，他会自动重复调用execute()方法

    ThreadPoolExecutor.DiscardOldestPolicy()  ---   抛弃旧的任务

   ThreadPoolExecutor.DiscardPolicy()           ---   抛弃当前的任务

生产者：

@Component
public class OrderMQProducer extends Producer<InvestReq> {

    @Autowired
    private AmqpTemplate amqpTemplate;

    private final static Logger LOG = LoggerFactory.getLogger(Producer.class);

    public void sendDataToQueue(String queueKey, InvestReq object) throws Exception{
            amqpTemplate.convertAndSend(queueKey, object);
    }

}

消费者：

public class OrderMQConsumer implements ChannelAwareMessageListener {

    private static Logger logger = LoggerFactory.getLogger(OrderMQConsumer.class);

    private static Long startTime = null;

    @Autowired
    private InvestService investService;

    public void onMessage(Message  message,Channel channel) throws IOException {
        if(startTime == null)startTime = System.currentTimeMillis();
        try{
            Jackson2JsonMessageConverter converter = new Jackson2JsonMessageConverter();
            Object o = converter.fromMessage(message);
            if (o instanceof InvestReq) {
                System.out.println("---"+((InvestReq) o).getId());
            }
        }catch (Exception e){
            logger.error("队列订单处理失败：【{}】", message.toString());
        }finally {
            channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);
            System.out.println("----"+ (System.currentTimeMillis() - startTime));
        }
    }

}