3 changed files with 32 additions and 91 deletions
@ -1,84 +0,0 @@
@@ -1,84 +0,0 @@
|
||||
/* |
||||
* Copyright 2012-2018 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 |
||||
* |
||||
* http://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.jdbc; |
||||
|
||||
import com.zaxxer.hikari.HikariDataSource; |
||||
import com.zaxxer.hikari.metrics.micrometer.MicrometerMetricsTrackerFactory; |
||||
import io.micrometer.core.instrument.MeterRegistry; |
||||
import org.apache.commons.logging.Log; |
||||
import org.apache.commons.logging.LogFactory; |
||||
|
||||
import org.springframework.beans.factory.config.BeanPostProcessor; |
||||
import org.springframework.context.ApplicationContext; |
||||
import org.springframework.core.Ordered; |
||||
|
||||
/** |
||||
* {@link BeanPostProcessor} that configures Hikari metrics. Such arrangement is necessary |
||||
* because a {@link HikariDataSource} instance cannot be modified once its configuration |
||||
* has completed. |
||||
* |
||||
* @author Stephane Nicoll |
||||
*/ |
||||
class HikariDataSourceMetricsPostProcessor implements BeanPostProcessor, Ordered { |
||||
|
||||
private static final Log logger = LogFactory |
||||
.getLog(HikariDataSourceMetricsPostProcessor.class); |
||||
|
||||
private final ApplicationContext context; |
||||
|
||||
private volatile MeterRegistry meterRegistry; |
||||
|
||||
HikariDataSourceMetricsPostProcessor(ApplicationContext context) { |
||||
this.context = context; |
||||
} |
||||
|
||||
@Override |
||||
public Object postProcessAfterInitialization(Object bean, String beanName) { |
||||
if (bean instanceof HikariDataSource) { |
||||
bindMetricsRegistryToHikariDataSource(getMeterRegistry(), |
||||
(HikariDataSource) bean); |
||||
} |
||||
return bean; |
||||
} |
||||
|
||||
private void bindMetricsRegistryToHikariDataSource(MeterRegistry registry, |
||||
HikariDataSource dataSource) { |
||||
if (dataSource.getMetricRegistry() == null |
||||
&& dataSource.getMetricsTrackerFactory() == null) { |
||||
try { |
||||
dataSource.setMetricsTrackerFactory( |
||||
new MicrometerMetricsTrackerFactory(registry)); |
||||
} |
||||
catch (Exception ex) { |
||||
logger.warn("Failed to bind Hikari metrics: " + ex.getMessage()); |
||||
} |
||||
} |
||||
} |
||||
|
||||
private MeterRegistry getMeterRegistry() { |
||||
if (this.meterRegistry == null) { |
||||
this.meterRegistry = this.context.getBean(MeterRegistry.class); |
||||
} |
||||
return this.meterRegistry; |
||||
} |
||||
|
||||
@Override |
||||
public int getOrder() { |
||||
return Ordered.HIGHEST_PRECEDENCE; |
||||
} |
||||
|
||||
} |
||||
Loading…
Reference in new issue