RabbitMQ踩坑记

 

之前我们给我们的系统加了一个使用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各插入一条,那么两个系统的用户就都能收到通知啦!