use of org.apache.kafka.clients.admin.Config in project strimzi by strimzi.
the class BaseKafkaImpl method topicMetadata.
/**
* Get a topic config via the Kafka AdminClient API, calling the given handler
* (in a different thread) with the result.
*/
@Override
public void topicMetadata(TopicName topicName, Handler<AsyncResult<TopicMetadata>> handler) {
LOGGER.debug("Getting metadata for topic {}", topicName);
ConfigResource resource = new ConfigResource(ConfigResource.Type.TOPIC, topicName.toString());
KafkaFuture<TopicDescription> descriptionFuture = adminClient.describeTopics(Collections.singleton(topicName.toString())).values().get(topicName.toString());
KafkaFuture<Config> configFuture = adminClient.describeConfigs(Collections.singleton(resource)).values().get(resource);
queueWork(new MetadataWork(descriptionFuture, configFuture, result -> handler.handle(result)));
}
use of org.apache.kafka.clients.admin.Config in project ksql by confluentinc.
the class KafkaTopicClientImplTest method topicConfigResponse.
private static DescribeConfigsResult topicConfigResponse(final String topicName, final ConfigEntry... entries) {
final Map<ConfigResource, Config> config = ImmutableMap.of(new ConfigResource(ConfigResource.Type.TOPIC, topicName), new Config(Arrays.asList(entries)));
final DescribeConfigsResult response = mock(DescribeConfigsResult.class);
expect(response.all()).andReturn(KafkaFuture.completedFuture(config));
replay(response);
return response;
}
use of org.apache.kafka.clients.admin.Config in project ksql by confluentinc.
the class KafkaTopicClientImplTest method withResourceConfig.
/*
* Config has broken hashCode & equals method:
* https://issues.apache.org/jira/browse/KAFKA-6727
*/
private static Map<ConfigResource, Config> withResourceConfig(final ConfigResource resource, final ConfigEntry... entries) {
final Set<ConfigEntry> expected = Arrays.stream(entries).collect(Collectors.toSet());
class ConfigMatcher implements IArgumentMatcher {
@SuppressWarnings("unchecked")
@Override
public boolean matches(final Object argument) {
final Map<ConfigResource, Config> request = (Map<ConfigResource, Config>) argument;
if (request.size() != 1) {
return false;
}
final Config config = request.get(resource);
if (config == null) {
return false;
}
final Set<ConfigEntry> actual = new HashSet<>(config.entries());
return actual.equals(expected);
}
@Override
public void appendTo(final StringBuffer buffer) {
buffer.append(resource).append("->").append("Config{").append(expected).append("}");
}
}
EasyMock.reportMatcher(new ConfigMatcher());
return null;
}
use of org.apache.kafka.clients.admin.Config in project ksql by confluentinc.
the class KafkaTopicClientImplTest method describeBrokerResult.
private DescribeConfigsResult describeBrokerResult() {
DescribeConfigsResult describeConfigsResult = mock(DescribeConfigsResult.class);
ConfigEntry configEntryDeleteEnable = new ConfigEntry("delete.topic.enable", "true");
List<ConfigEntry> configEntries = new ArrayList<>();
configEntries.add(configEntryDeleteEnable);
Map<ConfigResource, Config> config = ImmutableMap.of(new ConfigResource(ConfigResource.Type.BROKER, node.idString()), new Config(configEntries));
expect(describeConfigsResult.all()).andReturn(KafkaFuture.completedFuture(config));
replay(describeConfigsResult);
return describeConfigsResult;
}
use of org.apache.kafka.clients.admin.Config in project ksql by confluentinc.
the class KafkaTopicClientImpl method addTopicConfig.
@Override
public boolean addTopicConfig(final String topicName, final Map<String, ?> overrides) {
final ConfigResource resource = new ConfigResource(ConfigResource.Type.TOPIC, topicName);
try {
final Map<String, String> existingConfig = topicConfig(topicName, false);
final boolean changed = overrides.entrySet().stream().anyMatch(e -> !Objects.equals(existingConfig.get(e.getKey()), e.getValue()));
if (!changed) {
return false;
}
existingConfig.putAll(toStringConfigs(overrides));
final Set<ConfigEntry> entries = existingConfig.entrySet().stream().map(e -> new ConfigEntry(e.getKey(), e.getValue())).collect(Collectors.toSet());
final Map<ConfigResource, Config> request = Collections.singletonMap(resource, new Config(entries));
executeWithRetries(() -> adminClient.alterConfigs(request).all());
return true;
} catch (final Exception e) {
throw new KafkaResponseGetFailedException("Failed to set config for Kafka Topic " + topicName, e);
}
}
Aggregations