show topic partition info

src/main/java/com/xuxd/kafka/console/beans/vo/TopicPartitionVO.java

@@ -0,0 +1,34 @@
+package com.xuxd.kafka.console.beans.vo;
+
+import java.util.List;
+import java.util.stream.Collectors;
+import lombok.Data;
+import org.apache.kafka.common.Node;
+import org.apache.kafka.common.TopicPartitionInfo;
+
+/**
+ * kafka-console-ui.
+ *
+ * @author xuxd
+ * @date 2021-09-23 15:15:29
+ **/
+@Data
+public class TopicPartitionVO {
+
+    private int partition;
+
+    private String leader;
+
+    private List<String> replicas;
+
+    private List<String> isr;
+
+    public static TopicPartitionVO from(TopicPartitionInfo partitionInfo) {
+        TopicPartitionVO partitionVO = new TopicPartitionVO();
+        partitionVO.setPartition(partitionInfo.partition());
+        partitionVO.setLeader(partitionInfo.leader().toString());
+        partitionVO.setReplicas(partitionInfo.replicas().stream().map(Node::toString).collect(Collectors.toList()));
+        partitionVO.setIsr(partitionInfo.isr().stream().map(Node::toString).collect(Collectors.toList()));
+        return partitionVO;
+    }
+}

src/main/java/com/xuxd/kafka/console/controller/TopicController.java

@@ -31,4 +31,9 @@ public class TopicController {
     public Object deleteTopic(@RequestParam String topic) {
         return topicService.deleteTopic(topic);
     }
+
+    @GetMapping("/partition")
+    public Object getTopicPartitionInfo(@RequestParam String topic) {
+        return topicService.getTopicPartitionInfo(topic);
+    }
 }

src/main/java/com/xuxd/kafka/console/service/TopicService.java

@@ -16,4 +16,6 @@ public interface TopicService {
     ResponseData getTopicList(String topic, TopicType type);
 
     ResponseData deleteTopic(String topic);
+
+    ResponseData getTopicPartitionInfo(String topic);
 }

src/main/java/com/xuxd/kafka/console/service/impl/TopicServiceImpl.java

@@ -3,6 +3,7 @@ package com.xuxd.kafka.console.service.impl;
 import com.xuxd.kafka.console.beans.ResponseData;
 import com.xuxd.kafka.console.beans.enums.TopicType;
 import com.xuxd.kafka.console.beans.vo.TopicDescriptionVO;
+import com.xuxd.kafka.console.beans.vo.TopicPartitionVO;
 import com.xuxd.kafka.console.service.TopicService;
 import java.util.Collections;
 import java.util.Comparator;
@@ -70,4 +71,14 @@ public class TopicServiceImpl implements TopicService {
         Tuple2<Object, String> tuple2 = topicConsole.deleteTopic(topic);
         return (Boolean) tuple2._1 ? ResponseData.create().success() : ResponseData.create().failed(tuple2._2);
     }
+
+    @Override public ResponseData getTopicPartitionInfo(String topic) {
+        List<TopicDescription> list = topicConsole.getTopicList(Collections.singleton(topic));
+        if (list.isEmpty()) {
+            return ResponseData.create().success();
+        }
+        TopicDescription topicDescription = list.get(0);
+
+        return ResponseData.create().data(topicDescription.partitions().stream().map(p -> TopicPartitionVO.from(p))).success();
+    }
 }