Examples with PartitionProducerState org.apache.kafka.clients.admin.DescribeProducersResult.PartitionProducerState used on opensource projects

Example 6 with PartitionProducerState

use of org.apache.kafka.clients.admin.DescribeProducersResult.PartitionProducerState in project kafka by apache.

the class DescribeProducersHandlerTest method testUnmappedAfterNotLeaderError.

@Test
public void testUnmappedAfterNotLeaderError() {
    TopicPartition topicPartition = new TopicPartition("foo", 5);
    ApiResult<TopicPartition, PartitionProducerState> result = handleResponseWithError(new DescribeProducersOptions(), topicPartition, Errors.NOT_LEADER_OR_FOLLOWER);
    assertEquals(emptyMap(), result.failedKeys);
    assertEquals(emptyMap(), result.completedKeys);
    assertEquals(singletonList(topicPartition), result.unmappedKeys);
}

Example 7 with PartitionProducerState

use of org.apache.kafka.clients.admin.DescribeProducersResult.PartitionProducerState in project kafka by apache.

the class DescribeProducersHandlerTest method assertMatchingProducers.

private void assertMatchingProducers(PartitionResponse expected, PartitionProducerState actual) {
    List<ProducerState> expectedProducers = expected.activeProducers();
    List<org.apache.kafka.clients.admin.ProducerState> actualProducers = actual.activeProducers();
    assertEquals(expectedProducers.size(), actualProducers.size());
    Map<Long, ProducerState> expectedByProducerId = expectedProducers.stream().collect(Collectors.toMap(ProducerState::producerId, Function.identity()));
    for (org.apache.kafka.clients.admin.ProducerState actualProducerState : actualProducers) {
        ProducerState expectedProducerState = expectedByProducerId.get(actualProducerState.producerId());
        assertNotNull(expectedProducerState);
        assertEquals(expectedProducerState.producerEpoch(), actualProducerState.producerEpoch());
        assertEquals(expectedProducerState.lastSequence(), actualProducerState.lastSequence());
        assertEquals(expectedProducerState.lastTimestamp(), actualProducerState.lastTimestamp());
        assertEquals(expectedProducerState.currentTxnStartOffset(), actualProducerState.currentTransactionStartOffset().orElse(-1L));
    }
}

Example 8 with PartitionProducerState

use of org.apache.kafka.clients.admin.DescribeProducersResult.PartitionProducerState in project kafka by apache.

the class DescribeProducersHandlerTest method assertRetriableError.

private void assertRetriableError(TopicPartition topicPartition, Errors error) {
    ApiResult<TopicPartition, PartitionProducerState> result = handleResponseWithError(new DescribeProducersOptions(), topicPartition, error);
    assertEquals(emptyMap(), result.failedKeys);
    assertEquals(emptyMap(), result.completedKeys);
    assertEquals(emptyList(), result.unmappedKeys);
}

Example 9 with PartitionProducerState

use of org.apache.kafka.clients.admin.DescribeProducersResult.PartitionProducerState in project kafka by apache.

the class TransactionsCommandTest method testNewBrokerAbortTransaction.

@Test
public void testNewBrokerAbortTransaction() throws Exception {
    TopicPartition topicPartition = new TopicPartition("foo", 5);
    long startOffset = 9173;
    long producerId = 12345L;
    short producerEpoch = 15;
    int coordinatorEpoch = 76;
    String[] args = new String[] { "--bootstrap-server", "localhost:9092", "abort", "--topic", topicPartition.topic(), "--partition", String.valueOf(topicPartition.partition()), "--start-offset", String.valueOf(startOffset) };
    DescribeProducersResult describeResult = Mockito.mock(DescribeProducersResult.class);
    KafkaFuture<PartitionProducerState> describeFuture = completedFuture(new PartitionProducerState(singletonList(new ProducerState(producerId, producerEpoch, 1300, 1599509565L, OptionalInt.of(coordinatorEpoch), OptionalLong.of(startOffset)))));
    AbortTransactionResult abortTransactionResult = Mockito.mock(AbortTransactionResult.class);
    KafkaFuture<Void> abortFuture = completedFuture(null);
    AbortTransactionSpec expectedAbortSpec = new AbortTransactionSpec(topicPartition, producerId, producerEpoch, coordinatorEpoch);
    Mockito.when(describeResult.partitionResult(topicPartition)).thenReturn(describeFuture);
    Mockito.when(admin.describeProducers(singleton(topicPartition))).thenReturn(describeResult);
    Mockito.when(abortTransactionResult.all()).thenReturn(abortFuture);
    Mockito.when(admin.abortTransaction(expectedAbortSpec)).thenReturn(abortTransactionResult);
    execute(args);
    assertNormalExit();
}

Example 10 with PartitionProducerState

use of org.apache.kafka.clients.admin.DescribeProducersResult.PartitionProducerState in project kafka by apache.

the class TransactionsCommandTest method testDescribeProducers.

private void testDescribeProducers(TopicPartition topicPartition, String[] args, DescribeProducersOptions expectedOptions) throws Exception {
    DescribeProducersResult describeResult = Mockito.mock(DescribeProducersResult.class);
    KafkaFuture<PartitionProducerState> describeFuture = completedFuture(new PartitionProducerState(asList(new ProducerState(12345L, 15, 1300, 1599509565L, OptionalInt.of(20), OptionalLong.of(990)), new ProducerState(98765L, 30, 2300, 1599509599L, OptionalInt.empty(), OptionalLong.empty()))));
    Mockito.when(describeResult.partitionResult(topicPartition)).thenReturn(describeFuture);
    Mockito.when(admin.describeProducers(singleton(topicPartition), expectedOptions)).thenReturn(describeResult);
    execute(args);
    assertNormalExit();
    List<List<String>> table = readOutputAsTable();
    assertEquals(3, table.size());
    List<String> expectedHeaders = asList(TransactionsCommand.DescribeProducersCommand.HEADERS);
    assertEquals(expectedHeaders, table.get(0));
    Set<List<String>> expectedRows = Utils.mkSet(asList("12345", "15", "20", "1300", "1599509565", "990"), asList("98765", "30", "-1", "2300", "1599509599", "None"));
    assertEquals(expectedRows, new HashSet<>(table.subList(1, table.size())));
}