扩分区

kafka-manager-common/src/main/java/com/xiaojukeji/kafka/manager/common/entity/dto/normal/TopicExpandDTO.java

@@ -0,0 +1,57 @@
+package com.xiaojukeji.kafka.manager.common.entity.dto.normal;
+
+import com.xiaojukeji.kafka.manager.common.entity.dto.ClusterTopicDTO;
+import com.xiaojukeji.kafka.manager.common.utils.ValidateUtils;
+import io.swagger.annotations.ApiModel;
+import io.swagger.annotations.ApiModelProperty;
+
+import java.util.List;
+
+@ApiModel(description = "扩分区")
+public class TopicExpandDTO extends ClusterTopicDTO {
+
+  @ApiModelProperty(value = "regionId")
+  private Long regionId;
+
+  @ApiModelProperty(value = "brokerId列表")
+  private List<Integer> brokerIds;
+
+  @ApiModelProperty(value = "新增分区数")
+  private Integer partitionNum;
+
+  public Long getRegionId() {
+    return regionId;
+  }
+
+  public void setRegionId(Long regionId) {
+    this.regionId = regionId;
+  }
+
+  public List<Integer> getBrokerIds() {
+    return brokerIds;
+  }
+
+  public void setBrokerIds(List<Integer> brokerIds) {
+    this.brokerIds = brokerIds;
+  }
+
+  public Integer getPartitionNum() {
+    return partitionNum;
+  }
+
+  public void setPartitionNum(Integer partitionNum) {
+    this.partitionNum = partitionNum;
+  }
+
+  public boolean paramLegal() {
+    if (ValidateUtils.isNull(clusterId)
+        || ValidateUtils.isNull(topicName)
+        || ValidateUtils.isNull(partitionNum) || partitionNum <= 0) {
+      return false;
+    }
+    if (ValidateUtils.isEmptyList(brokerIds) && ValidateUtils.isNull(regionId)) {
+      return false;
+    }
+    return true;
+  }
+}

kafka-manager-core/src/main/java/com/xiaojukeji/kafka/manager/service/service/TopicService.java

@@ -7,6 +7,7 @@ import com.xiaojukeji.kafka.manager.common.entity.ao.*;
 import com.xiaojukeji.kafka.manager.common.entity.dto.normal.TopicAddDTO;
 import com.xiaojukeji.kafka.manager.common.entity.dto.normal.TopicDataSampleDTO;
 import com.xiaojukeji.kafka.manager.common.entity.ao.topic.*;
+import com.xiaojukeji.kafka.manager.common.entity.dto.normal.TopicExpandDTO;
 import com.xiaojukeji.kafka.manager.common.entity.dto.normal.TopicQuotaDTO;
 import com.xiaojukeji.kafka.manager.common.entity.metrics.BaseMetrics;
 import com.xiaojukeji.kafka.manager.common.entity.pojo.TopicMetricsDO;
@@ -122,4 +123,9 @@ public interface TopicService {
      * 配额调整
      */
     Result addTopicQuota(TopicQuotaDTO dto);
+
+    /**
+     * 扩分区
+     */
+    Result expandTopic(TopicExpandDTO dto);
 }

kafka-manager-core/src/main/java/com/xiaojukeji/kafka/manager/service/service/impl/TopicServiceImpl.java

@@ -6,6 +6,7 @@ import com.xiaojukeji.kafka.manager.common.entity.Result;
 import com.xiaojukeji.kafka.manager.common.entity.ResultStatus;
 import com.xiaojukeji.kafka.manager.common.entity.ao.gateway.TopicQuota;
 import com.xiaojukeji.kafka.manager.common.entity.dto.normal.TopicAddDTO;
+import com.xiaojukeji.kafka.manager.common.entity.dto.normal.TopicExpandDTO;
 import com.xiaojukeji.kafka.manager.common.entity.dto.normal.TopicQuotaDTO;
 import com.xiaojukeji.kafka.manager.common.entity.pojo.gateway.AppDO;
 import com.xiaojukeji.kafka.manager.common.bizenum.OffsetPosEnum;
@@ -911,6 +912,28 @@ public class TopicServiceImpl implements TopicService {
       return Result.buildFrom(ResultStatus.FAIL);
   }
 
+    @Override
+    public Result expandTopic(TopicExpandDTO dto) {
+        // 校验非空
+        if (!dto.paramLegal()) {
+            return Result.buildFrom(ResultStatus.PARAM_ILLEGAL);
+        }
+        //获取物理集群id
+        Long physicalClusterId = logicalClusterMetadataManager.getPhysicalClusterId(dto.getClusterId());
+        if (ValidateUtils.isNull(physicalClusterId)) {
+            return Result.buildFrom(ResultStatus.PARAM_ILLEGAL);
+        }
+        //获取集群信息
+        ClusterDO clusterDO = clusterService.getById(physicalClusterId);
+        if (ValidateUtils.isNull(clusterDO)) {
+            return Result.buildFrom(ResultStatus.CLUSTER_NOT_EXIST);
+        }
+        //扩分区
+        ResultStatus resultStatus = adminService.expandPartitions(clusterDO, dto.getTopicName(), dto.getPartitionNum(),
+            dto.getRegionId(), dto.getBrokerIds(), SpringTool.getUserName());
+        return Result.buildFrom(resultStatus);
+    }
+
     private Result<TopicOffsetChangedEnum> checkTopicOffsetChanged(ClusterDO clusterDO,
                                                                    String topicName,
                                                                    Map<TopicPartition, Long> endOffsetMap) {

kafka-manager-web/src/main/java/com/xiaojukeji/kafka/manager/web/api/versionone/normal/NormalTopicController.java

@@ -8,6 +8,7 @@ import com.xiaojukeji.kafka.manager.common.entity.ao.topic.TopicConnection;
 import com.xiaojukeji.kafka.manager.common.entity.ao.topic.TopicPartitionDTO;
 import com.xiaojukeji.kafka.manager.common.entity.dto.normal.TopicAddDTO;
 import com.xiaojukeji.kafka.manager.common.entity.dto.normal.TopicDataSampleDTO;
+import com.xiaojukeji.kafka.manager.common.entity.dto.normal.TopicExpandDTO;
 import com.xiaojukeji.kafka.manager.common.entity.dto.normal.TopicQuotaDTO;
 import com.xiaojukeji.kafka.manager.common.entity.metrics.BaseMetrics;
 import com.xiaojukeji.kafka.manager.common.entity.vo.common.RealTimeMetricsVO;
@@ -351,6 +352,17 @@ public class NormalTopicController {
         return topicService.addTopic(dto);
     }
 
+    @ApiOperation(value = "扩分区",notes = "扩分区")
+    @RequestMapping(value = "{/topics/expand}",method = RequestMethod.POST)
+    @ResponseBody
+    public Result expandTopic(@RequestBody TopicExpandDTO dto) {
+        if (ValidateUtils.isNull(dto)) {
+            return Result.buildFrom(ResultStatus.PARAM_ILLEGAL);
+        }
+        return topicService.expandTopic(dto);
+    }
+
+
     @ApiOperation(value = "删除topic",notes = "删除topic")
     @RequestMapping(value = {"{clusterId}/topics/{topicName}/delete"},method = RequestMethod.DELETE)
     @ResponseBody