49 lines
2.4 KiB
Java
49 lines
2.4 KiB
Java
package com.xuxd.kafka.console.scala;
|
|
|
|
import com.xuxd.kafka.console.config.ContextConfig;
|
|
import com.xuxd.kafka.console.config.ContextConfigHolder;
|
|
import com.xuxd.kafka.console.config.KafkaConfig;
|
|
import kafka.console.ClientQuotaConsole;
|
|
import org.apache.kafka.common.config.internals.QuotaConfigs;
|
|
import org.apache.kafka.common.quota.ClientQuotaEntity;
|
|
import org.junit.jupiter.api.Test;
|
|
|
|
import java.util.Arrays;
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
|
|
public class ClientQuotaConsoleTest {
|
|
|
|
String bootstrapServer = "localhost:9092";
|
|
|
|
@Test
|
|
void testGetClientQuotasConfigs() {
|
|
ClientQuotaConsole console = new ClientQuotaConsole(new KafkaConfig());
|
|
ContextConfig config = new ContextConfig();
|
|
config.setBootstrapServer(bootstrapServer);
|
|
ContextConfigHolder.CONTEXT_CONFIG.set(config);
|
|
Map<ClientQuotaEntity, Map<String, Object>> configs = console.getClientQuotasConfigs(Arrays.asList(ClientQuotaEntity.USER, ClientQuotaEntity.CLIENT_ID), Arrays.asList("user1", "clientA"));
|
|
configs.forEach((k, v) -> {
|
|
System.out.println(k);
|
|
System.out.println(v);
|
|
});
|
|
}
|
|
|
|
@Test
|
|
void testAlterClientQuotasConfigs() {
|
|
ClientQuotaConsole console = new ClientQuotaConsole(new KafkaConfig());
|
|
ContextConfig config = new ContextConfig();
|
|
config.setBootstrapServer(bootstrapServer);
|
|
ContextConfigHolder.CONTEXT_CONFIG.set(config);
|
|
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.addQuotaConfigs(Arrays.asList(ClientQuotaEntity.CLIENT_ID), Arrays.asList(""), configsToBeAddedMap);
|
|
console.addQuotaConfigs(Arrays.asList(ClientQuotaEntity.CLIENT_ID), Arrays.asList("clientA"), configsToBeAddedMap);
|
|
console.addQuotaConfigs(Arrays.asList(ClientQuotaEntity.USER, ClientQuotaEntity.CLIENT_ID), Arrays.asList("", ""), configsToBeAddedMap);
|
|
// console.deleteQuotaConfigs(Arrays.asList(ClientQuotaEntity.CLIENT_ID), Arrays.asList(""), Arrays.asList(QuotaConfigs.CONSUMER_BYTE_RATE_OVERRIDE_CONFIG));
|
|
}
|
|
}
|