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:
许晓东
2023-01-30 21:40:11 +08:00
parent 832b20a83e
commit 56621e0b8c
9 changed files with 127 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/main/java/com/xuxd/kafka/console/beans/dto/AlterClientQuotaDTO.java Normal file
View File

@@ -0,0 +1,23 @@
package com.xuxd.kafka.console.beans.dto;
import lombok.Data;
import java.util.List;
/**
* @author: xuxd
* @date: 2023/1/10 20:12
**/
@Data
public class AlterClientQuotaDTO {
private List<String> types;
private List<String> names;
private String consumerRate;
private String producerRate;
private String requestPercentage;
}

13
src/main/java/com/xuxd/kafka/console/controller/ClientQuotaController.java
View File

@@ -1,7 +1,10 @@
package com.xuxd.kafka.console.controller;
import com.xuxd.kafka.console.beans.ResponseData;
import com.xuxd.kafka.console.beans.dto.AlterClientQuotaDTO;
import com.xuxd.kafka.console.beans.dto.QueryClientQuotaDTO;
import com.xuxd.kafka.console.service.ClientQuotaService;
import org.apache.commons.collections.CollectionUtils;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
@@ -25,4 +28,14 @@ public class ClientQuotaController {
public Object getClientQuotaConfigs(@RequestBody QueryClientQuotaDTO request) {
return clientQuotaService.getClientQuotaConfigs(request.getTypes(), request.getNames());
}
@PostMapping
public Object alterClientQuotaConfigs(@RequestBody AlterClientQuotaDTO request) {
if (CollectionUtils.isEmpty(request.getTypes())
|| CollectionUtils.isEmpty(request.getNames())
|| request.getTypes().size() != request.getNames().size()) {
return ResponseData.create().failed("types length and names length is invalid.");
}
return clientQuotaService.alterClientQuotaConfigs(request);
}
}

3
src/main/java/com/xuxd/kafka/console/service/ClientQuotaService.java
View File

@@ -1,5 +1,6 @@
package com.xuxd.kafka.console.service;
import com.xuxd.kafka.console.beans.dto.AlterClientQuotaDTO;
import com.xuxd.kafka.console.beans.vo.ClientQuotaEntityVO;
import java.util.List;
@@ -10,4 +11,6 @@ import java.util.List;
public interface ClientQuotaService {
List<ClientQuotaEntityVO> getClientQuotaConfigs(List<String> types, List<String> names);
Object alterClientQuotaConfigs(AlterClientQuotaDTO request);
}

8
src/main/java/com/xuxd/kafka/console/service/impl/ClientQuotaServiceImpl.java
View File

@@ -1,5 +1,7 @@
package com.xuxd.kafka.console.service.impl;
import com.xuxd.kafka.console.beans.ResponseData;
import com.xuxd.kafka.console.beans.dto.AlterClientQuotaDTO;
import com.xuxd.kafka.console.beans.vo.ClientQuotaEntityVO;
import com.xuxd.kafka.console.service.ClientQuotaService;
import kafka.console.ClientQuotaConsole;
@@ -57,4 +59,10 @@ public class ClientQuotaServiceImpl implements ClientQuotaService {
return voList.stream().filter(e -> names.get(1).equals(e.getClient())).collect(Collectors.toList());
}
@Override
public Object alterClientQuotaConfigs(AlterClientQuotaDTO request) {
return ResponseData.create().failed();
}
}

8
src/test/java/com/xuxd/kafka/console/scala/ClientQuotaConsoleTest.java
View File

@@ -14,7 +14,7 @@ import java.util.Map;
public class ClientQuotaConsoleTest {
String bootstrapServer = "127.0.0.1:9092";
String bootstrapServer = "10.1.18.222:9092";
@Test
void testGetClientQuotasConfigs() {
@@ -38,8 +38,8 @@ public class ClientQuotaConsoleTest {
Map<String, String> configsToBeAddedMap = new HashMap<>();
configsToBeAddedMap.put(QuotaConfigs.PRODUCER_BYTE_RATE_OVERRIDE_CONFIG, "1024000000");
console.addQuotaConfigs(Arrays.asList(ClientQuotaEntity.USER), Arrays.asList("user-test"), configsToBeAddedMap);
console.addQuotaConfigs(Arrays.asList(ClientQuotaEntity.USER), Arrays.asList(""), configsToBeAddedMap);
console.deleteQuotaConfigs(Arrays.asList(ClientQuotaEntity.USER), Arrays.asList(""), Arrays.asList(QuotaConfigs.PRODUCER_BYTE_RATE_OVERRIDE_CONFIG));
// console.addQuotaConfigs(Arrays.asList(ClientQuotaEntity.USER), Arrays.asList("user-test"), configsToBeAddedMap);
// console.addQuotaConfigs(Arrays.asList(ClientQuotaEntity.USER), Arrays.asList(""), configsToBeAddedMap);
console.deleteQuotaConfigs(Arrays.asList(ClientQuotaEntity.CLIENT_ID), Arrays.asList(""), Arrays.asList(QuotaConfigs.CONSUMER_BYTE_RATE_OVERRIDE_CONFIG));
}
}