使用 RabbitMQ Binder 进行分区

RabbitMQ 本机不支持分区。

有时,将数据发送到特定分区是有利的——例如,当您希望严格按顺序处理消息时,特定客户的所有消息都应进入同一分区。

RabbitMessageChannelBinder 通过将每个分区的队列绑定到目标交换来提供分区。

以下 Java 和 YAML 示例展示了如何配置生产者

生产者
@SpringBootApplication
public class RabbitPartitionProducerApplication {

    private static final Random RANDOM = new Random(System.currentTimeMillis());

    private static final String[] data = new String[] {
            "abc1", "def1", "qux1",
            "abc2", "def2", "qux2",
            "abc3", "def3", "qux3",
            "abc4", "def4", "qux4",
            };

    public static void main(String[] args) {
        new SpringApplicationBuilder(RabbitPartitionProducerApplication.class)
            .web(false)
            .run(args);
    }

    @Bean
    public Supplier<Message<?>> generate() {
        return () -> {
            String value = data[RANDOM.nextInt(data.length)];
            System.out.println("Sending: " + value);
            return MessageBuilder.withPayload(value)
                    .setHeader("partitionKey", value)
                    .build();
        };
    }

}
application.yml
    spring:
      cloud:
        stream:
          bindings:
            generate-out-0:
              destination: partitioned.destination
              producer:
                partitioned: true
                partition-key-expression: headers['partitionKey']
                partition-count: 2
                required-groups:
                - myGroup

前一个示例中的配置使用默认分区(key.hashCode() % partitionCount)。这可能会或可能不会提供适当的平衡算法,具体取决于键值。您可以使用 partitionSelectorExpressionpartitionSelectorClass 属性覆盖此默认值。

只有在生产者部署时需要配置消费者队列时,才需要 required-groups 属性。否则,发送到分区的任何消息都会丢失,直到部署相应的消费者。

以下配置配置了一个主题交换

part exchange

以下队列绑定到该交换

part queues

以下绑定将队列与交换关联

part bindings

以下 Java 和 YAML 示例延续了之前的示例,并展示了如何配置使用者

使用者
@SpringBootApplication
public class RabbitPartitionConsumerApplication {

    public static void main(String[] args) {
        new SpringApplicationBuilder(RabbitPartitionConsumerApplication.class)
            .web(false)
            .run(args);
    }

    @Bean
    public Consumer<Message<String>> listen() {
        return message -> {
            String queue =- message.getHeaders().get(AmqpHeaders.CONSUMER_QUEUE);
            System.out.println(in + " received from queue " + queue);
        };
    }

}
application.yml
    spring:
      cloud:
        stream:
          bindings:
            listen-in-0:
              destination: partitioned.destination
              group: myGroup
              consumer:
                partitioned: true
                instance-index: 0
RabbitMessageChannelBinder 不支持动态扩展。每个分区必须至少有一个使用者。使用者的 instanceIndex 用于指示使用哪个分区。Cloud Foundry 等平台只能有一个 instanceIndex 的实例。