107 lines
3.7 KiB
Java
107 lines
3.7 KiB
Java
package com.xuxd.kafka.console.controller;
|
|
|
|
import com.xuxd.kafka.console.aspect.annotation.ControllerLog;
|
|
import com.xuxd.kafka.console.aspect.annotation.Permission;
|
|
import com.xuxd.kafka.console.beans.ReplicaAssignment;
|
|
import com.xuxd.kafka.console.beans.dto.AddPartitionDTO;
|
|
import com.xuxd.kafka.console.beans.dto.NewTopicDTO;
|
|
import com.xuxd.kafka.console.beans.dto.TopicThrottleDTO;
|
|
import com.xuxd.kafka.console.beans.enums.TopicType;
|
|
import com.xuxd.kafka.console.service.TopicService;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.web.bind.annotation.*;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.Collections;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
|
|
/**
|
|
* kafka-console-ui.
|
|
*
|
|
* @author xuxd
|
|
* @date 2021-09-08 20:28:35
|
|
**/
|
|
@RestController
|
|
@RequestMapping("/topic")
|
|
public class TopicController {
|
|
|
|
@Autowired
|
|
private TopicService topicService;
|
|
|
|
@GetMapping
|
|
public Object getTopicNameList() {
|
|
return topicService.getTopicNameList(false);
|
|
}
|
|
|
|
@Permission("topic:load")
|
|
@GetMapping("/list")
|
|
public Object getTopicList(@RequestParam(required = false, name = "topic") String topic, @RequestParam("type") String type) {
|
|
return topicService.getTopicList(topic, TopicType.valueOf(type.toUpperCase()));
|
|
}
|
|
|
|
@ControllerLog("删除topic")
|
|
@Permission({"topic:batch-del", "topic:del"})
|
|
@DeleteMapping
|
|
public Object deleteTopic(@RequestBody List<String> topics) {
|
|
return topicService.deleteTopics(topics);
|
|
}
|
|
|
|
@Permission("topic:partition-detail")
|
|
@GetMapping("/partition")
|
|
public Object getTopicPartitionInfo(@RequestParam("topic") String topic) {
|
|
return topicService.getTopicPartitionInfo(topic.trim());
|
|
}
|
|
|
|
@ControllerLog("创建topic")
|
|
@Permission("topic:add")
|
|
@PostMapping("/new")
|
|
public Object createNewTopic(@RequestBody NewTopicDTO topicDTO) {
|
|
return topicService.createTopic(topicDTO.toNewTopic());
|
|
}
|
|
|
|
@ControllerLog("增加topic分区")
|
|
@Permission("topic:partition-add")
|
|
@PostMapping("/partition/new")
|
|
public Object addPartition(@RequestBody AddPartitionDTO partitionDTO) {
|
|
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);
|
|
}
|
|
|
|
@GetMapping("/replica/assignment")
|
|
public Object getCurrentReplicaAssignment(@RequestParam("topic") String topic) {
|
|
return topicService.getCurrentReplicaAssignment(topic);
|
|
}
|
|
|
|
@ControllerLog("更新副本")
|
|
@Permission({"topic:replication-modify", "op:replication-reassign"})
|
|
@PostMapping("/replica/assignment")
|
|
public Object updateReplicaAssignment(@RequestBody ReplicaAssignment assignment) {
|
|
return topicService.updateReplicaAssignment(assignment);
|
|
}
|
|
|
|
@ControllerLog("配置限流")
|
|
@Permission("topic:replication-sync-throttle")
|
|
@PostMapping("/replica/throttle")
|
|
public Object configThrottle(@RequestBody TopicThrottleDTO dto) {
|
|
return topicService.configThrottle(dto.getTopic(), dto.getPartitions(), dto.getOperation());
|
|
}
|
|
|
|
@Permission("topic:send-count")
|
|
@GetMapping("/send/stats")
|
|
public Object sendStats(@RequestParam("topic") String topic) {
|
|
return topicService.sendStats(topic);
|
|
}
|
|
}
|