SpringBoot1.5.10+HikariCP+Dropwizard-Metrics统计连接池使用情况

SpringBoot+HikariCP+Dropwizard-Metrics统计连接池使用情况

背景，HikariCP是Java目前使用最广的连接池工具类，SpringBoot默认也是用这个，现在想获取连接池使用情况。

这里假设SpringBoot已集成HikariCP
1.pom.xml加上Dropwizard-Metrics配置

		<dependency>
			<groupId>io.dropwizard.metrics</groupId>
			<artifactId>metrics-core</artifactId>
		</dependency>
		<dependency>
			<groupId>io.dropwizard.metrics</groupId>
			<artifactId>metrics-healthchecks</artifactId>
		</dependency>

2在应用启动的时候连接池注册统计接口

import com.codahale.metrics.MetricRegistry;  
import com.codahale.metrics.Slf4jReporter;  
import com.zaxxer.hikari.HikariDataSource;  
  
import javax.sql.DataSource;  
import org.slf4j.Logger;  
import org.slf4j.LoggerFactory;  
@Component  
public class ApplicationRunner implements ApplicationRunner {  
	private static final Logger LOGGER = LoggerFactory.getLogger(ApplicationRunner.class);  
	@Autowired  
    private DataSource dataSource;  
  
	@Override  
    public void run(ApplicationArguments args) throws Exception {  
		try {  
            // see detail https://github.com/brettwooldridge/HikariCP/wiki/Dropwizard-Metrics  
            // 连接池注册统计接口  
            MetricRegistry metricRegistry = new MetricRegistry();  
            if(dataSource instanceof HikariDataSource) {  
                ((HikariDataSource) dataSource).setMetricRegistry(metricRegistry);  
            }  
            // 定时打印连接池使用情况  
            Slf4jReporter reporter = Slf4jReporter.forRegistry(metricRegistry).build();  
            reporter.start(1, TimeUnit.MINUTES);  
  
		} catch (Exception e) {  
            String msg = "服务启动异常";  
            LOGGER.error(msg, e);  
            throw new IllegalStateException(msg, e);  
        }   
  
  
	}  
}

3 提供http请求获取连接池使用情况

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.http.client.SimpleClientHttpRequestFactory;
import org.springframework.web.client.RestTemplate;


/**
 * 远程使用的RestTemplatebean，与服务间调用区分开来
 */
@Configuration
public class RestTemplateConfig {

    @Bean("remoteRestTemplate")
    public RestTemplate remoteRestTemplate() {
        SimpleClientHttpRequestFactory factory = new SimpleClientHttpRequestFactory();
        factory.setReadTimeout(180000);
        factory.setConnectTimeout(8000);
        return new RestTemplate(factory);
    }

}

import com.codahale.metrics.*;
import com.netflix.appinfo.InstanceInfo;
import com.zaxxer.hikari.HikariDataSource;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.cloud.client.ServiceInstance;
import org.springframework.cloud.client.discovery.DiscoveryClient;
import org.springframework.cloud.netflix.eureka.EurekaDiscoveryClient;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.client.RestTemplate;

import javax.sql.DataSource;
import java.util.List;
import java.util.Map;
import java.util.SortedMap;
import java.util.TreeMap;


@RestController
@RequestMapping("hikariCpStatsController")
public class HikariCpStatsController {
private static final String SERVICE_NAME ="xiaoniu";
	@Autowired
	private DataSource dataSource;

	@Autowired
	private DiscoveryClient discoveryClient;

	@Qualifier("remoteRestTemplate")
	@Autowired
	private RestTemplate restTemplate;

	@GetMapping("poolStatsHa")
	public TreeMap<String, Object> poolStatsHa() {
		MetricRegistry metricRegistry = null;
		if (dataSource instanceof HikariDataSource) {
			metricRegistry = (MetricRegistry) ((HikariDataSource) dataSource).getMetricRegistry();
		}
		if (metricRegistry == null)
			return null;
		TreeMap<String, Object> dataAll = new TreeMap<>();
		// 节点的信息
		List<ServiceInstance> instances = discoveryClient.getInstances(SERVICE_NAME);
		for (ServiceInstance instance : instances) {
			InstanceInfo instanceInfo = ((EurekaDiscoveryClient.EurekaServiceInstance) instance).getInstanceInfo();
			TreeMap data = restTemplate.getForObject(instance.getUri().toString() + "/hikariCpStatsController/poolStats", TreeMap.class);
			dataAll.put(instanceInfo.getInstanceId(), data);
		}
		return dataAll;
	}

	@GetMapping("poolStats")
	public TreeMap<String, Object> poolStats() {
		MetricRegistry metricRegistry = null;
		if (dataSource instanceof HikariDataSource) {
			metricRegistry = (MetricRegistry) ((HikariDataSource) dataSource).getMetricRegistry();
		}
		if (metricRegistry == null)
			return null;

		TreeMap<String, Object> data = new TreeMap<>();

		SortedMap<String, Gauge> gauges = metricRegistry.getGauges();
		for (Map.Entry<String, Gauge> gaugeEntry : gauges.entrySet()) {
			String key = gaugeEntry.getKey();
			Gauge value = gaugeEntry.getValue();
			data.put(key, value.getValue());

		}
		SortedMap<String, Timer> timers = metricRegistry.getTimers();
		for (Map.Entry<String, Timer> timerEntry : timers.entrySet()) {
			String key = timerEntry.getKey();
			Timer value = timerEntry.getValue();
			data.put(key, "获取连接时99%线程等待的纳秒=" + value.getSnapshot().get99thPercentile());
		}
		SortedMap<String, Meter> meters = metricRegistry.getMeters();
		for (Map.Entry<String, Meter> meterEntry : meters.entrySet()) {
			String key = meterEntry.getKey();
			Meter value = meterEntry.getValue();
			data.put(key, "count=" + value.getCount());
		}
		SortedMap<String, Histogram> histograms = metricRegistry.getHistograms();
		for (Map.Entry<String, Histogram> histogramEntry : histograms.entrySet()) {
			String key = histogramEntry.getKey();
			Histogram value = histogramEntry.getValue();
			data.put(key, "99%连接线程使用的毫秒=" + value.getSnapshot().get99thPercentile());
		}
		return data;
	}
}

在此大功告成

参考// see detail https://github.com/brettwooldridge/HikariCP/wiki/Dropwizard-Metrics