Search in sources :

Example 6 with DescribedGroup

use of org.apache.kafka.common.message.DescribeGroupsResponseData.DescribedGroup in project kafka by apache.

the class RequestResponseTest method createDescribeGroupResponse.

private DescribeGroupsResponse createDescribeGroupResponse() {
    String clientId = "consumer-1";
    String clientHost = "localhost";
    DescribeGroupsResponseData describeGroupsResponseData = new DescribeGroupsResponseData();
    DescribeGroupsResponseData.DescribedGroupMember member = DescribeGroupsResponse.groupMember("memberId", null, clientId, clientHost, new byte[0], new byte[0]);
    DescribedGroup metadata = DescribeGroupsResponse.groupMetadata("test-group", Errors.NONE, "STABLE", "consumer", "roundrobin", singletonList(member), DescribeGroupsResponse.AUTHORIZED_OPERATIONS_OMITTED);
    describeGroupsResponseData.groups().add(metadata);
    return new DescribeGroupsResponse(describeGroupsResponseData);
}
Also used : DescribeGroupsResponseData(org.apache.kafka.common.message.DescribeGroupsResponseData) DescribedGroup(org.apache.kafka.common.message.DescribeGroupsResponseData.DescribedGroup)

Example 7 with DescribedGroup

use of org.apache.kafka.common.message.DescribeGroupsResponseData.DescribedGroup in project kafka by apache.

the class DescribeConsumerGroupsHandler method handleResponse.

@Override
public ApiResult<CoordinatorKey, ConsumerGroupDescription> handleResponse(Node coordinator, Set<CoordinatorKey> groupIds, AbstractResponse abstractResponse) {
    final DescribeGroupsResponse response = (DescribeGroupsResponse) abstractResponse;
    final Map<CoordinatorKey, ConsumerGroupDescription> completed = new HashMap<>();
    final Map<CoordinatorKey, Throwable> failed = new HashMap<>();
    final Set<CoordinatorKey> groupsToUnmap = new HashSet<>();
    for (DescribedGroup describedGroup : response.data().groups()) {
        CoordinatorKey groupIdKey = CoordinatorKey.byGroupId(describedGroup.groupId());
        Errors error = Errors.forCode(describedGroup.errorCode());
        if (error != Errors.NONE) {
            handleError(groupIdKey, error, failed, groupsToUnmap);
            continue;
        }
        final String protocolType = describedGroup.protocolType();
        if (protocolType.equals(ConsumerProtocol.PROTOCOL_TYPE) || protocolType.isEmpty()) {
            final List<DescribedGroupMember> members = describedGroup.members();
            final List<MemberDescription> memberDescriptions = new ArrayList<>(members.size());
            final Set<AclOperation> authorizedOperations = validAclOperations(describedGroup.authorizedOperations());
            for (DescribedGroupMember groupMember : members) {
                Set<TopicPartition> partitions = Collections.emptySet();
                if (groupMember.memberAssignment().length > 0) {
                    final Assignment assignment = ConsumerProtocol.deserializeAssignment(ByteBuffer.wrap(groupMember.memberAssignment()));
                    partitions = new HashSet<>(assignment.partitions());
                }
                memberDescriptions.add(new MemberDescription(groupMember.memberId(), Optional.ofNullable(groupMember.groupInstanceId()), groupMember.clientId(), groupMember.clientHost(), new MemberAssignment(partitions)));
            }
            final ConsumerGroupDescription consumerGroupDescription = new ConsumerGroupDescription(groupIdKey.idValue, protocolType.isEmpty(), memberDescriptions, describedGroup.protocolData(), ConsumerGroupState.parse(describedGroup.groupState()), coordinator, authorizedOperations);
            completed.put(groupIdKey, consumerGroupDescription);
        } else {
            failed.put(groupIdKey, new IllegalArgumentException(String.format("GroupId %s is not a consumer group (%s).", groupIdKey.idValue, protocolType)));
        }
    }
    return new ApiResult<>(completed, failed, new ArrayList<>(groupsToUnmap));
}
Also used : HashMap(java.util.HashMap) MemberDescription(org.apache.kafka.clients.admin.MemberDescription) ArrayList(java.util.ArrayList) AclOperation(org.apache.kafka.common.acl.AclOperation) DescribedGroup(org.apache.kafka.common.message.DescribeGroupsResponseData.DescribedGroup) Assignment(org.apache.kafka.clients.consumer.ConsumerPartitionAssignor.Assignment) MemberAssignment(org.apache.kafka.clients.admin.MemberAssignment) MemberAssignment(org.apache.kafka.clients.admin.MemberAssignment) DescribedGroupMember(org.apache.kafka.common.message.DescribeGroupsResponseData.DescribedGroupMember) DescribeGroupsResponse(org.apache.kafka.common.requests.DescribeGroupsResponse) HashSet(java.util.HashSet) ConsumerGroupDescription(org.apache.kafka.clients.admin.ConsumerGroupDescription) Errors(org.apache.kafka.common.protocol.Errors) TopicPartition(org.apache.kafka.common.TopicPartition)

Aggregations

DescribedGroup (org.apache.kafka.common.message.DescribeGroupsResponseData.DescribedGroup)7 DescribedGroupMember (org.apache.kafka.common.message.DescribeGroupsResponseData.DescribedGroupMember)4 Test (org.junit.jupiter.api.Test)3 ArrayList (java.util.ArrayList)1 HashMap (java.util.HashMap)1 HashSet (java.util.HashSet)1 ConsumerGroupDescription (org.apache.kafka.clients.admin.ConsumerGroupDescription)1 MemberAssignment (org.apache.kafka.clients.admin.MemberAssignment)1 MemberDescription (org.apache.kafka.clients.admin.MemberDescription)1 Assignment (org.apache.kafka.clients.consumer.ConsumerPartitionAssignor.Assignment)1 TopicPartition (org.apache.kafka.common.TopicPartition)1 AclOperation (org.apache.kafka.common.acl.AclOperation)1 DescribeGroupsResponseData (org.apache.kafka.common.message.DescribeGroupsResponseData)1 Errors (org.apache.kafka.common.protocol.Errors)1 DescribeGroupsResponse (org.apache.kafka.common.requests.DescribeGroupsResponse)1