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

时间:2024-04-05 13:38:14

在高并发场景下,如大促销、秒杀等,我们可以采用 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);
    }
}