Kafka获取订阅某topic的所有consumer group【客户端版】

之前写过如何用服务器端的API代码来获取订阅某topic的所有consumer group，参见这里。使用服务器端的API需要用到kafka.admin.AdminClient类，但是这个类在0.11.0.0版本已经被标记为不推荐使用了，故目前最合适的方式还是通过客户端API：org.apache.kafka.clients.admin.AdminClient。今天碰到有人问这个问题，我就尝试写了一个。使用之前你需要引入kafka client包依赖（以2.2.0版本为例）

Maven：

<dependency>
<groupId>org.apache.kafka</groupId>
<artifactId>kafka-clients</artifactId>
<version>2.2.0</version>
</dependency>

Gradle：

compile group: 'org.apache.kafka', name: 'kafka-clients', version: '2.2.0'

下面是代码：

 private static List<String> getGroupsForTopic(String brokerServers, String topic)
             throws ExecutionException, InterruptedException, TimeoutException {
         Properties props = new Properties();
         props.put(AdminClientConfig.BOOTSTRAP_SERVERS_CONFIG, brokerServers);

         try (AdminClient client = AdminClient.create(props)) {
             List<String> allGroups = client.listConsumerGroups()
                     .valid()
                     .get(10, TimeUnit.SECONDS)
                     .stream()
                     .map(ConsumerGroupListing::groupId)
                     .collect(Collectors.toList());

             Map<String, ConsumerGroupDescription> allGroupDetails =
                     client.describeConsumerGroups(allGroups).all().get(10, TimeUnit.SECONDS);

             final List<String> filteredGroups = new ArrayList<>();
             allGroupDetails.entrySet().forEach(entry -> {
                 String groupId = entry.getKey();
                 ConsumerGroupDescription description = entry.getValue();
                 boolean topicSubscribed = description.members().stream().map(MemberDescription::assignment)
                         .map(MemberAssignment::topicPartitions)
                         .map(tps -> tps.stream().map(TopicPartition::topic).collect(Collectors.toSet()))
                         .anyMatch(tps -> tps.contains(topic));
                 if (topicSubscribed)
                     filteredGroups.add(groupId);
             });
             return filteredGroups;
         }
     }

我会假设你的集群中没有配置安全认证和授权机制或者发起此AdminClient的用户是合法用户且有CLUSTER以及GROUP的DESCRIBE权限。

另外值得注意的是，上面这个函数无法获取非运行中的consumer group，即虽然一个group订阅了某topic，但是若它所有的consumer成员都关闭的话这个函数是不会返回该group的。