Examples with DescribeConfigsRequestData org.apache.kafka.common.message.DescribeConfigsRequestData used on opensource projects

Example 1 with DescribeConfigsRequestData

use of org.apache.kafka.common.message.DescribeConfigsRequestData in project kafka by apache.

the class KafkaAdminClient method describeConfigs.

@Override
public DescribeConfigsResult describeConfigs(Collection<ConfigResource> configResources, final DescribeConfigsOptions options) {
    // Partition the requested config resources based on which broker they must be sent to with the
    // null broker being used for config resources which can be obtained from any broker
    final Map<Integer, Map<ConfigResource, KafkaFutureImpl<Config>>> brokerFutures = new HashMap<>(configResources.size());
    for (ConfigResource resource : configResources) {
        Integer broker = nodeFor(resource);
        brokerFutures.compute(broker, (key, value) -> {
            if (value == null) {
                value = new HashMap<>();
            }
            value.put(resource, new KafkaFutureImpl<>());
            return value;
        });
    }
    final long now = time.milliseconds();
    for (Map.Entry<Integer, Map<ConfigResource, KafkaFutureImpl<Config>>> entry : brokerFutures.entrySet()) {
        Integer broker = entry.getKey();
        Map<ConfigResource, KafkaFutureImpl<Config>> unified = entry.getValue();
        runnable.call(new Call("describeConfigs", calcDeadlineMs(now, options.timeoutMs()), broker != null ? new ConstantNodeIdProvider(broker) : new LeastLoadedNodeProvider()) {

            @Override
            DescribeConfigsRequest.Builder createRequest(int timeoutMs) {
                return new DescribeConfigsRequest.Builder(new DescribeConfigsRequestData().setResources(unified.keySet().stream().map(config -> new DescribeConfigsRequestData.DescribeConfigsResource().setResourceName(config.name()).setResourceType(config.type().id()).setConfigurationKeys(null)).collect(Collectors.toList())).setIncludeSynonyms(options.includeSynonyms()).setIncludeDocumentation(options.includeDocumentation()));
            }

            @Override
            void handleResponse(AbstractResponse abstractResponse) {
                DescribeConfigsResponse response = (DescribeConfigsResponse) abstractResponse;
                for (Map.Entry<ConfigResource, DescribeConfigsResponseData.DescribeConfigsResult> entry : response.resultMap().entrySet()) {
                    ConfigResource configResource = entry.getKey();
                    DescribeConfigsResponseData.DescribeConfigsResult describeConfigsResult = entry.getValue();
                    KafkaFutureImpl<Config> future = unified.get(configResource);
                    if (future == null) {
                        if (broker != null) {
                            log.warn("The config {} in the response from broker {} is not in the request", configResource, broker);
                        } else {
                            log.warn("The config {} in the response from the least loaded broker is not in the request", configResource);
                        }
                    } else {
                        if (describeConfigsResult.errorCode() != Errors.NONE.code()) {
                            future.completeExceptionally(Errors.forCode(describeConfigsResult.errorCode()).exception(describeConfigsResult.errorMessage()));
                        } else {
                            future.complete(describeConfigResult(describeConfigsResult));
                        }
                    }
                }
                completeUnrealizedFutures(unified.entrySet().stream(), configResource -> "The broker response did not contain a result for config resource " + configResource);
            }

            @Override
            void handleFailure(Throwable throwable) {
                completeAllExceptionally(unified.values(), throwable);
            }
        }, now);
    }
    return new DescribeConfigsResult(new HashMap<>(brokerFutures.entrySet().stream().flatMap(x -> x.getValue().entrySet().stream()).collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue))));
}