Browse Source
Add auto-configuration support to export Lettuce latency metrics. See gh-27865pull/27989/head
7 changed files with 142 additions and 1 deletions
@ -0,0 +1,55 @@
@@ -0,0 +1,55 @@
|
||||
/* |
||||
* Copyright 2012-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. |
||||
* 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.actuate.autoconfigure.metrics.redis; |
||||
|
||||
import io.lettuce.core.metrics.MicrometerCommandLatencyRecorder; |
||||
import io.lettuce.core.metrics.MicrometerOptions; |
||||
import io.micrometer.core.instrument.MeterRegistry; |
||||
|
||||
import org.springframework.boot.actuate.autoconfigure.metrics.CompositeMeterRegistryAutoConfiguration; |
||||
import org.springframework.boot.actuate.autoconfigure.metrics.MetricsAutoConfiguration; |
||||
import org.springframework.boot.autoconfigure.AutoConfigureAfter; |
||||
import org.springframework.boot.autoconfigure.AutoConfigureBefore; |
||||
import org.springframework.boot.autoconfigure.condition.ConditionalOnBean; |
||||
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass; |
||||
import org.springframework.boot.autoconfigure.data.redis.ClientResourcesBuilderCustomizer; |
||||
import org.springframework.boot.autoconfigure.data.redis.RedisAutoConfiguration; |
||||
import org.springframework.context.annotation.Bean; |
||||
import org.springframework.context.annotation.Configuration; |
||||
import org.springframework.data.redis.connection.lettuce.LettuceConnectionFactory; |
||||
|
||||
/** |
||||
* Auto-configuration for Lettuce metrics. |
||||
* |
||||
* @author Antonin Arquey |
||||
* @since 2.6.0 |
||||
*/ |
||||
@Configuration(proxyBeanMethods = false) |
||||
@AutoConfigureBefore(RedisAutoConfiguration.class) |
||||
@AutoConfigureAfter({ MetricsAutoConfiguration.class, CompositeMeterRegistryAutoConfiguration.class }) |
||||
@ConditionalOnClass(LettuceConnectionFactory.class) |
||||
@ConditionalOnBean(MeterRegistry.class) |
||||
public class LettuceMetricsAutoConfiguration { |
||||
|
||||
@Bean |
||||
public ClientResourcesBuilderCustomizer lettuceMetrics(MeterRegistry meterRegistry) { |
||||
MicrometerOptions options = MicrometerOptions.builder().histogram(true).build(); |
||||
return (client) -> client.commandLatencyRecorder(new MicrometerCommandLatencyRecorder(meterRegistry, options)) |
||||
.build(); |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,20 @@
@@ -0,0 +1,20 @@
|
||||
/* |
||||
* Copyright 2012-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. |
||||
* 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. |
||||
*/ |
||||
|
||||
/** |
||||
* Auto-configuration for Redis metrics. |
||||
*/ |
||||
package org.springframework.boot.actuate.autoconfigure.metrics.redis; |
||||
@ -0,0 +1,56 @@
@@ -0,0 +1,56 @@
|
||||
/* |
||||
* Copyright 2012-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. |
||||
* 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.actuate.autoconfigure.metrics.redis; |
||||
|
||||
import io.lettuce.core.metrics.MicrometerCommandLatencyRecorder; |
||||
import io.lettuce.core.resource.ClientResources; |
||||
import org.junit.jupiter.api.Test; |
||||
|
||||
import org.springframework.boot.actuate.autoconfigure.metrics.test.MetricsRun; |
||||
import org.springframework.boot.autoconfigure.AutoConfigurations; |
||||
import org.springframework.boot.autoconfigure.data.redis.RedisAutoConfiguration; |
||||
import org.springframework.boot.test.context.runner.ApplicationContextRunner; |
||||
import org.springframework.data.redis.connection.lettuce.LettuceConnectionFactory; |
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat; |
||||
|
||||
class LettuceMetricsAutoConfigurationTests { |
||||
|
||||
private final ApplicationContextRunner contextRunner = new ApplicationContextRunner() |
||||
.withConfiguration(AutoConfigurations.of(LettuceMetricsAutoConfiguration.class)); |
||||
|
||||
@Test |
||||
void whenThereIsAMeterRegistryThenCommandLatencyRecorderIsAdded() { |
||||
this.contextRunner.with(MetricsRun.simple()) |
||||
.withConfiguration(AutoConfigurations.of(RedisAutoConfiguration.class)).run((context) -> { |
||||
ClientResources clientResources = context.getBean(LettuceConnectionFactory.class) |
||||
.getClientResources(); |
||||
assertThat(clientResources.commandLatencyRecorder()).isNotNull() |
||||
.isInstanceOf(MicrometerCommandLatencyRecorder.class); |
||||
}); |
||||
} |
||||
|
||||
@Test |
||||
void whenThereIsNoMeterRegistryThenClientResourcesCustomizationBacksOff() { |
||||
this.contextRunner.withConfiguration(AutoConfigurations.of(RedisAutoConfiguration.class)).run((context) -> { |
||||
ClientResources clientResources = context.getBean(LettuceConnectionFactory.class).getClientResources(); |
||||
assertThat(clientResources.commandLatencyRecorder().isEnabled()) |
||||
.isNotInstanceOf(MicrometerCommandLatencyRecorder.class); |
||||
}); |
||||
} |
||||
|
||||
} |
||||
Loading…
Reference in new issue