mimaison commented on a change in pull request #10743: URL: https://github.com/apache/kafka/pull/10743#discussion_r659600886
########## File path: clients/src/main/java/org/apache/kafka/clients/admin/internals/CoordinatorStrategy.java ########## @@ -17,84 +17,160 @@ package org.apache.kafka.clients.admin.internals; import org.apache.kafka.common.errors.GroupAuthorizationException; +import org.apache.kafka.common.errors.InvalidGroupIdException; import org.apache.kafka.common.errors.TransactionalIdAuthorizationException; import org.apache.kafka.common.message.FindCoordinatorRequestData; +import org.apache.kafka.common.message.FindCoordinatorResponseData.Coordinator; import org.apache.kafka.common.protocol.Errors; import org.apache.kafka.common.requests.AbstractResponse; import org.apache.kafka.common.requests.FindCoordinatorRequest; +import org.apache.kafka.common.requests.FindCoordinatorRequest.CoordinatorType; import org.apache.kafka.common.requests.FindCoordinatorResponse; import org.apache.kafka.common.utils.LogContext; import org.slf4j.Logger; +import java.util.Collections; +import java.util.HashMap; +import java.util.Map; import java.util.Objects; import java.util.Set; +import java.util.stream.Collectors; public class CoordinatorStrategy implements AdminApiLookupStrategy<CoordinatorKey> { + + private static final ApiRequestScope GROUP_REQUEST_SCOPE = new ApiRequestScope() { }; + private static final ApiRequestScope TXN_REQUEST_SCOPE = new ApiRequestScope() { }; + private final Logger log; + private final FindCoordinatorRequest.CoordinatorType type; + private Set<CoordinatorKey> unrepresentableKeys = Collections.emptySet(); + + boolean batch = true; public CoordinatorStrategy( + FindCoordinatorRequest.CoordinatorType type, LogContext logContext ) { + this.type = type; this.log = logContext.logger(CoordinatorStrategy.class); } @Override public ApiRequestScope lookupScope(CoordinatorKey key) { - // The `FindCoordinator` API does not support batched lookups, so we use a - // separate lookup context for each coordinator key we need to lookup - return new LookupRequestScope(key); + if (batch) { + if (type == CoordinatorType.GROUP) { + return GROUP_REQUEST_SCOPE; + } else { + return TXN_REQUEST_SCOPE; + } Review comment: Right, I merged both as `BATCH_REQUEST_SCOPE` -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: jira-unsubscr...@kafka.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org