RabbitMQ踩坑记

Posted 2021-01-22 little-sheep

 

之前我们给我们的系统加了一个使用SpringAOP+RabbitMQ+WebSocket进行实时消息通知功能（https://www.cnblogs.com/little-sheep/p/9934887.html）。在测试环境下没有问题，但上到生产环境后部分用户反映出现了丢消息的情况，针对这个问题我们进行了排查，发现，原本我们的系统是单机的，但用户在之前做了调整，在内外网服务器分别部署了系统，两个服务器都公用一个RabbitMQ。那么问题就来了。

之前生产者向Exchange生产消息，消费者从queue消费消息使用的是direct模式，通过routing-key保证生产的消息只有指定的消费者可消费。但当两台服务器共用一个MQ时即有了两个消费者连接同一个queue，此时rabbitmq并不是一个消息两个消费者都能消费，而是采用默认的轮询发送方式，A服务器收到消息1、3、5 。。。而B服务器收到2、4、6 。。。

这就出现了用户感觉的丢消息现象。所以我们考虑将通知改为广播形式即fanout。

       RabbitMQ将消息中间件的实现分成了Exchange+Queue的形式,Exchange和Queue使用Binding；生产者向Exchange生产消息，消息根据指定的binding进入Queue，消费者从Queue取消息。Fanout模式如图：

一个消费者会对应一个queue，那么多个消费者要有多个queue。

修改后代码如下：

RabbitMQConfig.java

//声明exchange 
Connection connection = factory.createConnection();
Channel channel = connection.createChannel(false);
//生产环境中有多个server，每个server都是一个消费者，对同一个消息都要进行处理。选用广播模式
channel.exchangeDeclare("exchange.websocket.msg", BuiltinExchangeType.FANOUT);

RabbitMessageQueue.java

@Override
public void send(WebSocketMsgEntity entity) {
    logger.warn("::product msg to MQ-websocket_msg_queue!");
    //若没有指定exchange，则使用默认名为“”的exchange，binding名与queue名相同
    rabbitTemplate.convertAndSend("exchange.websocket.msg","", entity);
}

RabbitMQListener.java

@Component
public class RabbitMQListener {
    private static Logger logger = LoggerFactory.getLogger(RabbitMQListener.class);
    @Autowired
    private RabbitMQService mqService;
    /**
     * WebSocket推送监听器
     * @param socketEntity
     * @param deliveryTag
     * @param channel
     */
    @RabbitListener(bindings ={@QueueBinding(value = @Queue(exclusive = "true"), exchange = @Exchange(value = "exchange.websocket.msg", type = ExchangeTypes.FANOUT))})
    public void webSocketMsgListener(@Payload WebSocketMsgEntity socketMsgEntity, @Header(AmqpHeaders.DELIVERY_TAG) long deliveryTag, Channel channel) throws IOException {
        logger.warn("::consume msg from MQ-websocket_msg_queue!");
        mqService.handleWebSocketMsg(socketMsgEntity, deliveryTag, channel);
    }

}

原本的客户端监听是监听Queue，而现在改成监听Binding。并不显式的指定一个Queue，而是将Queue设置成exclusive = true，这样每个消费者在监听Binding时都会默认创建一个Queue与指定的Exchange绑定，在消费者断开连接后Queue自动删除。若有两个消费者，则创建两个Queue，他们绑定的Exchange相同，当生产者有消息时会向两个Queue各插入一条，那么两个系统的用户就都能收到通知啦！