07.RabbitMQ处理幂等性

在上篇中主要讲到了RabbitMQ的重试机制和配置，因为重试某些情况需要处理幂等性问题

解决幂等性，一般采用以下几种方式，

1. 使用全局消息ID（MQ自带，RabbitMQ没有，要自己实现）
2. 根据业务ID判断

代码中处理方式：

@RestController
@RequestMapping("/p2p")
public class P2PUserCtrl {

    @Autowired
    private AmqpTemplate amqpTemplate;
    
    @GetMapping("/addUserPower")
    public String addUserPower(){
        String queueName = "USER";
        Message message = MessageBuilder.withBody("terry".getBytes())
                .setContentEncoding("utf-8")
                .setMessageId(UUID.randomUUID().toString())
                .build();
        amqpTemplate.convertAndSend(queueName, message);
        return "terry";
    }
}

生产者处理方式：

import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import org.springframework.stereotype.Component;

import java.io.UnsupportedEncodingException;
import java.util.ArrayList;
import java.util.List;

@Component
public class P2PUserConsumer {

    // 定义变量存储messageId，实际项目使用redis，这种方式会引发内存泄漏
    private List<String> list = new ArrayList<>();

    /**
     * RabbitMQ解决幂等，此处需要将RabbitListener写在方法上面，否则报错
     * @param msg
     */
    @RabbitListener(queues = "USER")
    public void receivePower(Message msg) {
        String messageId = msg.getMessageProperties().getMessageId();
        if (list.contains(messageId)) {
            System.out.println("messageId重复");
        } else {
            list.add(messageId);
            // 正常消费逻辑
            try {
                String str = new String(msg.getBody(),"utf-8");
                System.out.println("消费者消费：" + str);
            } catch (UnsupportedEncodingException e) {
                e.printStackTrace();
            }
        }
        // 抛出一个运行时异常，模仿消费异常
        throw new RuntimeException();
    }
}