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

Example 1 with InitProducerIdResponse

use of org.apache.kafka.common.requests.InitProducerIdResponse in project kafka by apache.

the class TransactionManagerTest method prepareInitPidResponse.

private void prepareInitPidResponse(Errors error, boolean shouldDisconnect, long producerId, short producerEpoch) {
    InitProducerIdResponseData responseData = new InitProducerIdResponseData().setErrorCode(error.code()).setProducerEpoch(producerEpoch).setProducerId(producerId).setThrottleTimeMs(0);
    client.prepareResponse(body -> {
        InitProducerIdRequest initProducerIdRequest = (InitProducerIdRequest) body;
        assertEquals(transactionalId, initProducerIdRequest.data().transactionalId());
        assertEquals(transactionTimeoutMs, initProducerIdRequest.data().transactionTimeoutMs());
        return true;
    }, new InitProducerIdResponse(responseData), shouldDisconnect);
}

Example 2 with InitProducerIdResponse

use of org.apache.kafka.common.requests.InitProducerIdResponse in project kafka by apache.

the class TransactionManagerTest method initializeIdempotentProducerId.

private void initializeIdempotentProducerId(long producerId, short epoch) {
    InitProducerIdResponseData responseData = new InitProducerIdResponseData().setErrorCode(Errors.NONE.code()).setProducerEpoch(epoch).setProducerId(producerId).setThrottleTimeMs(0);
    client.prepareResponse(body -> {
        InitProducerIdRequest initProducerIdRequest = (InitProducerIdRequest) body;
        assertNull(initProducerIdRequest.data().transactionalId());
        return true;
    }, new InitProducerIdResponse(responseData), false);
    runUntil(transactionManager::hasProducerId);
}

Example 3 with InitProducerIdResponse

use of org.apache.kafka.common.requests.InitProducerIdResponse in project kafka by apache.

the class KafkaProducerTest method testCommitTransactionWithRecordTooLargeException.

@Test
public void testCommitTransactionWithRecordTooLargeException() throws Exception {
    Map<String, Object> configs = new HashMap<>();
    configs.put(ProducerConfig.TRANSACTIONAL_ID_CONFIG, "some.id");
    configs.put(ProducerConfig.BOOTSTRAP_SERVERS_CONFIG, "localhost:9000");
    configs.put(ProducerConfig.MAX_REQUEST_SIZE_CONFIG, 1000);
    Time time = new MockTime(1);
    MetadataResponse initialUpdateResponse = RequestTestUtils.metadataUpdateWith(1, singletonMap("topic", 1));
    ProducerMetadata metadata = mock(ProducerMetadata.class);
    MockClient client = new MockClient(time, metadata);
    client.updateMetadata(initialUpdateResponse);
    client.prepareResponse(FindCoordinatorResponse.prepareResponse(Errors.NONE, "some.id", NODE));
    client.prepareResponse(initProducerIdResponse(1L, (short) 5, Errors.NONE));
    when(metadata.fetch()).thenReturn(onePartitionCluster);
    String largeString = IntStream.range(0, 1000).mapToObj(i -> "*").collect(Collectors.joining());
    ProducerRecord<String, String> largeRecord = new ProducerRecord<>(topic, "large string", largeString);
    try (KafkaProducer<String, String> producer = kafkaProducer(configs, new StringSerializer(), new StringSerializer(), metadata, client, null, time)) {
        producer.initTransactions();
        client.prepareResponse(endTxnResponse(Errors.NONE));
        producer.beginTransaction();
        TestUtils.assertFutureError(producer.send(largeRecord), RecordTooLargeException.class);
        assertThrows(KafkaException.class, producer::commitTransaction);
    }
}

Example 4 with InitProducerIdResponse

use of org.apache.kafka.common.requests.InitProducerIdResponse in project apache-kafka-on-k8s by banzaicloud.

the class Sender method maybeWaitForProducerId.

private void maybeWaitForProducerId() {
    while (!transactionManager.hasProducerId() && !transactionManager.hasError()) {
        try {
            Node node = awaitLeastLoadedNodeReady(requestTimeout);
            if (node != null) {
                ClientResponse response = sendAndAwaitInitProducerIdRequest(node);
                InitProducerIdResponse initProducerIdResponse = (InitProducerIdResponse) response.responseBody();
                Errors error = initProducerIdResponse.error();
                if (error == Errors.NONE) {
                    ProducerIdAndEpoch producerIdAndEpoch = new ProducerIdAndEpoch(initProducerIdResponse.producerId(), initProducerIdResponse.epoch());
                    transactionManager.setProducerIdAndEpoch(producerIdAndEpoch);
                    return;
                } else if (error.exception() instanceof RetriableException) {
                    log.debug("Retriable error from InitProducerId response", error.message());
                } else {
                    transactionManager.transitionToFatalError(error.exception());
                    break;
                }
            } else {
                log.debug("Could not find an available broker to send InitProducerIdRequest to. " + "We will back off and try again.");
            }
        } catch (UnsupportedVersionException e) {
            transactionManager.transitionToFatalError(e);
            break;
        } catch (IOException e) {
            log.debug("Broker {} disconnected while awaiting InitProducerId response", e);
        }
        log.trace("Retry InitProducerIdRequest in {}ms.", retryBackoffMs);
        time.sleep(retryBackoffMs);
        metadata.requestUpdate();
    }
}

Example 5 with InitProducerIdResponse

use of org.apache.kafka.common.requests.InitProducerIdResponse in project apache-kafka-on-k8s by banzaicloud.

the class SenderTest method prepareAndReceiveInitProducerId.

private void prepareAndReceiveInitProducerId(long producerId, Errors error) {
    short producerEpoch = 0;
    if (error != Errors.NONE)
        producerEpoch = RecordBatch.NO_PRODUCER_EPOCH;
    client.prepareResponse(new MockClient.RequestMatcher() {

        @Override
        public boolean matches(AbstractRequest body) {
            return body instanceof InitProducerIdRequest && ((InitProducerIdRequest) body).transactionalId() == null;
        }
    }, new InitProducerIdResponse(0, error, producerId, producerEpoch));
    sender.run(time.milliseconds());
}