《RabbitMQ》如何保证消息不被重复消费

一 重复消息

为何会出现消息重复?消息重复的缘由有两个:1.生产时消息重复,2.消费时消息重复。java

1.1 生产时消息重复

因为生产者发送消息给MQ,在MQ确认的时候出现了网络波动,生产者没有收到确认,实际上MQ已经接收到了消息。这时候生产者就会从新发送一遍这条消息。程序员

生产者中若是消息未被确认,或确认失败,咱们可使用定时任务+(redis/db)来进行消息重试。web

@Component
@Slf4J
public class SendMessage {
    @Autowired
    private MessageService messageService;

    @Autowired
    private RabbitTemplate rabbitTemplate;

    // 最大投递次数
    private static final int MAX_TRY_COUNT = 3;

    /**
     * 每30s拉取投递失败的消息, 从新投递
     */

    @Scheduled(cron = "0/30 * * * * ?")
    public void resend() {
        log.info("开始执行定时任务(从新投递消息)");

        List<MsgLog> msgLogs = messageService.selectTimeoutMsg();
        msgLogs.forEach(msgLog -> {
            String msgId = msgLog.getMsgId();
            if (msgLog.getTryCount() >= MAX_TRY_COUNT) {
                messageService.updateStatus(msgId, Constant.MsgLogStatus.DELIVER_FAIL);
                log.info("超过最大重试次数, 消息投递失败, msgId: {}", msgId);
            } else {
                messageService.updateTryCount(msgId, msgLog.getNextTryTime());// 投递次数+1

                CorrelationData correlationData = new CorrelationData(msgId);
                rabbitTemplate.convertAndSend(msgLog.getExchange(), msgLog.getRoutingKey(), MessageHelper.objToMsg(msgLog.getMsg()), correlationData);// 从新投递

                log.info("第 " + (msgLog.getTryCount() + 1) + " 次从新投递消息");
            }
        });

        log.info("定时任务执行结束(从新投递消息)");
    }
}

1.2消费时消息重复

消费者消费成功后,再给MQ确认的时候出现了网络波动,MQ没有接收到确认,为了保证消息被消费,MQ就会继续给消费者投递以前的消息。这时候消费者就接收到了两条同样的消息。redis

修改消费者,模拟异常spring

@RabbitListener(queuesToDeclare = @Queue(value = "javatrip", durable = "true"))
public void receive(String message, @Headers Map<String,Object> headers, Channel channel) throws Exception{

    System.out.println("重试"+System.currentTimeMillis());
    System.out.println(message);
    int i = 1 / 0;
}

配置yml重试策略数据库

spring:
  rabbitmq:
    listener:
      simple:
        retry:
          enabled: true # 开启消费者进行重试
          max-attempts: 5 # 最大重试次数
          initial-interval: 3000 # 重试时间间隔

因为重复消息是因为网络缘由形成的,所以不可避免重复消息。可是咱们须要保证消息的幂等性json

二 如何保证消息幂等性

让每一个消息携带一个全局的惟一ID,便可保证消息的幂等性,具体消费过程为:微信

  1. 消费者获取到消息后先根据id去查询redis/db是否存在该消息。网络

  2. 若是不存在,则正常消费,消费完毕后写入redis/db。app

  3. 若是存在,则证实消息被消费过,直接丢弃。

生产者

@PostMapping("/send")
public void sendMessage(){

    JSONObject jsonObject = new JSONObject();
    jsonObject.put("message","Java旅途");
    String json = jsonObject.toJSONString();
    Message message = MessageBuilder.withBody(json.getBytes()).setContentType(MessageProperties.CONTENT_TYPE_JSON).setContentEncoding("UTF-8").setMessageId(UUID.randomUUID()+"").build();
    amqpTemplate.convertAndSend("javatrip",message);
}

消费者

@Component
@RabbitListener(queuesToDeclare = @Queue(value = "javatrip", durable = "true"))
public class Consumer {

    @RabbitHandler
    public void receiveMessage(Message message) throws Exception {

        Jedis jedis = new Jedis("localhost"6379);

        String messageId = message.getMessageProperties().getMessageId();
        String msg = new String(message.getBody(),"UTF-8");
        System.out.println("接收到的消息为:"+msg+"==消息id为:"+messageId);

        String messageIdRedis = jedis.get("messageId");

        if(messageId == messageIdRedis){
            return;
        }
        JSONObject jsonObject = JSONObject.parseObject(msg);
        String email = jsonObject.getString("message");
        jedis.set("messageId",messageId);
    }
}

若是须要存入db的话,能够直接将这个ID设为消息的主键,下次若是获取到重复消息进行消费时,因为数据库主键的惟一性,则会直接抛出异常。

< END >

往期精选
《RabbitMQ》如何保证消息的可靠性
  一文搞懂TCP和UDP的区别
  程序员接私活的19个平台
  Spring AOP实现原理
  Spring IOC实现原理
  Nginx超简单教程

本文分享自微信公众号 - Java旅途(Javatrip)。
若有侵权,请联系 support@oschina.cn 删除。
本文参与“OSC源创计划”,欢迎正在阅读的你也加入,一块儿分享。

相关文章
相关标签/搜索