kafka java获取topic_Java API获取topic所占磁盘空间(Kafka 1.0.0)

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();

}

}

}