取消正在进行的副本重分配,消费组->消费详情增加刷新按钮

This commit is contained in:
许晓东
2021-11-30 19:49:02 +08:00
parent 222ba34702
commit 20535027bf
9 changed files with 288 additions and 5 deletions

View File

@@ -0,0 +1,25 @@
package com.xuxd.kafka.console.beans.vo;
import com.xuxd.kafka.console.beans.TopicPartition;
import java.util.List;
import lombok.Data;
/**
* kafka-console-ui.
*
* @author xuxd
* @date 2021-11-30 16:03:41
**/
@Data
public class CurrentReassignmentVO {
private final String topic;
private final int partition;
private final List<Integer> replicas;
private final List<Integer> addingReplicas;
private final List<Integer> removingReplicas;
}

View File

@@ -1,5 +1,6 @@
package com.xuxd.kafka.console.controller;
import com.xuxd.kafka.console.beans.TopicPartition;
import com.xuxd.kafka.console.beans.dto.BrokerThrottleDTO;
import com.xuxd.kafka.console.beans.dto.ReplicationDTO;
import com.xuxd.kafka.console.beans.dto.SyncDataDTO;
@@ -63,4 +64,14 @@ public class OperationController {
public Object removeThrottle(@RequestBody BrokerThrottleDTO dto) {
return operationService.removeThrottle(dto.getBrokerList());
}
@GetMapping("/replication/reassignments")
public Object currentReassignments() {
return operationService.currentReassignments();
}
@DeleteMapping("/replication/reassignments")
public Object cancelReassignment(@RequestBody TopicPartition partition) {
return operationService.cancelReassignment(new org.apache.kafka.common.TopicPartition(partition.getTopic(), partition.getPartition()));
}
}

View File

@@ -3,6 +3,7 @@ package com.xuxd.kafka.console.service;
import com.xuxd.kafka.console.beans.ResponseData;
import java.util.List;
import java.util.Properties;
import org.apache.kafka.common.TopicPartition;
/**
* kafka-console-ui.
@@ -25,4 +26,8 @@ public interface OperationService {
ResponseData configThrottle(List<Integer> brokerList, long size);
ResponseData removeThrottle(List<Integer> brokerList);
ResponseData currentReassignments();
ResponseData cancelReassignment(TopicPartition partition);
}

View File

@@ -5,17 +5,21 @@ import com.google.gson.Gson;
import com.google.gson.JsonObject;
import com.xuxd.kafka.console.beans.ResponseData;
import com.xuxd.kafka.console.beans.dos.MinOffsetAlignmentDO;
import com.xuxd.kafka.console.beans.vo.CurrentReassignmentVO;
import com.xuxd.kafka.console.beans.vo.OffsetAlignmentVO;
import com.xuxd.kafka.console.dao.MinOffsetAlignmentMapper;
import com.xuxd.kafka.console.service.OperationService;
import com.xuxd.kafka.console.utils.GsonUtil;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Properties;
import java.util.Set;
import java.util.stream.Collectors;
import kafka.console.OperationConsole;
import org.apache.kafka.clients.admin.PartitionReassignment;
import org.apache.kafka.common.TopicPartition;
import org.springframework.beans.factory.ObjectProvider;
import org.springframework.beans.factory.annotation.Autowired;
@@ -135,4 +139,27 @@ public class OperationServiceImpl implements OperationService {
return (boolean) tuple2._1() ? ResponseData.create().success() : ResponseData.create().failed(tuple2._2());
}
@Override public ResponseData currentReassignments() {
Map<TopicPartition, PartitionReassignment> reassignmentMap = operationConsole.currentReassignments();
List<CurrentReassignmentVO> vos = reassignmentMap.entrySet().stream().map(entry -> {
TopicPartition partition = entry.getKey();
PartitionReassignment reassignment = entry.getValue();
return new CurrentReassignmentVO(partition.topic(),
partition.partition(), reassignment.replicas(), reassignment.addingReplicas(), reassignment.removingReplicas());
}).collect(Collectors.toList());
return ResponseData.create().data(vos).success();
}
@Override public ResponseData cancelReassignment(TopicPartition partition) {
Map<TopicPartition, Throwable> res = operationConsole.cancelPartitionReassignments(Collections.singleton(partition));
if (!res.isEmpty()) {
StringBuilder sb = new StringBuilder("Failed: ");
res.forEach((p, t) -> {
sb.append(p.toString()).append(": ").append(t.getMessage()).append(System.lineSeparator());
});
return ResponseData.create().failed(sb.toString());
}
return ResponseData.create().success();
}
}