jiazhizhong/kafka-console-ui
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

增加分区

Browse Source
This commit is contained in:
许晓东
2021-10-19 14:40:17 +08:00
parent 0f4663d58f
commit 83c8217d73
6 changed files with 77 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/main/java/com/xuxd/kafka/console/controller/TopicController.java
View File

@@ -4,6 +4,10 @@ import com.xuxd.kafka.console.beans.dto.AddPartitionDTO;
import com.xuxd.kafka.console.beans.dto.NewTopicDTO;
import com.xuxd.kafka.console.beans.enums.TopicType;
import com.xuxd.kafka.console.service.TopicService;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.GetMapping;
@@ -48,6 +52,18 @@ public class TopicController {
@PostMapping("/partition/new")
public Object addPartition(@RequestBody AddPartitionDTO partitionDTO) {
return topicService.addPartitions(partitionDTO.getTopic().trim(), partitionDTO.getAddNum(), partitionDTO.getAssignment());
String topic = partitionDTO.getTopic().trim();
int addNum = partitionDTO.getAddNum();
Map<Integer, List<Integer>> assignmentMap = partitionDTO.getAssignment();
List<List<Integer>> assignment = Collections.emptyList();
if (!assignmentMap.isEmpty()) {
assignment = new ArrayList<>(addNum);
for (int i = 1; i <= addNum; i++) {
assignment.add(assignmentMap.containsKey(i) ? assignmentMap.get(i) : Collections.emptyList());
}
}
return topicService.addPartitions(topic, addNum, assignment);
}
}