GH-1727: Close Producer if initTransactions Fails

Resolves https://github.com/spring-projects/spring-kafka/issues/1727
This commit is contained in:
Gary Russell
2021-03-03 11:19:37 -05:00
parent 0cdd1d955f
commit 52408e4aad

View File

@@ -1,5 +1,5 @@
/*
* Copyright 2016-2020 the original author or authors.
* Copyright 2016-2021 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@@ -57,6 +57,7 @@ import org.springframework.context.ApplicationContextAware;
import org.springframework.context.ApplicationListener;
import org.springframework.context.event.ContextStoppedEvent;
import org.springframework.core.log.LogAccessor;
import org.springframework.kafka.KafkaException;
import org.springframework.kafka.support.TransactionSupport;
import org.springframework.lang.Nullable;
import org.springframework.util.Assert;
@@ -508,7 +509,20 @@ public class DefaultKafkaProducerFactory<K, V> implements ProducerFactory<K, V>,
this.clientIdPrefix + "-" + this.clientIdCounter.incrementAndGet());
}
newProducer = createRawProducer(newProducerConfigs);
newProducer.initTransactions();
try {
newProducer.initTransactions();
}
catch (RuntimeException ex) {
try {
newProducer.close(this.physicalCloseTimeout);
}
catch (RuntimeException ex2) {
KafkaException newEx = new KafkaException("initTransactions() failed and then close() failed", ex);
newEx.addSuppressed(ex2);
throw newEx; // NOSONAR - lost stack trace
}
throw new KafkaException("initTransactions() failed", ex);
}
return new CloseSafeProducer<>(newProducer, getCache(prefix), remover,
(String) newProducerConfigs.get(ProducerConfig.TRANSACTIONAL_ID_CONFIG), this.physicalCloseTimeout,
this.epoch);