使用spring-boot-starter-amqp开发生产者应用

上一篇我们介绍了如何使用spring AMQP和RabbitMQ结合,开发消费者应用程序,使用的是Xml配置的Spring框架。

本篇我们仍然使用Spring AMQP开发生产者应用,不过我们使用零 XML配置的Spring Boot环境进行开发,使用的库是spring-boot-starter-amqp库。

 

使用Spring-boot-starter-amqp搭建框架

我们使用Spring Boot 1.4.3 RELEASE版本,pom.xml文件内容如下:
xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
    xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0modelVersion>
    <groupId>com.dxzgroupId>
    <artifactId>rabbitmqstudyartifactId>
    <version>0.0.1-SNAPSHOTversion>
    <packaging>jarpackaging>

    <name>rabbitmqstudyname>
    <url>http://maven.apache.orgurl>

    <parent>
        <groupId>org.springframework.bootgroupId>
        <artifactId>spring-boot-starter-parentartifactId>
        <version>1.4.3.RELEASEversion>
        <relativePath />
    parent>

    <properties>
        <project.build.sourceEncoding>UTF-8project.build.sourceEncoding>
        <project.reporting.outputEncoding>UTF-8project.reporting.outputEncoding>
        <java.version>1.8java.version>
    properties>

    <dependencies>
        <dependency>
            <groupId>org.springframework.bootgroupId>
            <artifactId>spring-boot-starter-webartifactId>
        dependency>

        <dependency>
            <groupId>org.springframework.bootgroupId>
            <artifactId>spring-boot-starter-testartifactId>
            <scope>testscope>
        dependency>

        <dependency>
            <groupId>org.springframework.bootgroupId>
            <artifactId>spring-boot-starter-amqpartifactId>
        dependency>

        <dependency>
            <groupId>org.springframework.bootgroupId>
            <artifactId>spring-boot-configuration-processorartifactId>
            <optional>trueoptional>
        dependency>
        <dependency>
            <groupId>com.rabbitmqgroupId>
            <artifactId>rabbitmq-clientartifactId>
            <version>1.3.0version>
        dependency>
        <dependency>
            <groupId>org.springframework.amqpgroupId>
            <artifactId>spring-rabbitartifactId>
            <version>1.7.2.RELEASEversion>
        dependency>
        <dependency>
            <groupId>org.springframework.amqpgroupId>
            <artifactId>spring-amqpartifactId>
            <version>1.7.2.RELEASEversion>
        dependency>
        <dependency>
            <groupId>junitgroupId>
            <artifactId>junitartifactId>
            <version>3.8.1version>
            <scope>testscope>
        dependency>
    dependencies>

    <build>
        <plugins>
            <plugin>
                <groupId>org.springframework.bootgroupId>
                <artifactId>spring-boot-maven-pluginartifactId>
            plugin>
        plugins>
    build>

project>
需要注意的是,当使用spring-boot-configuration-processor库时会默认打开RabbitAutoConfiguration,它会自动创建一个 CachingConnectionFactory对象,一个RabbitTemplate对象,一个AmqpAdmin对象以及一个RabbitMessageTemplate对象。
 
[java]  view plain  copy
 
  1. "code" class="java">@Configuration  
  2. @ConditionalOnClass({ RabbitTemplate.class, Channel.class })  
  3. @EnableConfigurationProperties(RabbitProperties.class)  
  4. @Import(RabbitAnnotationDrivenConfiguration.class)  
  5. public class RabbitAutoConfiguration {  
  6.     @Configuration  
  7.     @ConditionalOnMissingBean(ConnectionFactory.class)  
  8.     protected static class RabbitConnectionFactoryCreator {  
  9.         @Bean  
  10.         public CachingConnectionFactory rabbitConnectionFactory(RabbitProperties config)  
  11.                 {  
  12.                  ........  
  13.    
  14.         @Configuration  
  15.     @Import(RabbitConnectionFactoryCreator.class)  
  16.     protected static class RabbitTemplateConfiguration {  
  17.            @Bean  
  18.        @ConditionalOnSingleCandidate(ConnectionFactory.class)  
  19.        @ConditionalOnMissingBean(RabbitTemplate.class)  
  20.        public RabbitTemplate rabbitTemplate(ConnectionFactory connectionFactory) {  
  21.            ........  
  22.    
  23.         @Bean  
  24.     @ConditionalOnSingleCandidate(ConnectionFactory.class)  
  25.     @ConditionalOnProperty(prefix = "spring.rabbitmq", name = "dynamic", matchIfMissing = true)  
  26.     @ConditionalOnMissingBean(AmqpAdmin.class)  
  27.     public AmqpAdmin amqpAdmin(ConnectionFactory connectionFactory) {  
  28.         return new RabbitAdmin(connectionFactory);  
  29.     }  
  30.    
  31.         .......  
  32.         @Configuration  
  33.     @ConditionalOnClass(RabbitMessagingTemplate.class)  
  34.     @ConditionalOnMissingBean(RabbitMessagingTemplate.class)  
  35.     @Import(RabbitTemplateConfiguration.class)  
  36.     protected static class MessagingTemplateConfiguration {  
  37.    
  38.     @Bean  
  39.     @ConditionalOnSingleCandidate(RabbitTemplate.class)  
  40.     public RabbitMessagingTemplate rabbitMessagingTemplate(  
  41.             RabbitTemplate rabbitTemplate) {  
  42.         return new RabbitMessagingTemplate(rabbitTemplate);  
  43.     }  
  44.   
  45. "font-family:Arial,sans-serif; color:rgb(51,51,51); font-size:10px">如果我们想定制自己的ConnectionFactory,RabbitTemplate等对象,建议在Configuration中将其禁用"color:rgb(51,51,51); background-color:rgb(251,250,249); font-size:17.5px">"font-family:Consolas,Andale Mono WT,Andale Mono,Bitstream Vera Sans Mono,Nimbus Mono L,Monaco,Courier New,monospace">:  
  46.   
  47.   
  48.   
  49.   
[java]  view plain  copy
 
  1. "font-size:10px;">@Configuration  
  2. @ComponentScan("com.qf.rabbitmq")  
  3. @EnableAutoConfiguration(exclude = RabbitAutoConfiguration.class)  
  4. @SpringBootApplication  
  5. public class SpringMqProducerApplication {  
  6.    
  7.     public static void main(String[] args) {  
  8.         SpringApplication.run(SpringMqProducerApplication.class, args);  
  9.     }  
  10. }  

在src/main/resources目录下创建application.properties文件,添加rabbitmq相关设置,以spring.rabbitmq为前缀,这样它会自动映射到RabbitProperties配置类
[plain]  view plain  copy
 
  1. spring.rabbitmq.host=localhost  
  2. spring.rabbitmq.port=5672  
  3. spring.rabbitmq.username=rabbitmq_producer  
  4. spring.rabbitmq.password=123456  
  5. spring.rabbitmq.virtualHost=test_vhosts  

添加RabbitConfig类,用于创建自定义ConnectionFactory,RabbitAdmin,RabbitTemplate等。 这个类头部如下所示
[java]  view plain  copy
 
  1. @Configuration  
  2. @EnableConfigurationProperties(RabbitProperties.class)  
  3. public class RabbitConfig  
  4. {  
  5.     @Autowired  
  6.     private RabbitProperties rabbitProperties;  

这里用到了org.springframework.boot.autoconfigure.amqp.RabbitProperties类,记录application.properties中的RabbitMQ连接设置。

再创建ConnectionFactory:

 

[java]  view plain  copy
 
  1. @Bean("connectionFactory")  
  2.  public ConnectionFactory getConnectionFactory() {  
  3.         com.rabbitmq.client.ConnectionFactory rabbitConnectionFactory =  
  4.                 new com.rabbitmq.client.ConnectionFactory();  
  5.         rabbitConnectionFactory.setHost(rabbitProperties.getHost());  
  6.         rabbitConnectionFactory.setPort(rabbitProperties.getPort());  
  7.         rabbitConnectionFactory.setUsername(rabbitProperties.getUsername());  
  8.         rabbitConnectionFactory.setPassword(rabbitProperties.getPassword());  
  9.         rabbitConnectionFactory.setVirtualHost(rabbitProperties.getVirtualHost());  
  10.    
  11.         ConnectionFactory connectionFactory = new CachingConnectionFactory(rabbitConnectionFactory);  
  12.         return connectionFactory;  
  13.  }  
这里可以像消费者应用里提到的那样,设置CacheMode和最大缓存数,我们这里暂不设置。 再创建RabbitAdmin对象

 

 

[java]  view plain  copy
 
  1. @Bean(name="rabbitAdmin")  
  2. public RabbitAdmin getRabbitAdmin()  
  3. {  
  4.     RabbitAdmin rabbitAdmin = new RabbitAdmin(getConnectionFactory());  
  5.     rabbitAdmin.setAutoStartup(true);  
  6.     return rabbitAdmin;  
  7. }  
定义MessageConverter和MessagePropertiesConverter对象

 

 

[java]  view plain  copy
 
  1. @Bean(name="serializerMessageConverter")  
  2. public MessageConverter getMessageConverter(){  
  3.         return new SimpleMessageConverter();  
  4. }  
  5.    
  6. @Bean(name="messagePropertiesConverter")  
  7. public MessagePropertiesConverter getMessagePropertiesConverter()  
  8. {  
  9.     return new DefaultMessagePropertiesConverter();  
  10. }  
定义发送消息所用的RabbitTemplate对象,由于我们的场景要求发送之后立即从消费者处获得返回消息,因此我们在RabbitTemplate对象中设置了ReplyAddress,而且在下面的MessageListenerContainer中将这个对象作为Listener设置到Container中。

[java]  view plain  copy
 
  1. @Bean(name="rabbitTemplate")  
  2. public RabbitTemplate getRabbitTemplate()  
  3. {  
  4.    RabbitTemplate rabbitTemplate = new RabbitTemplate(getConnectionFactory());  
  5.    rabbitTemplate.setUseTemporaryReplyQueues(false);  
  6.    rabbitTemplate.setMessageConverter(getMessageConverter());  
  7.    rabbitTemplate.setMessagePropertiesConverter(getMessagePropertiesConverter());  
  8.    rabbitTemplate.setReplyAddress(AppConstants.REPLY_QUEUE_NAME);  
  9.    rabbitTemplate.setReceiveTimeout(60000);  
  10.    return rabbitTemplate;  
  11. }  
使用RabbitAdmin对象定义发送消息Exchange/Queue/Binding,返回消息Exchange/Queue/Binding对象,如果它们在RabbitMQ中已经存在,下面的定义代码可以省略.这里的Exchange/Queue名称和前面的消费者应用使用的相同。

 

 

[java]  view plain  copy
 
  1. @Bean(name="springMessageQueue")      
  2. public Queue createQueue(@Qualifier("rabbitAdmin")RabbitAdmin rabbitAdmin)  
  3. {  
  4.     Queue sendQueue = new Queue(AppConstants.SEND_QUEUE_NAME,true,false,false);  
  5.     rabbitAdmin.declareQueue(sendQueue);  
  6.     return sendQueue;  
  7. }  
  8.    
  9. @Bean(name="springMessageExchange")  
  10. public Exchange createExchange(@Qualifier("rabbitAdmin")RabbitAdmin rabbitAdmin)  
  11. {  
  12.     DirectExchange sendExchange = new DirectExchange(AppConstants.SEND_EXCHANGE_NAME,true,false);  
  13.     rabbitAdmin.declareExchange(sendExchange);  
  14.     return sendExchange;  
  15. }  
  16.    
  17. @Bean(name="springMessageBinding")  
  18. public Binding createMessageBinding(@Qualifier("rabbitAdmin")RabbitAdmin rabbitAdmin)  
  19. {  
  20.     Map arguments = new HashMap();  
  21.     Binding sendMessageBinding =  
  22.             new Binding(AppConstants.SEND_QUEUE_NAME, Binding.DestinationType.QUEUE,  
  23.                         AppConstants.SEND_EXCHANGE_NAME, AppConstants.SEND_MESSAGE_KEY, arguments);  
  24.     rabbitAdmin.declareBinding(sendMessageBinding);  
  25.     return sendMessageBinding;  
  26. }  
  27.    
  28. @Bean(name="springReplyMessageQueue")  
  29. public Queue createReplyQueue(@Qualifier("rabbitAdmin")RabbitAdmin rabbitAdmin)  
  30. {  
  31.     Queue replyQueue = new Queue(AppConstants.REPLY_QUEUE_NAME,true,false,false);  
  32.     rabbitAdmin.declareQueue(replyQueue);  
  33.     return replyQueue;  
  34. }  
  35.    
  36. @Bean(name="springReplyMessageExchange")  
  37. public Exchange createReplyExchange(@Qualifier("rabbitAdmin")RabbitAdmin rabbitAdmin)  
  38. {  
  39.     DirectExchange replyExchange = new DirectExchange(AppConstants.REPLY_EXCHANGE_NAME,true,false);  
  40.     rabbitAdmin.declareExchange(replyExchange);  
  41.     return replyExchange;  
  42. }  
  43.    
  44. @Bean(name="springReplyMessageBinding")  
  45. public Binding createReplyMessageBinding(@Qualifier("rabbitAdmin")RabbitAdmin rabbitAdmin)  
  46. {  
  47.     Map arguments = new HashMap();  
  48.     Binding replyMessageBinding =  
  49.             new Binding(AppConstants.REPLY_QUEUE_NAME, Binding.DestinationType.QUEUE,  
  50.                         AppConstants.REPLY_EXCHANGE_NAME, AppConstants.REPLY_MESSAGE_KEY, arguments);  
  51.     rabbitAdmin.declareBinding(replyMessageBinding);  
  52.     return replyMessageBinding;  
  53. }  
最后定义接收返回消息的Message Listener Container,这里的Listener属性设置的是上面创建的RabbitTemplate对象。

 

[java]  view plain  copy
 
  1. @Bean(name="replyMessageListenerContainer")  
  2. public SimpleMessageListenerContainer createReplyListenerContainer() {  
  3.      SimpleMessageListenerContainer listenerContainer = new SimpleMessageListenerContainer();  
  4.      listenerContainer.setConnectionFactory(getConnectionFactory());  
  5.      listenerContainer.setQueueNames(AppConstants.REPLY_QUEUE_NAME);  
  6.      listenerContainer.setMessageConverter(getMessageConverter());  
  7.      listenerContainer.setMessageListener(getRabbitTemplate());  
  8.      listenerContainer.setRabbitAdmin(getRabbitAdmin());  
  9.      listenerContainer.setAcknowledgeMode(AcknowledgeMode.AUTO);  
  10.      return listenerContainer;  
  11. }  
我们还定义了一个ThreadPoolExecutor对象,用于RabbitTemplate异步执行,发送和接收消息使用。
[java]  view plain  copy
 
  1. @Bean(name="threadExecutor")  
  2. public ThreadPoolTaskExecutor createThreadPoolTaskExecutor()  
  3. {  
  4.     ThreadPoolTaskExecutor threadPoolTaskExecutor =  
  5.             new ThreadPoolTaskExecutor();  
  6.     threadPoolTaskExecutor.setCorePoolSize(5);  
  7.     threadPoolTaskExecutor.setMaxPoolSize(10);  
  8.     threadPoolTaskExecutor.setQueueCapacity(200);  
  9.     threadPoolTaskExecutor.setKeepAliveSeconds(20000);  
  10.     return threadPoolTaskExecutor;  
  11. }  

我们的生产者-消费者场景是系列1中提到的RPC方式计算阶乘,使用Rest接口调用生产者应用,生产者应用发送消息给消费者应用,计算出阶乘结果返回给生产者。 
为了实现这个场景,我们先在生产者程序中定义用于发送请求消息和接收返回消息的服务接口SendMessageService和它的实现类SendMessageServiceImpl类。
[java]  view plain  copy
 
  1. public interface SendMessageService  
  2. {  
  3.     String sendMessage(String message);  
  4. }  
  5.    
  6. @Service("sendMessageService")  
  7. public class SendMessageServiceImpl implements SendMessageService  
  8. {  
  9.     @Autowired  
  10.     private ThreadPoolTaskExecutor executor;  
  11.    
  12.     public String sendMessage(String message)  
  13.     {  
  14.         CompletableFuture resultCompletableFuture =  
  15.                 CompletableFuture.supplyAsync(new MQSenderSupplier(message), executor);  
  16.         try  
  17.         {  
  18.             String result = resultCompletableFuture.get();  
  19.             return result;  
  20.         }  
  21.         catch (InterruptedException e)  
  22.         {  
  23.             e.printStackTrace();  
  24.         }  
  25.         catch (ExecutionException e)  
  26.         {  
  27.             e.printStackTrace();  
  28.         }  
  29.         return null;  
  30.     }  
  31. }  
在实现类中,我们将发送请求消息和接收返回消息交给一个Supplier对象(JDK 1.8引入)在后台线程池中异步执行,我们自定义了MQSenderSupplier类,实现了发送请求和接收返回消息的操作。
[java]  view plain  copy
 
  1. public class  MQSenderSupplier implements Supplier {  
  2.    
  3.     private String message;  
  4.    
  5.     public MQSenderSupplier(String message)  
  6.     {  
  7.         this.message = message;  
  8.     }  
  9.    
  10.     public String get()  
  11.     {  
  12.         Date sendTime = new Date();  
  13.         String correlationId = UUID.randomUUID().toString();  
  14.    
  15.         MessagePropertiesConverter messagePropertiesConverter =  
  16.                 (MessagePropertiesConverter) ApplicationContextUtil.getBean("messagePropertiesConverter");  
  17.    
  18.         RabbitTemplate rabbitTemplate =  
  19.                  (RabbitTemplate)ApplicationContextUtil.getBean("rabbitTemplate");  
  20.    
  21.         AMQP.BasicProperties props =  
  22.                 new AMQP.BasicProperties("text/plain",  
  23.                         "UTF-8",  
  24.                         null,  
  25.                         2,  
  26.                         0, correlationId, AppConstants.REPLY_EXCHANGE_NAME, null,  
  27.                         null, sendTime, null, null,  
  28.                         "SpringProducer", null);  
  29.    
  30.         MessageProperties sendMessageProperties =  
  31.                 messagePropertiesConverter.toMessageProperties(props, null,"UTF-8");  
  32.         sendMessageProperties.setReceivedExchange(AppConstants.REPLY_EXCHANGE_NAME);  
  33.         sendMessageProperties.setReceivedRoutingKey(AppConstants.REPLY_MESSAGE_KEY);  
  34.         sendMessageProperties.setRedelivered(true);  
  35.    
  36.         Message sendMessage = MessageBuilder.withBody(message.getBytes())  
  37.                 .andProperties(sendMessageProperties)  
  38.                 .build();  
  39.    
  40.         Message replyMessage =  
  41.                 rabbitTemplate.sendAndReceive(AppConstants.SEND_EXCHANGE_NAME,  
  42.                         AppConstants.SEND_MESSAGE_KEY, sendMessage);  
  43.    
  44.         String replyMessageContent = null;  
  45.         try {  
  46.             replyMessageContent = new String(replyMessage.getBody(),"UTF-8");  
  47.         } catch (UnsupportedEncodingException e) {  
  48.             e.printStackTrace();  
  49.         }     
  50.         return replyMessageContent;  
  51.     }  
  52. }                
由于这里的MQSenderSupplier对象是临时构建,因此它使用的RabbitTemplate和MessagePropertiesConverter对象都是从Spring Context环境中直接获取,而没有使用自动注入方式。ApplicationContextUtil是我们定义的一个Context辅助类(类似代码可以从网上搜索到,这里就不列出了),使用RabbitTemplate.sendAndReceive方法发送请求消息和接收返回消息,没有另行定义Listener类。
再添加发送消息的Rest接口类SendMessageController
[java]  view plain  copy
 
  1. @RestController  
  2. public class SendMessageController  
  3. {  
  4.     @Autowired  
  5.     private SendMessageService sendMessageService;  
  6.    
  7.     @RequestMapping(value = "/caculate", method = RequestMethod.POST)  
  8.     @ResponseBody  
  9.     public String sendMessage(@RequestBody String number)  
  10.     {  
  11.         String result =  
  12.                 sendMessageService.sendMessage(number);  
  13.          return "Factorial(" + number + ") = " + result;  
  14.     }  
  15. }  
修改消费者应用程序,添加一个用于计算阶乘的Message Listener,绑定springMessageQueue队列,替换之前定义的Message Listener。
[java]  view plain  copy
 
  1. "font-size:10px;">public class CaculateListener implements ChannelAwareMessageListener {  
  2.    
  3.     @Autowired  
  4.     private MessagePropertiesConverter messagePropertiesConverter;  
  5.    
  6.     @Autowired  
  7.     private RabbitTemplate rabbitTemplate;  
  8.    
  9.     public void onMessage(Message message, Channel channel) throws Exception   
  10.     {  
  11.         MessageProperties messageProperties = message.getMessageProperties();  
  12.         AMQP.BasicProperties rabbitMQProperties =  
  13.                 messagePropertiesConverter.fromMessageProperties(messageProperties, "UTF-8");  
  14.         String numberContent = null;  
  15.         numberContent = new String(message.getBody(),"UTF-8");  
  16.         System.out.println("The received number is:" + numberContent);  
  17.         String consumerTag = messageProperties.getConsumerTag();  
  18.         int number = Integer.parseInt(numberContent);  
  19.    
  20.         String result = factorial(number);  
  21.    
  22.         AMQP.BasicProperties replyRabbitMQProps =  
  23.                 new AMQP.BasicProperties("text/plain",  
  24.                         "UTF-8",  
  25.                         null,  
  26.                         2,  
  27.                         0, rabbitMQProperties.getCorrelationId(), null, null,  
  28.                         null, null, null, null,  
  29.                         consumerTag, null);  
  30.         Envelope replyEnvelope =  
  31.                 new Envelope(messageProperties.getDeliveryTag(), true,  
  32.                         "springReplyMessageExchange", "springReplyMessage");  
  33.    
  34.         MessageProperties replyMessageProperties =  
  35.                 messagePropertiesConverter.toMessageProperties(replyRabbitMQProps,  
  36.                         replyEnvelope,"UTF-8");  
  37.    
  38.         Message replyMessage = MessageBuilder.withBody(result.getBytes())  
  39.                 .andProperties(replyMessageProperties)  
  40.                 .build();  
  41.    
  42.         rabbitTemplate.send("springReplyMessageExchange","springReplyMessage", replyMessage);  
  43.         channel.basicAck(messageProperties.getDeliveryTag(), false);  
  44.     }  
  45.     ..........  

这里计算阶乘的方法factorial和系列1中相同,代码不再列出。

修改applicationContext.xml文件,替换Message Listener Container中的Listener对象为CaculateListener对象。

 

[html]  view plain  copy
 
  1. <bean id="caculateListener"  
  2.             class="com.qf.rabbitmq.listener.CaculateListener" />  
  3.    
  4. <rabbit:listener-container message-converter="serializerMessageConverter"  
  5.                                ..................  
  6.                                recovery-back-off="backOff">  
  7.         <rabbit:listener ref="caculateListener" queues="springMessageQueue"/>  
  8. rabbit:listener-container>  

修改完成后启动消费者和生产者程序,使用生产者的rest接口计算20的阶乘值(生产者程序使用8180端口启动),我们可以看到消费者应用计算出了20的阶乘结果,并返回给生产者应用。

 

假如消费者应用计算时间较长,生产者是否还能获取正确的结算结果呢?
 
我们修改消费者应用,在计算后睡眠10s,再返回计算结果。 重新启动消费者和生产者程序,调用计算阶乘的rest接口,我们会看到生产者端抛出了空引用异常,没有获取到返回消息。
 


这是因为RabbitTemplate对象默认的接收Reply消息的超时时间只有5000ms(5s),超过这个时间sendAndReceive方法会直接返回空对象。 查看RabbitTemplate的代码,我们可以看到是内置的PendingReply对象从它自身的阻塞队列queue中获取返回消息。 RabbitTemplate对象是在onMessage方法中获取到返回消息时才把消息放入这个阻塞队列的。
 
[java]  view plain  copy
 
  1. public class RabbitTemplate {  
  2.   //默认Reply Timeout时间5000ms  
  3.   private volatile long replyTimeout = DEFAULT_REPLY_TIMEOUT;  
  4.   ..............  
  5.   //这个方法被sendAndReceive调用,实际处理消息的发送和返回消息的接收。  
  6.   private Message exchangeMessages(final String exchange, final String routingKey, final Message message,  
  7.     final CorrelationData correlationData, Channel channel, final PendingReply pendingReply, String messageTag)  
  8.         throws Exception {  
  9.                 ............          
  10.         //发送消息到阻塞队列  
  11.                 doSend(channel, exchange, routingKey, message, mandatory, correlationData);  
  12.                 //从pendingReply的阻塞队列queue中获取返回消息。  
  13.         reply = this.replyTimeout < 0 ? pendingReply.get() : pendingReply.get(this.replyTimeout, TimeUnit.MILLISECONDS);  
  14.         return reply;  
  15.     }  
  16.    
  17.   public void onMessage(Message message) {  
  18.         try {  
  19.             String messageTag;  
  20.                         ...............  
  21.                         PendingReply pendingReply = this.replyHolder.get(messageTag);  
  22.                         ...............  
  23.                         message.getMessageProperties().setReplyTo(savedReplyTo);  
  24.                         //PendingReply对象将返回消息放入阻塞队列中  
  25.             pendingReply.reply(message);  
  26.   .............  
  27.   private static class PendingReply {  
  28.   .............  
  29.   //从阻塞队列中获取返回消息对象。  
  30.   public Message get(long timeout, TimeUnit unit) throws InterruptedException   
  31.   {  
  32.     Object reply = this.queue.poll(timeout, unit);  
  33.         //如果返回消息不为空,将返回对象转换为Amqp Message对象。  
  34.     return reply == null ? null : processReply(reply);  
  35.   }  
  36.   private Message processReply(Object reply)   
  37.   {  
  38.       if (reply instanceof Message) {  
  39.     return (Message) reply;  
  40.       }   
  41.       ............        
  42.   }  
  43.    
  44.   public void reply(Message reply)   
  45.   {  
  46.     this.queue.add(reply);  
  47.   }  
由于我们设置的消费者应用处理时间为10000ms,导致PendingReply从阻塞队列取返回消息时,onMessage方法还没有被触发,阻塞队列为空,因此返回的消息对象为空,从而抛出异常。 
为了解决这个问题,我们可以把RabbitTemplate的replyTimeout时间设置的稍微长一点,例如1分钟(60000ms)。
[java]  view plain  copy
 
  1. @Bean(name="rabbitTemplate")  
  2. public RabbitTemplate getRabbitTemplate()  
  3. {  
  4.  ................  
  5.  rabbitTemplate.setReplyTimeout(60000);  
  6.  return rabbitTemplate;  
  7. }  
再次启动生产者程序,计算30的阶乘值,我们看到空引用异常不再抛出,阶乘计算的结果也正常返回。

使用AsyncRabbitTemplate发送异步消息

使用RabbitTemplate.sendAndReceive方法发送消息给消费者,获得返回消息,这是一个同步的过程,发送消息和等待接收消息是 在同一个线程中执行,如果等待返回消息的时间过长,当前方法的后续操作将被卡住,无法执行。 
为了验证这一点,我们在生产者程序中添加日志输出代码。
[java]  view plain  copy
 
  1. Message replyMessage = rabbitTemplate.sendAndReceive(AppConstants.SEND_EXCHANGE_NAME,  
  2.                         AppConstants.SEND_MESSAGE_KEY, sendMessage);  
  3.    
  4. logger.info("Send the caculate number to consumer");  
  5.    
  6. String replyMessageContent = null;  
  7. try {  
  8.        replyMessageContent = new String(replyMessage.getBody(),"UTF-8");  
  9.        logger.info("The reply message is:" + replyMessageContent);  
  10.     }   
  11.     catch (UnsupportedEncodingException e)   
  12.     {  
  13.        e.printStackTrace();  
  14.     }  
  15.    
  16. logger.info("The following operation");  
 

我们再调用阶乘接口,生产者端控制台的输出日志如下图

从日志可以看出“The following operation”日志信息的输出是在获得返回消息之后执行的,假如当前方法的后续操作不依赖于返回消息的内容,但却要等待返回消息的获取完成后才能执行,这样的同步操作是不合适的,我们需要分离消息发送和返回消息的接收,使这两个操作在不同的线程中异步执行。

要做到这一点有两种方式。

第一种方式是使用RabbitTemplate.send方法发送消息,定义一个Message Listener侦听返回队列,获取并处理返回消息,这种方式在前面的代码中已经使用过,不再赘述。

第二种方式是使用spring-amqp 1.6引入的AsyncRabbitTemplate类,实现消息的异步发送,下面我们详细介绍这种方式的实现。

为了区分同步消息和异步消息的发送,我们新定义两个消息队列,一个名为“springAsyncMessageQueue”,用于接收AsyncRabbitTemplate发送的异步消息,一个名为“springAsyncReplyMessageQueue”,用于接收异步消息处理后的返回消息,同时把这两个消息队列分别绑定到 “springMessageExchange”和“springReplyMessageExchange”上,routingKey分别是“springAsyncMessage”和“springAsyncReplyMessage”。 消费者端的applicationContext.xml修改如下:

 

[html]  view plain  copy
 
  1. <rabbit:queue id="springAsyncMessageQueue" name="springAsyncMessageQueue" auto-delete="false"  
  2.                   durable="true" exclusive="false" auto-declare="true" declared-by="rabbitAdmin" />  
  3. <rabbit:direct-exchange id="springMessageExchange" name="springMessageExchange" durable="true"  
  4.                             auto-declare="false" auto-delete="false" declared-by="rabbitAdmin">  
  5.         <rabbit:bindings>  
  6.             <rabbit:binding queue="springMessageQueue" key="springMessage" />  
  7.             <rabbit:binding queue="springAsyncMessageQueue" key="springAsyncMessage" />  
  8.         rabbit:bindings>  
  9. rabbit:direct-exchange>  
  10.    
  11. <rabbit:queue id="springAsyncReplyMessageQueue" name="springAsyncReplyMessageQueue" auto-delete="false"  
  12.                   durable="true" exclusive="false" auto-declare="true" declared-by="rabbitAdmin" />  
  13.    
  14. <rabbit:direct-exchange id="springReplyMessageExchange" name="springReplyMessageExchange"  
  15.             durable="true" auto-delete="false" auto-declare="true" declared-by="rabbitAdmin">  
  16.         <rabbit:bindings>  
  17.             <rabbit:binding queue="springReplyMessageQueue" key="springReplyMessage" />  
  18.             <rabbit:binding queue="springAsyncReplyMessageQueue" key="springAsyncReplyMessage" />  
  19.         rabbit:bindings>  
  20. rabbit:direct-exchange>  

再添加一个处理异步消息的MessageListener类AsyncMessageListener,这个Listener类接收异步消息后,休眠10s后发送返回消息。

 

[java]  view plain  copy
 
  1. public class AsyncMessageListener implements ChannelAwareMessageListener {  
  2.    
  3.     @Autowired  
  4.     private MessagePropertiesConverter messagePropertiesConverter;  
  5.    
  6.     @Autowired  
  7.     private RabbitTemplate rabbitTemplate;  
  8.    
  9.     @Override  
  10.     public void onMessage(Message message, Channel channel) throws Exception  
  11.     {  
  12.         MessageProperties messageProperties = message.getMessageProperties();  
  13.         AMQP.BasicProperties rabbitMQProperties =  
  14.                 messagePropertiesConverter.fromMessageProperties(messageProperties, "UTF-8");  
  15.    
  16.          String messageContent = new String(message.getBody(), "UTF-8");  
  17.    
  18.          System.out.println("The received message is:" + messageContent);  
  19.    
  20.         String consumerTag = messageProperties.getConsumerTag();  
  21.    
  22.         String replyMessageContent = consumerTag + " have received the message '" + messageContent + "'";  
  23.    
  24.         Thread.sleep(10000);  
  25.    
  26.         AMQP.BasicProperties replyRabbitMQProps =  
  27.                 new AMQP.BasicProperties("text/plain",  
  28.                         "UTF-8",  
  29.                         null,  
  30.                         2,  
  31.                         0, rabbitMQProperties.getCorrelationId(), null, null,  
  32.                         null, null, null, null,  
  33.                         consumerTag, null);  
  34.         //创建返回消息的信封头  
  35.         Envelope replyEnvelope =  
  36.                 new Envelope(messageProperties.getDeliveryTag(), true,  
  37.                         "springReplyMessageExchange", "springAsyncReplyMessage");  
  38.    
  39.         MessageProperties replyMessageProperties =  
  40.                 messagePropertiesConverter.toMessageProperties(replyRabbitMQProps,  
  41.                         replyEnvelope,"UTF-8");  
  42.    
  43.         Message replyMessage = MessageBuilder.withBody(replyMessageContent.getBytes())  
  44.                 .andProperties(replyMessageProperties)  
  45.                 .build();  
  46.    
  47.         rabbitTemplate.send("springReplyMessageExchange","springAsyncReplyMessage", replyMessage);  
  48.         channel.basicAck(messageProperties.getDeliveryTag(), false);  
  49.     }  
  50. }  

在Message Listener Container中追加这个Listener
[html]  view plain  copy
 
  1. <bean id="asyncMessageListener"  
  2.           class="com.qf.rabbitmq.listener.AsyncMessageListener" />  
  3. <rabbit:listener-container message-converter="serializerMessageConverter"  
  4.                                ........................                                 
  5.                                recovery-back-off="backOff">  
  6.         <rabbit:listener ref="caculateListener" queues="springMessageQueue"/>  
  7.         <rabbit:listener ref="asyncMessageListener" queues="springAsyncMessageQueue" />  
  8. rabbit:listener-container>  

在生产者端,我们在RabitConfig中添加一个AsyncRabbitTemplate的bean定义
[java]  view plain  copy
 
  1. @Bean(name="asyncRabbitTemplate")  
  2. public AsyncRabbitTemplate getAsyncRabbitTemplate()  
  3. {  
  4.    AsyncRabbitTemplate asyncRabbitTemplate=  
  5.         new AsyncRabbitTemplate(getConnectionFactory(),  
  6.             AppConstants.SEND_EXCHANGE_NAME,  
  7.             AppConstants.SEND_ASYNC_MESSAGE_KEY,  
  8.             AppConstants.REPLY_ASYNC_QUEUE_NAME);  
  9.         asyncRabbitTemplate.setReceiveTimeout(60000);  
  10.         asyncRabbitTemplate.setAutoStartup(true);  
  11.         return rabbitTemplate;  
  12. }  

这里设置的ReceiveTimeout属性类似于RabbitTemplate的ReplyTimeout属性,是获取返回消息的超时时间。 AsyncRabbitTemplate构造函数中的四个参数分别是连接工厂,发送消息的Exchange名,发送异步消息的RoutingKey(“springAsyncMessage”,用于Exchange将异步消息转发到springAsyncMessageQueue队列),返回消息队列名(“springAsyncMessageQueue”,用于接收异步消息对应的返回消息)

在SendMessageService和SendMessageServiceImpl中添加一个发送异步消息的接口方法sendAsyncMessage

 

[java]  view plain  copy
 
  1. public interface SendMessageService  
  2. {  
  3.   ..........  
  4.   void sendAsyncMessage(String message);  
  5. }  
  6.    
  7. @Service("sendMessageService")  
  8. public class SendMessageServiceImpl implements SendMessageService  
  9. {  
  10.   @Autowired  
  11.   private ThreadPoolTaskExecutor executor;  
  12.    
  13.   ..........  
  14.    @Override  
  15.    public void sendAsyncMessage(String message)  
  16.    {  
  17.         CompletableFuture resultCompletableFuture =  
  18.                 CompletableFuture.runAsync(new AsyncMQSenderThread(message), executor);  
  19.    
  20.    }  
  21. }  
在SendMessageServiceImpl.sendAsyncMessage方法中,我们把发送异步消息的任务交给了一个后台线程异步执行,
使用了CompletableFuture的runAsync方法,没有返回值。
负责发送消息的后台任务类AsyncMQSenderThread主要代码如下:

 

 

[java]  view plain  copy
 
  1. public class AsyncMQSenderThread implements Runnable  
  2. {  
  3.   .................  
  4.   private String message;  
  5.   
  6.   
  7.   public AsyncMQSenderThread(String message)  
  8.   {  
  9.         this.message = message;  
  10.   }  
  11.     
  12.   public void run()  
  13.   {  
  14.      AsyncRabbitTemplate rabbitTemplate =  
  15.        (AsyncRabbitTemplate) ApplicationContextUtil.getBean("asyncRabbitTemplate");  
  16.      .........  
  17.      Message sendMessage = MessageBuilder.withBody(message.getBytes())  
  18.                 .andProperties(sendMessageProperties)  
  19.                 .build();  
  20.   
  21.   
  22.      logger.info("Send message to consumer");  
  23.        
  24.      AsyncRabbitTemplate.RabbitMessageFuture replyMessageFuture =  
  25.                 rabbitTemplate.sendAndReceive("springMessageExchange", "springAsyncMessage", sendMessage);  
  26.   
  27.   
  28.         replyMessageFuture.addCallback(new ListenableFutureCallback() {  
  29.             @Override  
  30.             public void onFailure(Throwable ex)  
  31.             {  
  32.   
  33.   
  34.             }  
  35.   
  36.   
  37.             @Override  
  38.             public void onSuccess(Message result)  
  39.             {  
  40.                 logger.info("Received the reply message");  
  41.                 try {  
  42.                     String replyMessageContent =  
  43.                             new String(result.getBody(), "UTF-8");  
  44.                     logger.info("The reply message is:" + replyMessageContent);  
  45.                 } catch (UnsupportedEncodingException e) {  
  46.                     e.printStackTrace();  
  47.                 }  
  48.             }  
  49.         });  
  50.         logger.info("The following operation");  
  51.     }     

 

在run方法中,AsyncRabbitTemplate.sendAndReceive方法返回的结果是一个RabbitMessageFuture对象,它实现了Google Guava提供的ListenableFuture接口,可以通过添加CallBack对返回消息进行处理。我们这里是添加了一个匿名的ListenableFutureCallback对象。

在SendMessageController中添加一个Rest接口sendAsyncMessage
[java]  view plain  copy
 
  1. @RestController  
  2. public class SendMessageController  
  3. {  
  4.  ............  
  5.  @RequestMapping(value = "/sendMessage", method = RequestMethod.POST)  
  6.  public void sendAsyncMessage(@RequestBody String message)  
  7.  {  
  8.     sendMessageService.sendAsyncMessage(message);  
  9.  }  
  10. }  
启动生产者和消费者应用,调用sendMessage接口,生产者端控制台的日志如下所示:


 

从日志可以看出,发送消息和处理返回消息是由独立的两个线程异步处理的,在发送完消息后,当前线程并没有阻塞,等待返回消息,而是直接执行后续的代码,对返回消息的处理是由RabbitMessageFuture对象绑定的Callback进行处理的。

AsyncRabbitTemplate方式和Message Listener方式都可以实现对返回消息的异步处理,它们的区别在于对返回消息中包含的CorrelationId与发送消息中包含的CorrelationId的比较,前一种方式是交给AsyncRabbitTemplate对象的内置方法处理, 而Message Listener方式需要开发者自行处理。

 

使用BatchingRabbitTemplate批量发送消息

 

使用RabbitTemplate和AsyncRabbitTemplate都是发送单条消息,每条消息发送完进行确认,但如果我们想发送20条消息,对这些消息进行一次性的确认,应该如何实现呢?我们可以使用spring-amqp 1.4.1引入的BatchingRabbitTemplate进行消息发送。

BatchingRabbitTemplate的构造函数形式如下:

 

[java]  view plain  copy
 
  1. public BatchingRabbitTemplate(BatchingStrategy batchingStrategy, TaskScheduler scheduler)   
可以看出BatchingRabbitTemplate有两个属性,一个是批量策略,定义了批量发送消息时消息的组合方式以及消息Properties的设定, 另一个是任务调度器,用于定义发送批量消息的定时任务。 我们这里使用的批量策略类是spring-ampq自带的SimpleBatchingStrategy类,它的定义大致如下:
[java]  view plain  copy
 
  1. public class SimpleBatchingStrategy implements BatchingStrategy   
  2. {  
  3.     private final int batchSize;  
  4.    
  5.     private final int bufferLimit;  
  6.    
  7.     private final long timeout;  
  8.    
  9.     public SimpleBatchingStrategy(int batchSize, int bufferLimit, long timeout)   
  10.     {  
  11.         this.batchSize = batchSize;  
  12.         this.bufferLimit = bufferLimit;  
  13.         this.timeout = timeout;  
  14.     }  
  15.     .......  
  16.     @Override  
  17.     public MessageBatch addToBatch(String exchange, String routingKey, Message message) {  
  18.     .........  
这个批量策略的思想是设置一个消息缓存区,以及批量消息数的上限,当待发送消息的条数达到上限(batchSize),或者待发送的消息的总大小超过缓存区上限(bufferLimit),即将消息拼装成一个大消息进行发送。 使用BatchingRabbitTemplate发送消息的流程如下图所示


 
1)调用端调用BatchingRabbitTemplate的send方法发送消息。

2)send方法调用BatchingRabbitTemplate对象设置的BatchingStrategy属性的addToBatch方法。

3)以SimpleBatchingStrategy.addToBatch方法为例,如果累计的未发送消息数达到消息数上限,或者消息总长度大于缓存区大小,
会将未发送消息队列里的消息拼装成一条完整消息,封装在一个MessageBatch中返回给BatchingRabbitTemplate对象。
 
[java]  view plain  copy
 
  1. @Override  
  2. public MessageBatch addToBatch(String exchange, String routingKey, Message message) {  
  3.    ............  
  4.    //消息队列里现存消息加待发送消息大小超过缓存区上限,先将消息队列里已有消息拼装为  
  5.    //一条消息返回。  
  6.    if (this.messages.size() > 0 && this.currentSize + bufferUse > this.bufferLimit)   
  7.    {  
  8.          //拼装消息队列里已有消息到MessageBatch  
  9.          batch = doReleaseBatch();  
  10.          //使用新消息的exchange,routingKey初始化exchange和routingKey  
  11.      this.exchange = exchange;  
  12.      this.routingKey = routingKey;  
  13.    }  
  14.    this.currentSize += bufferUse;  
  15.    this.messages.add(message);  
  16.    //如果添加了新消息的消息队列的消息数超过消息上限或者消息总大小超过缓存区上限,  
  17.    //将消息队列里所有消息拼装成一条消息,封装到MessageBatch对象中返回。  
  18.    if (batch == null && (this.messages.size() >= this.batchSize  
  19.     || this.currentSize >= this.bufferLimit))   
  20.    {  
  21.     batch = doReleaseBatch();  
  22.    }  
  23.    ...................  
  24. }  
  25.   
  26.   
  27. private MessageBatch doReleaseBatch()   
  28. {  
  29.    if (this.messages.size() < 1)   
  30.    {  
  31.     return null;  
  32.    }  
  33.    Message message = assembleMessage();  
  34.    MessageBatch messageBatch = new MessageBatch(this.exchange, this.routingKey, message);  
  35.    this.messages.clear();  
  36.    this.currentSize = 0;  
  37.    this.exchange = null;  
  38.    this.routingKey = null;  
  39.    return messageBatch;  
  40. }  

4)BatchingRabbitTemplate对象使用父类(RabbitTemplate)的send方法发送MessageBatch中包含的消息。
 
我们修改生产者程序,添加BatchingRabbitTemplate对象
[java]  view plain  copy
 
  1. @Bean(name="batchingStrategy")  
  2. public BatchingStrategy createBatchingStrategy()  
  3. {  
  4.    SimpleBatchingStrategy batchingStrategy =  
  5.         new SimpleBatchingStrategy(20,1000,60000);  
  6.    return batchingStrategy;  
  7. }  
  8.    
  9. @Bean(name="batchRabbitTemplate")  
  10. public BatchingRabbitTemplate createBatchRabbitTemplate()  
  11. {  
  12.     BatchingRabbitTemplate batchTemplate =  
  13.             new BatchingRabbitTemplate(createBatchingStrategy(),  new ConcurrentTaskScheduler());  
  14.     batchTemplate.setConnectionFactory(getConnectionFactory());  
  15.    
  16.     return batchTemplate;  
  17. }  
 

我们这里的BatchingStrategy设置的消息条数上限为20条,消息总大小上限为1000。使用的TaskScheduler是Java自带的 ConcurrentTaskScheduler对象,使用单线程池执行后续发送批量消息的计划任务,实际使用时可以设置为自定义的TaskScheduler.

为了测试方便,我们再添加两个消息队列,一个消息队列名为springBatchMessageQueue,用于接收发送的批量消息,一个消息队列名为springBatchReplyMessageQueue。在生产者和消费者端分别添加定义队列和消息绑定的代码,由于和异步消息类似,这里不再列出。 在消费者端定义一个BatchMessageListener,用于从springBatchMessageQueue接收和处理消息,它的主要代码如下:

 

[java]  view plain  copy
 
  1. public class BatchMessageListener implements ChannelAwareMessageListener  
  2. {  
  3.     private Logger logger = LoggerFactory.getLogger(BatchMessageListener.class);  
  4.    
  5.     public void onMessage(Message message, Channel channel) throws Exception  
  6.     {  
  7.       MessageProperties messageProperties = message.getMessageProperties();  
  8.       AMQP.BasicProperties rabbitMQProperties =  
  9.                 messagePropertiesConverter.fromMessageProperties(messageProperties, "UTF-8");  
  10.    
  11.       String messageContent = new String(message.getBody(), "UTF-8");  
  12.       String correlationId = rabbitMQProperties.getCorrelationId();  
  13.       //打印接收到的批量消息内容和correlationId,用于检验是否使用第一条消息的correaltionId.  
  14.       logger.info("The received batch message is:" + messageContent);  
  15.       logger.info("The correlation id is:" + correlationId);  
  16.       ......  
  17.    }         

在生产者端,我们在SendMessageService接口中定义发送批量消息的接口sendBatchMessages方法。

 

 

[java]  view plain  copy
 
  1. public interface SendMessageService  
  2. {  
  3.   ........  
  4.   void sendBatchMessage(List messages);  
  5. }  
  6.    
  7. public class SendMessageServiceImpl implements SendMessageService  
  8. {  
  9.  .........  
  10.  @Override  
  11.  public void sendBatchMessage(List messages)   
  12.  {  
  13.        CompletableFuture.runAsync(new BatchMQSenderThread(messages), executor);  
  14.  }  
我们定义了BatchMQSenderThread类来实现后台发送批量消息,它的主要代码如下:
[java]  view plain  copy
 
  1. public class BatchMQSenderThread  implements Runnable  
  2. {  
  3.     private Logger logger = LoggerFactory.getLogger(BatchMQSenderThread.class);  
  4.    
  5.     private List messageList;  
  6.    
  7.     public BatchMQSenderThread(List messageList)  
  8.     {  
  9.         this.messageList = messageList;  
  10.     }  
  11.    
  12.     public void run()  
  13.     {  
  14.         BatchingRabbitTemplate rabbitTemplate =  
  15.                 (BatchingRabbitTemplate) ApplicationContextUtil.getBean("batchRabbitTemplate");  
  16.         for(String message:messageList)  
  17.         {  
  18.             Date sendTime = new Date();  
  19.             String correlationId = UUID.randomUUID().toString();  
  20.    
  21.             AMQP.BasicProperties props =  
  22.                     new AMQP.BasicProperties("text/plain",  
  23.                             "UTF-8",  
  24.                             null,  
  25.                             2,  
  26.                             0, correlationId, AppConstants.REPLY_EXCHANGE_NAME, null,  
  27.                             null, sendTime, null, null,  
  28.                             "SpringProducer", null);  
  29.    
  30.             MessageProperties sendMessageProperties =  
  31.                     messagePropertiesConverter.toMessageProperties(props, null,"UTF-8");  
  32.             sendMessageProperties.setReceivedExchange(AppConstants.REPLY_EXCHANGE_NAME);  
  33.             sendMessageProperties.setReceivedRoutingKey(AppConstants.REPLY_BATCH_MESSAGE_KEY);  
  34.             sendMessageProperties.setRedelivered(true);  
  35.    
  36.             Message sendMessage = MessageBuilder.withBody(message.getBytes())  
  37.                     .andProperties(sendMessageProperties)  
  38.                     .build();  
  39.    
  40.             logger.info("Send message '" + message + "' to batch");  
  41.             logger.info("The message's correlation id is:" + correlationId);  
  42.    
  43.             rabbitTemplate.send(AppConstants.SEND_EXCHANGE_NAME, AppConstants.SEND_BATCH_MESSAGE_KEY, sendMessage, null);  
  44.         }  
  45. }  

 

BatchMQSenderThread.run方法循环读取消息List中的消息,调用batchingRabbitTemplate对象进行消息发送,每条消息我们都生成一个uuid作为消息的correlationId,我们接下来会看到,在最后发送的批量消息中,只有第一条消息的correlationId被采用。

最后我们定义一个发送批量消息的rest接口,批量消息使用逗号进行分割。

 

[java]  view plain  copy
 
  1. @RestController  
  2. public class SendMessageController  
  3. {  
  4.   ............  
  5.   @RequestMapping(value = "/sendMessages", method = RequestMethod.POST)  
  6.   public void sendMessages(@RequestBody String messages)  
  7.   {  
  8.      List messageList = Splitter.on(",").splitToList(messages);  
  9.      sendMessageService.sendBatchMessage(messageList);  
  10.   }    
修改完成后,我们启动生产者和消费者应用,通过生产者rest接口发送30条消息(消息正文从Message0到Message29)

 

从生产者端日志可以看出,SendMessageServiceImpl调用BatchRabbitTemplate.send方法,发送了30条消息,但是由于消息条数超过了我们预设的20条限制,实际只批量发送了20条消息(从消费者端的日志截图可以看到),消费者端仍然是把发送的批量消息作为20条独立的消息各自接收的。

由于我们定义的SimpleBatchingStrategy策略对象的超时时间是60000ms(1分钟),1分钟后,剩余的10条消息被BatchRabbitTemplate对象批量发出,被消费者接收,消费者日志显示了这一点。

 

消费者日志还显示,BatchRabbitTemplate对象发送的30条消息,它们都具有相同的correlationId,是第一条消息的correlationId,而不是创建这些消息时生成的uuid,这是由SimpleBatchingStrategy类的批量策略决定的。

下面我们修改一下生产者的SimpleBatchingStrategy属性设定,设定消息缓存上限为200,超时时间为120000ms(2分钟),再发送30条消息。 消费者端的输出日志如下:


 

我们可以看出,由于消息缓存上限的限制,BatchRabbitTemplate只批量发送了16条消息,后14条消息是在2分钟后延时发送。

在实际使用时,我们也可以根据业务需要,自定义BatchStrategy策略。

 

Rabbit Annotation的使用

在系列2里我们提到了使用spring-rabbit的rabbit前缀简化applicationContext.xml配置文件中的RabbitMQ beans配置,spring-amqp还提供了RabbitMQ常用对象对应的Annoation,这些注解如下图所示:
 

 

 

下面我们使用这些Annotation简化我们的生产者应用代码。

我们直接修改ReplyBatchMessageListener,在它的onMessage方法头部添加以下注解:

 

[java]  view plain  copy
 
  1. @RabbitListener(  
  2.         bindings = @QueueBinding(  
  3.         value = @Queue(value="springBatchReplyMessageQueue", durable = "true", exclusive = "false",autoDelete = "false"),  
  4.         exchange = @Exchange(value="springReplyMessageExchange", durable = "true"),  
  5.         key = "springBatchReplyMessage"),  
  6.         admin="rabbitAdmin")  
  7. public void onMessage(Message message)  
  8. {  
  9.      try   
  10.      {  
  11.           String messageContent = new String(message.getBody(), "UTF-8");  
  12.           logger.info("The reply message content is:" + messageContent);  
  13.      }   
  14.      catch (UnsupportedEncodingException e)   
  15.      {  
  16.           e.printStackTrace();  
  17.       }  
  18. }  

 

这个注解中使用到了@RabbitListener,将onMessage方法定义为一个Message Listener, @QueueBinding用于定义Binding对象,@Queue定义Queue对象,@Exchange定义Exchange对象,key属性是Bind对象的routingKey. spring-amqp内置的RabbitAdmin对象根据@Queue和@Exchange的属性,在RabbitMQ服务器上创建消息队列和Exchange, 根据@QueueBinding的属性,在RabbitMQ服务器上建立对应的Bind关系。 admin属性是创建Message Listener Container使用的RabbitAdmin对象,我们这里引用RabbitConfig中定义的“rabbitAdmin”对象。

@RabbitListener定义的Message Listener,所在的Message Listener Container是spring-amqp根据默认设置创建的Message Listener Container对象。如果我们想根据需要使用定制的Message Listener Container,我们需要在RabbitConfig类中定制SimpleMessageListenerContainerFactory对象:

 

[java]  view plain  copy
 
  1. @Bean  
  2. public SimpleRabbitListenerContainerFactory rabbitListenerContainerFactory()   
  3. {  
  4.         SimpleRabbitListenerContainerFactory factory = new SimpleRabbitListenerContainerFactory();  
  5.         factory.setConnectionFactory(getConnectionFactory());  
  6.         factory.setConcurrentConsumers(5);  
  7.         factory.setMaxConcurrentConsumers(5);  
  8.         factory.setConsumerTagStrategy(new ReplyConsumerTagStrategy());  
  9.         factory.setPrefetchCount(5);  
  10.         factory.setMessageConverter(getMessageConverter());  
  11.         factory.setAcknowledgeMode(AcknowledgeMode.AUTO);  
  12.         BackOff backOff = new FixedBackOff(60000,100);  
  13.         factory.setRecoveryBackOff(backOff);  
  14.         return factory;  
  15. }  

 

我们可以看出这个方法设置的factory对象属性,和我们之前定义的Message Listener Container bean对象的属性有相似之处, 不过factory对象的属性,并没有完全覆盖Message Listener Container对象的属性定义,例如messagePropertiesConverter属性的设置,在SimpleRabbitListenerContainerFactory类中就找不到,因此,如果想使用Message Listener Container比较复杂的属性,建议还是使用@Bean定义。

如果@RabbitListener关联的消息队列,Exchange和Bind关系在RabbitMQ服务器中已经创建,我们可以在@RabbitListener中不使用@QueueBinding,直接使用queues属性,此时onMessage方法头部的注解如下:

 

[java]  view plain  copy
 
  1. @RabbitListener(admin="rabbitAdmin", queues="springBatchReplyMessageQueue")  
  2. public void onMessage(Message message)  

为了使Rabbit注解生效,还需要添加@EnableRabbit注解,我们在RabbitConfig类头部添加这个注解

 

 

[java]  view plain  copy
 
  1. @Configuration  
  2. @EnableRabbit  
  3. @EnableConfigurationProperties(RabbitProperties.class)  
  4. public class RabbitConfig  
  5. {  
  6.   ........  

启动生产者和消费者应用,使用批量接口发送20条消息,从生产者日志我们可以看出我们定义在ReplyBatchMessageListener.onMessage方法上的添加的@RabbitListener将onMessage方法变成了一个Message Listener,接收了springBatchReplyMessageQueue队列的返回消息。

 

 

如果有很多方法想定义为一个Message Listener,而他们使用的@RabbitListener注解内容又相同,我们可以自定义RabbitListener注解,例如,我们可以将上面的@RabbitListener注解定义为ReplyRabbitListener注解接口

 

[java]  view plain  copy
 
  1. @Target({ElementType.TYPE, ElementType.METHOD, ElementType.ANNOTATION_TYPE})  
  2. @Retention(RetentionPolicy.RUNTIME)  
  3. @RabbitListener(admin="rabbitAdmin", queues="springBatchReplyMessageQueue")  
  4. public @interface ReplyRabbitListener {  
  5. }  
我们注释掉ReplyBatchMessageListener.onMessage方法头部的@RabbitListener注解,重新定义一个POJO类ReplyConsumerDelegate, 在这个类中添加一个消息处理方法processReplyMessage,在这个方法头部添加我们自定义的注解ReplyRabbitListener,把这个方法变成了一个Message Listener,在这个类头部我们还添加了@Component注解,使这个对象在SpringBoot启动时被初始化。

 

 

[java]  view plain  copy
 
  1. @Component  
  2. public class ReplyConsumerDelegate  
  3. {  
  4.     private Logger logger = LoggerFactory.getLogger(ReplyConsumerDelegate.class);  
  5.    
  6.      @ReplyRabbitListener  
  7.      public void processReplyMessage(String message)  
  8.      {  
  9.          logger.info("The reply message content is:" + message);  
  10.      }  
  11. }  

 

我们再启动生产者应用,发送批量消息,从生产者日志可以看出,@ReplyRabbitListener注解使处理返回消息的方法变成了ReplyConsumerDelegate.processReplyMessage

对象类型消息的发送

前面的例子中发送的消息消息本体都是String类型,如果我们想发送一个对象类型的消息时,应该怎么处理呢?下面的实例将演示如何发送对象消息。

我们定义两个实体类Company和Employee

 

[java]  view plain  copy
 
  1. public class Company implements java.io.Serializable  
  2. {  
  3.     private int id;  
  4.     private String companyName;  
  5.     private String establishDate;  
  6.    
  7.    
  8.     public int getId() {  
  9.         return id;  
  10.     }  
  11.    
  12.     public void setId(int id) {  
  13.         this.id = id;  
  14.     }  
  15.    
  16.     public String getCompanyName() {  
  17.         return companyName;  
  18.     }  
  19.    
  20.     public void setCompanyName(String companyName) {  
  21.         this.companyName = companyName;  
  22.     }  
  23.    
  24.     public String getEstablishDate() {  
  25.         return establishDate;  
  26.     }  
  27.    
  28.     public void setEstablishDate(String establishDate) {  
  29.         this.establishDate = establishDate;  
  30.     }  
  31.    
  32.     public String toString()  
  33.     {  
  34.         return "companyId=" + Integer.toString(getId()) +   
  35.         ",companyName=" + getCompanyName() + ",establishDate=" + getEstablishDate();  
  36.     }  
  37. }  
  38.    
  39. public class Employee implements java.io.Serializable  
  40. {  
  41.         private String employeeId;  
  42.         private String employeeName;  
  43.         private String employeeDate;  
  44.    
  45.     public String getEmployeeId() {  
  46.         return employeeId;  
  47.     }  
  48.    
  49.     public void setEmployeeId(String employeeId) {  
  50.         this.employeeId = employeeId;  
  51.     }  
  52.    
  53.     public String getEmployeeName() {  
  54.         return employeeName;  
  55.     }  
  56.    
  57.     public void setEmployeeName(String employeeName) {  
  58.         this.employeeName = employeeName;  
  59.     }  
  60.    
  61.     public String getEmployeeDate() {  
  62.         return employeeDate;  
  63.     }  
  64.    
  65.     public void setEmployeeDate(String employeeDate) {  
  66.         this.employeeDate = employeeDate;  
  67.     }  
  68.    
  69.     public String toString()  
  70.     {  
  71.         return "EmployeeId=" + getEmployeeId() + ",EmployeeName=" + getEmployeeName() + ",EmployeeDate=" + getEmployeeDate();  
  72.     }  
  73. }  

我们创建一个名为“springRabbitHandlerQueue”的消息队列,绑定到“springMessageExchange”Exchange对象上。

 

 

我们设定分别发送一条Company类型的消息,一条Employee类型的消息到springRabbitHandlerQueue消息队列里,消费者在同一个Message Listener中使用不同方法处理不同类型的消息。

我们在生产者程序中添加HandlerMQSenderThread类,用于发送对象类型的消息,它的主要代码如下:

[java]  view plain  copy
 
  1. public class HandlerMQSenderThread  implements Runnable  
  2. {  
  3.     private Object message;  
  4.    
  5.     public HandlerMQSenderThread(Object message)  
  6.     {  
  7.         this.message = message;  
  8.     }  
  9.    
  10.     @Override  
  11.     public void run()  
  12.     {  
  13.         ............  
  14.         RabbitTemplate rabbitTemplate =  
  15.                     (RabbitTemplate)ApplicationContextUtil.getBean("rabbitTemplate");  
  16.    
  17.         AMQP.BasicProperties props =  
  18.                     new AMQP.BasicProperties(MessageProperties.CONTENT_TYPE_SERIALIZED_OBJECT,  
  19.                             "UTF-8",  
  20.                             null,  
  21.                             2,  
  22.                             0, correlationId, AppConstants.REPLY_EXCHANGE_NAME, null,  
  23.                             null, sendTime, null, null,  
  24.                             "SpringProducer", null);  
  25.         Message sendMessage = MessageBuilder.withBody(toByteArray(message))  
  26.                     .andProperties(sendMessageProperties)  
  27.                     .build();  
  28.         ......  
  29.         rabbitTemplate.send(AppConstants.SEND_EXCHANGE_NAME, AppConstants.SEND_HANDLER_MESSAGE_KEY, sendMessage);              
  30.    }  
  31.    
  32.    public byte[] toByteArray (Object obj) {  
  33.         byte[] bytes = null;  
  34.         ByteArrayOutputStream bos = new ByteArrayOutputStream();  
  35.         try {  
  36.             ObjectOutputStream oos = new ObjectOutputStream(bos);  
  37.             oos.writeObject(obj);  
  38.             oos.flush();  
  39.             bytes = bos.toByteArray ();  
  40.             oos.close();  
  41.             bos.close();  
  42.         } catch (IOException ex) {  
  43.             ex.printStackTrace();  
  44.         }  
  45.         return bytes;  
  46.     }  

run方法里调用toByteArray方法,把对象消息转换成byte数组。在MessageProperties中,我们设定content-type为 “application/x-java-serialized-object”,表示发送的消息体类型为Java对象,便于消费者端进行反序列化。

在SendMessageService中我们定义一个接口方法sendObjectMessage,用于发送对象消息。

[java]  view plain  copy
 
  1. public interface SendMessageService  
  2. {  
  3.  ..........  
  4.  void sendObjectMessage(Object message);  
  5. }  
  6.    
  7. public class SendMessageServiceImpl implements SendMessageService  
  8. {  
  9.  .......  
  10.  @Override  
  11.  public void sendObjectMessage(Object message)   
  12.  {  
  13.     CompletableFuture.runAsync(new HandlerMQSenderThread(message), executor);  
  14.  }  
  15. }  

在SendMessageController类中添加两个Rest接口方法sendCompanyMessage和sendEmployeeMessage,这两个方法从客户端接收Json形式的报文,将其转换为Company对象和Employee对象。
[java]  view plain  copy
 
  1. @RequestMapping(value = "/sendCompanyMessage", method = RequestMethod.POST)  
  2. public void sendCompanyMessage(@RequestBody String message)  
  3. {  
  4.    JSONObject obj = (JSONObject)JSON.parse(message);  
  5.    Company company = JSONObject.toJavaObject(obj, Company.class);  
  6.    sendMessageService.sendObjectMessage(company);  
  7. }  
  8.    
  9. @RequestMapping(value = "/sendEmployeeMessage", method = RequestMethod.POST)  
  10. public void sendEmployeeMessage(@RequestBody String message)  
  11. {  
  12.     JSONObject obj = (JSONObject)JSON.parse(message);  
  13.     Employee employee = JSONObject.toJavaObject(obj, Employee.class);  
  14.     sendMessageService.sendObjectMessage(employee);  
  15. }  

消费者端,我们使用spring-boot-starter-amqp创建一个消费者应用,启动类和Config类的主要代码如下:

[java]  view plain  copy
 
  1. @SpringBootApplication  
  2. @ComponentScan("com.qf.rabbitmq")  
  3. @EnableAutoConfiguration(exclude = RabbitAutoConfiguration.class)  
  4. public class SpringConsumerTestApplication  
  5. {  
  6.     public static void main(String[] args)   
  7.     {  
  8.     SpringApplication.run(SpringConsumerTestApplication.class, args);  
  9.     }  
  10. }  
  11.    
  12. @Configuration  
  13. @EnableRabbit  
  14. @EnableConfigurationProperties(RabbitProperties.class)  
  15. //public class RabbitConfig implements RabbitListenerConfigurer  
  16. public class RabbitConfig  
  17. {  
  18.     @Autowired  
  19.     private RabbitProperties rabbitProperties;  
  20.    
  21.     /** 
  22.       * 创建RabbitMQ连接工厂 
  23.       * @return 根据application.properties设定参数建立的连接工厂对象 
  24.       */  
  25.     @Bean("connectionFactory")  
  26.     public ConnectionFactory getConnectionFactory() {  
  27.         com.rabbitmq.client.ConnectionFactory rabbitConnectionFactory =  
  28.                 new com.rabbitmq.client.ConnectionFactory();  
  29.         rabbitConnectionFactory.setHost(rabbitProperties.getHost());  
  30.         rabbitConnectionFactory.setPort(rabbitProperties.getPort());  
  31.         rabbitConnectionFactory.setUsername(rabbitProperties.getUsername());  
  32.         rabbitConnectionFactory.setPassword(rabbitProperties.getPassword());  
  33.         rabbitConnectionFactory.setVirtualHost(rabbitProperties.getVirtualHost());  
  34.         ConnectionFactory connectionFactory = new CachingConnectionFactory(rabbitConnectionFactory);  
  35.         return connectionFactory;  
  36.     }  
  37.    
  38.    
  39.     /** 
  40.       * 创建Message Listener Container工厂,它负责创建RabbitListener注解的 
  41.       * Message Listener所在的Container. 
  42.       * @return Message Listener Container工厂对象 
  43.       */  
  44.     @Bean  
  45.     public SimpleRabbitListenerContainerFactory rabbitListenerContainerFactory() {  
  46.         SimpleRabbitListenerContainerFactory factory =   
  47.                 new SimpleRabbitListenerContainerFactory();  
  48.         factory.setConnectionFactory(getConnectionFactory());  
  49.         factory.setConcurrentConsumers(5);  
  50.         factory.setMaxConcurrentConsumers(5);  
  51.         factory.setPrefetchCount(5);  
  52.         return factory;  
  53.     }  
  54. }  

创建一个POJO类RabbitHandlerConsumer,通过添加@RabbitListener注解和@RabbitHandler注解,使它成为springRabbitHandlerQueue 消息的Message Listener对象,并且根据不同类型的消息分别处理。
[java]  view plain  copy
 
  1. @Component  
  2. @RabbitListener(queues="springRabbitHandlerQueue")  
  3. public class RabbitHandlerConsumer  
  4. {  
  5.     private Logger logger = LoggerFactory.getLogger(RabbitHandlerConsumer.class);  
  6.    
  7.     @RabbitHandler  
  8.     public void processCompanyMessage(Company message)  
  9.     {  
  10.         logger.info("Received the message having Company type.");  
  11.         logger.info("The message content is:" + message.toString());  
  12.     }  
  13.    
  14.     @RabbitHandler  
  15.     public void processEmployeeMessage(Employee message)  
  16.     {  
  17.         logger.info("Received the message having Employee type.");  
  18.         logger.info("The message content is:" + message.toString());  
  19.     }  
  20. }  

RabbitHandler注解是spring-amqp 1.5开始引入的,它用于将POJO类方法转换为消息处理方法,RabbitListener注解则提升到类级别, 可以在POJO类头部添加,作为全局设定。需要指出的是RabbitHandler起作用必须是在RabbitMQ消息本体被正常转换为对象后,否则它将无法根据方法参数类型确定实际的handlerMethod,在实际使用时需要对MessageConverter做相关设定,具体做法请参考spring-amqp文档。

最后将Employee和Company实体类添加到消费者项目中。

启动生产者和消费者应用。

先通过sendCompanyMessage接口发送一个Company类型的消息。

 

消费者端日志显示这条消息被RabbitHandlerConsumer.processCompanyMessage方法接收并处理。

再发送一条Employee类型的消息
 
 
消费者端日志显示这条消息被RabbitHandlerConsumer.processEmployeeMessage方法接收并处理。


 

我们上面的消费者应用是基于Spring Boot环境的,如果我们想在Spring AMQP框架接收并处理对象消息,我们不需要添加@EnableRabbit和@Configuration注解,只需要在applicationContext.xml文件中添加标签即可。

我们新建一个基于Maven的消费者项目,pom.xml添加的库为:

 

[html]  view plain  copy
 
  1. <span style="font-size:10px;"><dependencies>  
  2.     <dependency>  
  3.       <groupId>junitgroupId>  
  4.       <artifactId>junitartifactId>  
  5.       <version>3.8.1version>  
  6.       <scope>testscope>  
  7.     dependency>  
  8.     <dependency>  
  9.       <groupId>org.springframework.amqpgroupId>  
  10.       <artifactId>spring-amqpartifactId>  
  11.       <version>1.6.5.RELEASEversion>  
  12.     dependency>  
  13.     <dependency>  
  14.       <groupId>org.springframework.amqpgroupId>  
  15.       <artifactId>spring-rabbitartifactId>  
  16.       <version>1.6.5.RELEASEversion>  
  17.     dependency>  
  18.     <dependency>  
  19.       <groupId>org.slf4jgroupId>  
  20.       <artifactId>jcl-over-slf4jartifactId>  
  21.       <version>1.7.21version>  
  22.     dependency>  
  23.     <dependency>  
  24.       <groupId>org.slf4jgroupId>  
  25.       <artifactId>slf4j-log4j12artifactId>  
  26.       <version>1.7.21version>  
  27.     dependency>  
  28.     <dependency>  
  29.       <groupId>org.slf4jgroupId>  
  30.       <artifactId>slf4j-apiartifactId>  
  31.       <version>1.7.21version>  
  32.     dependency>  
  33. dependencies>span>  

 

在src/main/resources目录下添加applicationContext.xml文件,它的主要内容如下:

 

[html]  view plain  copy
 
  1. <span style="font-size:10px;"><beans xmlns="http://www.springframework.org/schema/beans"  
  2.     xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"  
  3.     xmlns:context="http://www.springframework.org/schema/context"  
  4.     xmlns:rabbit="http://www.springframework.org/schema/rabbit"  
  5.     xsi:schemaLocation="  
  6.     http://www.springframework.org/schema/beans  
  7.     http://www.springframework.org/schema/beans/spring-beans-4.0.xsd  
  8.     http://www.springframework.org/schema/rabbit  
  9.     http://www.springframework.org/schema/rabbit/spring-rabbit-1.6.xsd  
  10.     http://www.springframework.org/schema/context  
  11.     http://www.springframework.org/schema/context/spring-context-4.0.xsd" >  
  12.    
  13.     <context:annotation-config/>  
  14.    
  15.     <context:property-placeholder  
  16.         ignore-unresolvable="true" location="classpath*:/application.properties" />      
  17.    
  18.     <context:component-scan base-package="com.qf.rabbitmq"  />    
  19.    
  20.     <rabbit:annotation-driven container-factory="rabbitListenerContainerFactory" />  
  21.    
  22.     <bean id="rabbitMQConnectionFactory" class="com.rabbitmq.client.ConnectionFactory">  
  23.         <property name="username" value="${mq.userName}" />  
  24.         <property name="password" value="${mq.password}" />  
  25.         <property name="host" value="${mq.ip}" />  
  26.         <property name="port" value="${mq.port}" />  
  27.         <property name="virtualHost" value="${mq.virutalHost}" />  
  28.         <property name="automaticRecoveryEnabled" value="false" />  
  29.         <property name="topologyRecoveryEnabled" value="false" />  
  30.         <property name="networkRecoveryInterval" value="60000" />  
  31.     bean>  
  32.    
  33.     <rabbit:connection-factory id ="connectionFactory"  
  34.         connection-factory="rabbitMQConnectionFactory"  
  35.         connection-timeout="10000"  
  36.         cache-mode="CHANNEL"  
  37.         channel-cache-size="20"/>  
  38.    
  39.     <bean id="rabbitListenerContainerFactory"  
  40.         class="org.springframework.amqp.rabbit.config.SimpleRabbitListenerContainerFactory">  
  41.         <property name="connectionFactory" ref="connectionFactory"/>  
  42.         <property name="concurrentConsumers" value="3"/>  
  43.         <property name="maxConcurrentConsumers" value="3"/>  
  44.     bean>  
  45.    
  46.     <rabbit:admin id="rabbitAdmin"   
  47.         connection-factory="connectionFactory" auto-startup="true"/>  
  48. beans>span>  

 

再添加RabbitHandlerConsumer,Employee,Company类。

最后在主程序中加载applicationContext.xml文件,启动消费者应用

[java]  view plain  copy
 
  1. "font-size:10px;">public class RabbitCustomerTest   
  2. {  
  3.     public static void main( String[] args )  
  4.     {  
  5.         ClassPathXmlApplicationContext context = new ClassPathXmlApplicationContext(  
  6.                 new String[] { "classpath:applicationContext.xml" });  
  7.         context.start();  
  8.     }  
  9. }  

启动生产者应用,发送Company消息和Employee消息,我们可以看到消费者应用的响应与基于SpringBoot的消费者应用程序相同。

你可能感兴趣的:(使用spring-boot-starter-amqp开发生产者应用)