KafkaMessageListenerContainer.getAssignedPartitions是否线程安全？

这个方法被设为公开方法并非bug，而是有其设计场景和考量，但跨任意线程调用确实存在线程安全风险，以下是具体说明：

• 设计场景限定
  Spring Kafka提供getAssignedPartitions()，主要是供容器自身线程上下文内使用——比如在ConsumerAwareMessageListener的消费回调里，或者容器生命周期钩子（如start()/stop()相关逻辑）中调用。这些场景下，调用线程与修改assignedPartitions的ListenerConsumerRebalanceListener线程不会并发操作：Rebalance操作触发时，ListenerConsumer的主线程会暂停消费逻辑，此时不会有其他线程读取该字段，自然不存在并发问题。

• 跨线程调用的风险
  当你在外部任意线程中定期调用该方法时，确实会触发线程安全问题：assignedPartitions是普通LinkedHashSet，没有同步保护，ListenerConsumerRebalanceListener修改集合的同时，外部线程可能正在读取，轻则拿到不一致的分区数据，重则抛出ConcurrentModificationException。这属于超出方法设计预期场景的未定义行为，而非方法实现本身的bug。

• 安全获取分区信息的正确方式
  如果需要在外部线程获取分区状态，建议通过Spring Kafka的容器事件机制实现：注册ApplicationListener监听ListenerContainerPartitionAssignmentsEvent事件，在事件回调中获取最新分区信息并缓存到线程安全容器（如CopyOnWriteArraySet、ConcurrentHashMap），外部线程直接读取缓存副本即可。示例代码：

  @EventListener
  public void onPartitionAssigned(ListenerContainerPartitionAssignmentsEvent event) {
      Set<TopicPartition> latestPartitions = event.getAssignedPartitions();
      // 将分区信息存入线程安全的缓存
  }
  

• 为何不使用线程安全集合
  这是性能权衡的结果：在ListenerConsumer的常规运行流程中，assignedPartitions的读写都在同一线程内完成，使用普通集合能避免线程安全集合带来的额外性能开销。仅当跨线程调用时才会出现问题，而这种场景并非该方法的核心设计目标。

内容的提问来源于stack exchange，提问作者Tim Barabanov