优化partition

@@ -15,6 +15,7 @@ import org.springframework.web.bind.annotation.*;
 import javax.annotation.Resource;
 import javax.annotation.Resource;
 import java.io.IOException;
 import java.io.IOException;
 import java.util.List;
 import java.util.List;
+import java.util.concurrent.ExecutionException;
 import java.util.concurrent.TimeoutException;
 import java.util.concurrent.TimeoutException;
 
 
 import static com.sunyo.wlpt.message.bus.service.common.Constant.EXIST_QUEUE_INFO;
 import static com.sunyo.wlpt.message.bus.service.common.Constant.EXIST_QUEUE_INFO;
@@ -112,10 +113,11 @@ public class BusQueueController {
      * @return {@link ResultJson}
      * @return {@link ResultJson}
      */
      */
     @PutMapping("/update")
     @PutMapping("/update")
-    public ResultJson updateBusQueue(@RequestBody BusQueue busQueue)
+    public ResultJson updateBusQueue(@RequestBody BusQueue busQueue)throws ExecutionException, InterruptedException
     {
     {
         // 先校验队列名称
         // 先校验队列名称
         String message = validateBusQueue(busQueue);
         String message = validateBusQueue(busQueue);
+        kafkaService.ediPartition(busQueue);
         return message == null
         return message == null
                 ? busQueueService.updateByPrimaryKeySelective(busQueue) > 0
                 ? busQueueService.updateByPrimaryKeySelective(busQueue) > 0
                 ? new ResultJson<>("200", "编辑MQ消息队列信息，成功")
                 ? new ResultJson<>("200", "编辑MQ消息队列信息，成功")

 package com.sunyo.wlpt.message.bus.service.service;
 package com.sunyo.wlpt.message.bus.service.service;
 
 
+import com.sunyo.wlpt.message.bus.service.domain.BusQueue;
 import com.sunyo.wlpt.message.bus.service.model.ConsumerGroupOffsets;
 import com.sunyo.wlpt.message.bus.service.model.ConsumerGroupOffsets;
 
 
 import java.util.List;
 import java.util.List;
+import java.util.concurrent.ExecutionException;
 
 
 public interface KafkaService {
 public interface KafkaService {
 
 
     boolean addTopic(String TopicName,int partitionNum);
     boolean addTopic(String TopicName,int partitionNum);
 
 
     public void updateAdminclient();
     public void updateAdminclient();
+    //by xyh
+    boolean ediPartition(BusQueue record)throws ExecutionException, InterruptedException;
 
 
     List<ConsumerGroupOffsets> queueMonitor();
     List<ConsumerGroupOffsets> queueMonitor();
 }
 }

 package com.sunyo.wlpt.message.bus.service.service.kafka;
 package com.sunyo.wlpt.message.bus.service.service.kafka;
 
 
 
 
+import com.sunyo.wlpt.message.bus.service.domain.BusQueue;
 import com.sunyo.wlpt.message.bus.service.domain.BusServer;
 import com.sunyo.wlpt.message.bus.service.domain.BusServer;
 import com.sunyo.wlpt.message.bus.service.mapper.BusServerMapper;
 import com.sunyo.wlpt.message.bus.service.mapper.BusServerMapper;
 import com.sunyo.wlpt.message.bus.service.mapper.ConsumerGroupMapper;
 import com.sunyo.wlpt.message.bus.service.mapper.ConsumerGroupMapper;
@@ -82,6 +83,17 @@ public class KafkaServiceImp implements KafkaService {
         KafkaAdmin admin = new KafkaAdmin(configs);
         KafkaAdmin admin = new KafkaAdmin(configs);
         KAFKA_ADMIN_CLIENT = AdminClient.create(admin.getConfig());
         KAFKA_ADMIN_CLIENT = AdminClient.create(admin.getConfig());
     }
     }
+
+    @Override
+    public boolean ediPartition(BusQueue record) throws ExecutionException, InterruptedException{
+        Map<String, NewPartitions> newPartitions = new HashMap<>();
+        // 将MyTopic的Partition数量调整为2
+        newPartitions.put(record.getQueueName(), NewPartitions.increaseTo(record.getPartitionCount()));
+        CreatePartitionsResult result = KAFKA_ADMIN_CLIENT.createPartitions(newPartitions);
+        System.out.println(result.all().get());
+        return true;
+    }
+
     /**
     /**
      * 获取topicList
      * 获取topicList
      */
      */