并发与限流实战：如何利用 RabbitMQ 在 SpringBoot 应用中实现并发控制与流量限制

成都怡乐轩科技2024-04-05 10:56

在高并发场景下，如大促销、秒杀等，我们可以采用 RabbitMQ 配合 SpringBoot 来实现并发控制与流量限制。你可以将 RabbitMQ 作为一个缓冲区，暂存大量并发请求，然后消费者可以根据自身处理能力去处理这些请求。下面就以一个高并发订单处理的案例实现来说明。

Step 1：配置 RabbitMQ

首先，我们需要在 SpringBoot 的配置文件中，设定 RabbitMQ 的配置，如以下的例子：

yaml

复制代码

spring:
  rabbitmq:
    host: localhost
    port: 5672
    username: guest
    password: guest
    virtual-host: /

Step 2：定义 Message Model

我们需要定义一个明确的消息模型，假设我们有一个订单 Model ，含有订单号 orderId：

java

复制代码

public class Order implements Serializable {
    private String orderId;
    // getters and setters
}

Step 3：创建 Producer

我们实现一个订单生产者，这个生产者会将订单发送到 RabbitMQ 的队列中：

java

复制代码

@Service
public class OrderProducer {
    private static final String EXCHANGE   = "exchange";
    private static final String ROUTING_KEY = "routingKey";

    @Autowired
    private RabbitTemplate rabbitTemplate;

    public void send(Order order){
        rabbitTemplate.convertAndSend(EXCHANGE, ROUTING_KEY, order);
    }
}

Step 4：创建 Consumer

然后我们实现一个订单消费者，它会从 RabbitMQ 接收订单并处理：

java

复制代码

@Service
public class OrderConsumer {
    @RabbitListener(queues = "queue")
    public void receive(Order order) {
        // 处理订单操作
    }
}

在高并发的环境中，订单请求会快速发送到 RabbitMQ，然后消费者可以按照自身处理能力慢慢的从队列中取出订单并处理，这样就实现了海量订单的并发处理和流量限制。

注意: 限流需要设置一些参数，如预取数量和确认模式。如下配置会使得每个消费者在确认之前都只会接收10个消息。同时我们也开启了手动确认机式，保证只有在消息被正确处理后才会从队列中删除：

yaml

复制代码

spring:
  rabbitmq:
    listener:
      simple:
        prefetch: 10
        acknowledge-mode: manual

在消费者端，我们需要手动确认收到这个消息并且已经处理完毕：

java

复制代码

@Service
public class OrderConsumer {
    @RabbitListener(queues = "queue")
    public void receive(Order order, Channel channel, @Headers Map<String,Object> headers) {
        // 处理订单操作
        ...
        // 最后确认消息已被成功消费
        channel.basicAck((Long)headers.get(AmqpHeaders.DELIVERY_TAG),false);
    }
}