packagehuxihx;importorg.apache.kafka.clients.admin.AdminClient;importorg.apache.kafka.clients.admin.AdminClientConfig;importorg.apache.kafka.clients.admin.DescribeLogDirsResult;importorg.apache.kafka.common.TopicPartition;importorg.apache.kafka.common.requests.DescribeLogDirsResponse;importjava.util.Collections;importjava.util.Map;importjava.util.Properties;importjava.util.concurrent.ExecutionException;public classTopicDiskSizeSummary {private staticAdminClient admin;public static void main(String[] args) throwsExecutionException, InterruptedException {
String brokers= "localhost:9092";
initialize(brokers);try{long topic1InBroker1 = getTopicDiskSizeForSomeBroker("t2", 1);long topic2InBroker0 = getTopicDiskSizeForSomeBroker("t1", 0);
System.out.println(topic1InBroker1);
System.out.println(topic2InBroker0);
}finally{
shutdown();
}
}public static long getTopicDiskSizeForSomeBroker(String topic, intbrokerID)throwsExecutionException, InterruptedException {long sum = 0;
DescribeLogDirsResult ret=admin.describeLogDirs(Collections.singletonList(brokerID));
Map> tmp =ret.all().get();for (Map.Entry>entry : tmp.entrySet()) {
Map tmp1 =entry.getValue();for (Map.Entryentry1 : tmp1.entrySet()) {
DescribeLogDirsResponse.LogDirInfo info=entry1.getValue();
Map replicaInfoMap =info.replicaInfos;for (Map.Entryreplicas : replicaInfoMap.entrySet()) {if(topic.equals(replicas.getKey().topic())) {
sum+=replicas.getValue().size;
}
}
}
}returnsum;
}private static voidinitialize(String bootstrapServers) {
Properties props= newProperties();
props.put(AdminClientConfig.BOOTSTRAP_SERVERS_CONFIG, bootstrapServers);
admin=AdminClient.create(props);
}private static voidshutdown() {if (admin != null) {
admin.close();
}
}
}