Merge pull request #46975 from dungdm93

* pr/46975:
  Polish "Auto-configure observation of Redis with Lettuce"
  Auto-configure observation of Redis with Lettuce

Closes gh-46975

Author: snicoll

documentation/spring-boot-docs/src/docs/antora/modules/reference/pages/actuator/metrics.adoc

@@ -1053,7 +1053,7 @@ Metrics for Jetty's javadoc:org.eclipse.jetty.server.Connector[] instances are b
 [[actuator.metrics.supported.redis]]
 === Redis Metrics
 
-Auto-configuration registers a javadoc:io.lettuce.core.metrics.MicrometerCommandLatencyRecorder[] for the auto-configured javadoc:org.springframework.data.redis.connection.lettuce.LettuceConnectionFactory[].
+Auto-configuration registers a javadoc:io.lettuce.core.tracing.MicrometerTracing[] for the auto-configured javadoc:org.springframework.data.redis.connection.lettuce.LettuceConnectionFactory[].
 For more detail, see the {url-lettuce-docs}/advanced-usage/#observability[Observability section] of the Lettuce documentation.
 
 

module/spring-boot-data-redis/build.gradle

@@ -56,6 +56,7 @@ dependencies {
 
 	testImplementation(project(":core:spring-boot-test"))
 	testImplementation(project(":test-support:spring-boot-test-support"))
+	testImplementation("io.micrometer:micrometer-observation-test")
 	testImplementation("io.projectreactor:reactor-test")
 
 	testRuntimeOnly("ch.qos.logback:logback-classic")

module/spring-boot-data-redis/src/main/java/org/springframework/boot/data/redis/autoconfigure/metrics/LettuceMetricsAutoConfiguration.java renamed to module/spring-boot-data-redis/src/main/java/org/springframework/boot/data/redis/autoconfigure/observation/LettuceObservationAutoConfiguration.java

@@ -14,43 +14,36 @@
  * limitations under the License.
  */
 
-package org.springframework.boot.data.redis.autoconfigure.metrics;
+package org.springframework.boot.data.redis.autoconfigure.observation;
 
 import io.lettuce.core.RedisClient;
-import io.lettuce.core.metrics.MicrometerCommandLatencyRecorder;
-import io.lettuce.core.metrics.MicrometerOptions;
-import io.micrometer.core.instrument.MeterRegistry;
+import io.lettuce.core.tracing.MicrometerTracing;
+import io.micrometer.observation.ObservationRegistry;
 
 import org.springframework.boot.autoconfigure.AutoConfiguration;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnBean;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
-import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
 import org.springframework.boot.data.redis.autoconfigure.ClientResourcesBuilderCustomizer;
 import org.springframework.boot.data.redis.autoconfigure.DataRedisAutoConfiguration;
 import org.springframework.context.annotation.Bean;
 
 /**
- * Auto-configuration for Lettuce metrics.
+ * Auto-configuration for Lettuce observability.
  *
  * @author Antonin Arquey
  * @author Yanming Zhou
+ * @author Dũng Đăng Minh
  * @since 4.0.0
  */
 @AutoConfiguration(before = DataRedisAutoConfiguration.class,
-		afterName = "org.springframework.boot.micrometer.metrics.autoconfigure.CompositeMeterRegistryAutoConfiguration")
-@ConditionalOnClass({ RedisClient.class, MicrometerCommandLatencyRecorder.class, MeterRegistry.class })
-@ConditionalOnBean(MeterRegistry.class)
-public final class LettuceMetricsAutoConfiguration {
+		afterName = "org.springframework.boot.micrometer.observation.autoconfigure.ObservationAutoConfiguration")
+@ConditionalOnClass({ RedisClient.class, MicrometerTracing.class, ObservationRegistry.class })
+@ConditionalOnBean(ObservationRegistry.class)
+public final class LettuceObservationAutoConfiguration {
 
 	@Bean
-	@ConditionalOnMissingBean
-	MicrometerOptions micrometerOptions() {
-		return MicrometerOptions.create();
-	}
-
-	@Bean
-	ClientResourcesBuilderCustomizer lettuceMetrics(MeterRegistry meterRegistry, MicrometerOptions options) {
-		return (client) -> client.commandLatencyRecorder(new MicrometerCommandLatencyRecorder(meterRegistry, options));
+	ClientResourcesBuilderCustomizer lettuceObservation(ObservationRegistry observationRegistry) {
+		return (client) -> client.tracing(new MicrometerTracing(observationRegistry, "Redis"));
 	}
 
 }

module/spring-boot-data-redis/src/main/java/org/springframework/boot/data/redis/autoconfigure/metrics/package-info.java renamed to module/spring-boot-data-redis/src/main/java/org/springframework/boot/data/redis/autoconfigure/observation/package-info.java

@@ -15,9 +15,9 @@
  */
 
 /**
- * Auto-configuration for Spring Data Redis metrics.
+ * Auto-configuration for Spring Data Redis observation.
  */
 @NullMarked
-package org.springframework.boot.data.redis.autoconfigure.metrics;
+package org.springframework.boot.data.redis.autoconfigure.observation;
 
 import org.jspecify.annotations.NullMarked;

module/spring-boot-data-redis/src/main/resources/META-INF/spring/org.springframework.boot.autoconfigure.AutoConfiguration.imports

@@ -3,4 +3,4 @@ org.springframework.boot.data.redis.autoconfigure.DataRedisReactiveAutoConfigura
 org.springframework.boot.data.redis.autoconfigure.DataRedisRepositoriesAutoConfiguration
 org.springframework.boot.data.redis.autoconfigure.health.DataRedisHealthContributorAutoConfiguration
 org.springframework.boot.data.redis.autoconfigure.health.DataRedisReactiveHealthContributorAutoConfiguration
-org.springframework.boot.data.redis.autoconfigure.metrics.LettuceMetricsAutoConfiguration
+org.springframework.boot.data.redis.autoconfigure.observation.LettuceObservationAutoConfiguration

module/spring-boot-data-redis/src/test/java/org/springframework/boot/data/redis/autoconfigure/metrics/LettuceMetricsAutoConfigurationTests.java

@@ -0,0 +1,61 @@
+/*
+ * Copyright 2012-present 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.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.springframework.boot.data.redis.autoconfigure.observation;
+
+import io.lettuce.core.resource.ClientResources;
+import io.lettuce.core.tracing.MicrometerTracing;
+import io.micrometer.observation.tck.TestObservationRegistry;
+import org.junit.jupiter.api.Test;
+
+import org.springframework.boot.autoconfigure.AutoConfigurations;
+import org.springframework.boot.data.redis.autoconfigure.DataRedisAutoConfiguration;
+import org.springframework.boot.test.context.runner.ApplicationContextRunner;
+import org.springframework.data.redis.connection.lettuce.LettuceConnectionFactory;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+/**
+ * Tests for {@link LettuceObservationAutoConfiguration}.
+ *
+ * @author Antonin Arquey
+ * @author Stephane Nicoll
+ */
+class LettuceObservationAutoConfigurationTests {
+
+	private final ApplicationContextRunner contextRunner = new ApplicationContextRunner()
+		.withConfiguration(AutoConfigurations.of(LettuceObservationAutoConfiguration.class));
+
+	@Test
+	void whenThereIsAnObservationRegistryThenMicrometerTracingIsAdded() {
+		this.contextRunner.withBean(TestObservationRegistry.class, TestObservationRegistry::create)
+			.withConfiguration(AutoConfigurations.of(DataRedisAutoConfiguration.class))
+			.run((context) -> {
+				ClientResources clientResources = context.getBean(LettuceConnectionFactory.class).getClientResources();
+				assertThat(clientResources.tracing()).isInstanceOf(MicrometerTracing.class);
+			});
+
+	}
+
+	@Test
+	void whenThereIsNoObservationRegistryThenClientResourcesCustomizationBacksOff() {
+		this.contextRunner.withConfiguration(AutoConfigurations.of(DataRedisAutoConfiguration.class)).run((context) -> {
+			ClientResources clientResources = context.getBean(LettuceConnectionFactory.class).getClientResources();
+			assertThat(clientResources.tracing()).isNotInstanceOf(MicrometerTracing.class);
+		});
+	}
+
+}