Examples with Builder org.apache.kafka.common.requests.OffsetFetchRequest.Builder used on opensource projects

Example 1 with Builder

use of org.apache.kafka.common.requests.OffsetFetchRequest.Builder in project kafka by apache.

the class OffsetFetchRequestTest method testConstructor.

@Test
public void testConstructor() {
    List<TopicPartition> partitions = Arrays.asList(new TopicPartition(topicOne, partitionOne), new TopicPartition(topicTwo, partitionTwo));
    int throttleTimeMs = 10;
    Map<TopicPartition, PartitionData> expectedData = new HashMap<>();
    for (TopicPartition partition : partitions) {
        expectedData.put(partition, new PartitionData(OffsetFetchResponse.INVALID_OFFSET, Optional.empty(), OffsetFetchResponse.NO_METADATA, Errors.NONE));
    }
    for (short version : ApiKeys.OFFSET_FETCH.allVersions()) {
        if (version < 8) {
            builder = new OffsetFetchRequest.Builder(group1, false, partitions, false);
            assertFalse(builder.isAllTopicPartitions());
            OffsetFetchRequest request = builder.build(version);
            assertFalse(request.isAllPartitions());
            assertEquals(group1, request.groupId());
            assertEquals(partitions, request.partitions());
            OffsetFetchResponse response = request.getErrorResponse(throttleTimeMs, Errors.NONE);
            assertEquals(Errors.NONE, response.error());
            assertFalse(response.hasError());
            assertEquals(Collections.singletonMap(Errors.NONE, version <= (short) 1 ? 3 : 1), response.errorCounts(), "Incorrect error count for version " + version);
            if (version <= 1) {
                assertEquals(expectedData, response.responseDataV0ToV7());
            }
            if (version >= 3) {
                assertEquals(throttleTimeMs, response.throttleTimeMs());
            } else {
                assertEquals(DEFAULT_THROTTLE_TIME, response.throttleTimeMs());
            }
        } else {
            builder = new Builder(Collections.singletonMap(group1, partitions), false, false);
            OffsetFetchRequest request = builder.build(version);
            Map<String, List<TopicPartition>> groupToPartitionMap = request.groupIdsToPartitions();
            Map<String, List<OffsetFetchRequestTopics>> groupToTopicMap = request.groupIdsToTopics();
            assertFalse(request.isAllPartitionsForGroup(group1));
            assertTrue(groupToPartitionMap.containsKey(group1) && groupToTopicMap.containsKey(group1));
            assertEquals(partitions, groupToPartitionMap.get(group1));
            OffsetFetchResponse response = request.getErrorResponse(throttleTimeMs, Errors.NONE);
            assertEquals(Errors.NONE, response.groupLevelError(group1));
            assertFalse(response.groupHasError(group1));
            assertEquals(Collections.singletonMap(Errors.NONE, 1), response.errorCounts(), "Incorrect error count for version " + version);
            assertEquals(throttleTimeMs, response.throttleTimeMs());
        }
    }
}

Example 2 with Builder

use of org.apache.kafka.common.requests.OffsetFetchRequest.Builder in project kafka by apache.

the class OffsetFetchRequestTest method testConstructorWithMultipleGroups.

@Test
public void testConstructorWithMultipleGroups() {
    List<TopicPartition> topic1Partitions = Arrays.asList(new TopicPartition(topicOne, partitionOne), new TopicPartition(topicOne, partitionTwo));
    List<TopicPartition> topic2Partitions = Arrays.asList(new TopicPartition(topicTwo, partitionOne), new TopicPartition(topicTwo, partitionTwo));
    List<TopicPartition> topic3Partitions = Arrays.asList(new TopicPartition(topicThree, partitionOne), new TopicPartition(topicThree, partitionTwo));
    Map<String, List<TopicPartition>> groupToTp = new HashMap<>();
    groupToTp.put(group1, topic1Partitions);
    groupToTp.put(group2, topic2Partitions);
    groupToTp.put(group3, topic3Partitions);
    groupToTp.put(group4, null);
    groupToTp.put(group5, null);
    int throttleTimeMs = 10;
    for (short version : ApiKeys.OFFSET_FETCH.allVersions()) {
        if (version >= 8) {
            builder = new Builder(groupToTp, false, false);
            OffsetFetchRequest request = builder.build(version);
            Map<String, List<TopicPartition>> groupToPartitionMap = request.groupIdsToPartitions();
            Map<String, List<OffsetFetchRequestTopics>> groupToTopicMap = request.groupIdsToTopics();
            assertEquals(groupToTp.keySet(), groupToTopicMap.keySet());
            assertEquals(groupToTp.keySet(), groupToPartitionMap.keySet());
            assertFalse(request.isAllPartitionsForGroup(group1));
            assertFalse(request.isAllPartitionsForGroup(group2));
            assertFalse(request.isAllPartitionsForGroup(group3));
            assertTrue(request.isAllPartitionsForGroup(group4));
            assertTrue(request.isAllPartitionsForGroup(group5));
            OffsetFetchResponse response = request.getErrorResponse(throttleTimeMs, Errors.NONE);
            for (String group : groups) {
                assertEquals(Errors.NONE, response.groupLevelError(group));
                assertFalse(response.groupHasError(group));
            }
            assertEquals(Collections.singletonMap(Errors.NONE, 5), response.errorCounts(), "Incorrect error count for version " + version);
            assertEquals(throttleTimeMs, response.throttleTimeMs());
        }
    }
}

Example 3 with Builder

use of org.apache.kafka.common.requests.OffsetFetchRequest.Builder in project kafka by apache.

the class OffsetFetchRequestTest method testBuildThrowForUnsupportedBatchRequest.

@Test
public void testBuildThrowForUnsupportedBatchRequest() {
    for (int version : listOfVersionsNonBatchOffsetFetch) {
        Map<String, List<TopicPartition>> groupPartitionMap = new HashMap<>();
        groupPartitionMap.put(group1, null);
        groupPartitionMap.put(group2, null);
        builder = new Builder(groupPartitionMap, true, false);
        final short finalVersion = (short) version;
        assertThrows(NoBatchedOffsetFetchRequestException.class, () -> builder.build(finalVersion));
    }
}

Example 4 with Builder

use of org.apache.kafka.common.requests.OffsetFetchRequest.Builder in project kafka by apache.

the class OffsetFetchRequestTest method testConstructorFailForUnsupportedRequireStable.

@Test
public void testConstructorFailForUnsupportedRequireStable() {
    for (short version : ApiKeys.OFFSET_FETCH.allVersions()) {
        if (version < 8) {
            // The builder needs to be initialized every cycle as the internal data `requireStable` flag is flipped.
            builder = new OffsetFetchRequest.Builder(group1, true, null, false);
            final short finalVersion = version;
            if (version < 2) {
                assertThrows(UnsupportedVersionException.class, () -> builder.build(finalVersion));
            } else {
                OffsetFetchRequest request = builder.build(finalVersion);
                assertEquals(group1, request.groupId());
                assertNull(request.partitions());
                assertTrue(request.isAllPartitions());
                if (version < 7) {
                    assertFalse(request.requireStable());
                } else {
                    assertTrue(request.requireStable());
                }
            }
        } else {
            builder = new Builder(Collections.singletonMap(group1, null), true, false);
            OffsetFetchRequest request = builder.build(version);
            Map<String, List<TopicPartition>> groupToPartitionMap = request.groupIdsToPartitions();
            Map<String, List<OffsetFetchRequestTopics>> groupToTopicMap = request.groupIdsToTopics();
            assertTrue(groupToPartitionMap.containsKey(group1) && groupToTopicMap.containsKey(group1));
            assertNull(groupToPartitionMap.get(group1));
            assertTrue(request.isAllPartitionsForGroup(group1));
            assertTrue(request.requireStable());
        }
    }
}

Example 5 with Builder

use of org.apache.kafka.common.requests.OffsetFetchRequest.Builder in project kafka by apache.

the class OffsetFetchRequestTest method testBuildThrowForUnsupportedRequireStable.

@Test
public void testBuildThrowForUnsupportedRequireStable() {
    for (int version : listOfVersionsNonBatchOffsetFetch) {
        builder = new OffsetFetchRequest.Builder(group1, true, null, true);
        if (version < 7) {
            final short finalVersion = (short) version;
            assertThrows(UnsupportedVersionException.class, () -> builder.build(finalVersion));
        } else {
            OffsetFetchRequest request = builder.build((short) version);
            assertTrue(request.requireStable());
        }
    }
}