GH-3146: Propagate KT observation callback

Fixes: #3146

Propagate `KafkaTemplate` observation callback.

There is a requirement to see logs correlated in the `CompletableFuture` result
after `KafkaTemplate.send()` operation.

* Add `observation.openScope()` into the `buildCallback()` logic

# Conflicts:
#	spring-kafka/src/main/java/org/springframework/kafka/core/KafkaTemplate.java

(cherry picked from commit fffd5ef)

Author: artembilan

spring-kafka/src/main/java/org/springframework/kafka/core/KafkaTemplate.java

@@ -827,6 +827,7 @@ private ProducerRecord<K, V> interceptorProducerRecord(ProducerRecord<K, V> prod
 		return producerRecord;
 	}
 
+	@SuppressWarnings("try")
 	private Callback buildCallback(final ProducerRecord<K, V> producerRecord, final Producer<K, V> producer,
 			final CompletableFuture<SendResult<K, V>> future, @Nullable Object sample, Observation observation) {
 
@@ -839,10 +840,9 @@ private Callback buildCallback(final ProducerRecord<K, V> producerRecord, final
 			catch (Exception e) {
 				KafkaTemplate.this.logger.warn(e, () ->  "Error executing interceptor onAcknowledgement callback");
 			}
-			try {
+			try (Observation.Scope ignored = observation.openScope()) {
 				if (exception == null) {
 					successTimer(sample, producerRecord);
-					observation.stop();
 					future.complete(new SendResult<>(producerRecord, metadata));
 					if (KafkaTemplate.this.producerListener != null) {
 						KafkaTemplate.this.producerListener.onSuccess(producerRecord, metadata);
@@ -853,7 +853,6 @@ private Callback buildCallback(final ProducerRecord<K, V> producerRecord, final
 				else {
 					failureTimer(sample, exception, producerRecord);
 					observation.error(exception);
-					observation.stop();
 					future.completeExceptionally(
 							new KafkaProducerException(producerRecord, "Failed to send", exception));
 					if (KafkaTemplate.this.producerListener != null) {
@@ -864,6 +863,7 @@ private Callback buildCallback(final ProducerRecord<K, V> producerRecord, final
 				}
 			}
 			finally {
+				observation.stop();
 				if (!KafkaTemplate.this.transactional) {
 					closeProducer(producer, false);
 				}

spring-kafka/src/test/java/org/springframework/kafka/support/micrometer/ObservationTests.java

@@ -29,6 +29,7 @@
 import java.util.concurrent.ExecutionException;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.TimeoutException;
+import java.util.concurrent.atomic.AtomicReference;
 
 import org.apache.kafka.clients.admin.AdminClientConfig;
 import org.apache.kafka.clients.consumer.ConsumerConfig;
@@ -104,7 +105,14 @@ void endToEnd(@Autowired Listener listener, @Autowired KafkaTemplate<Integer, St
 			@Autowired KafkaTemplate<Integer, String> customTemplate, @Autowired Config config)
 					throws InterruptedException, ExecutionException, TimeoutException {
 
-		template.send("observation.testT1", "test").get(10, TimeUnit.SECONDS);
+		AtomicReference<SimpleSpan> spanFromCallback = new AtomicReference<>();
+
+		template.send("observation.testT1", "test")
+				.thenAccept((sendResult) -> spanFromCallback.set(tracer.currentSpan()))
+				.get(10, TimeUnit.SECONDS);
+
+		assertThat(spanFromCallback.get()).isNotNull();
+
 		assertThat(listener.latch1.await(10, TimeUnit.SECONDS)).isTrue();
 		assertThat(listener.record).isNotNull();
 		Headers headers = listener.record.headers();