diff --git a/src/main/java/com/chinaunicom/mall/ebtp/extend/ExtendDataSourceHealthContributorAutoConfiguration.java b/src/main/java/com/chinaunicom/mall/ebtp/extend/ExtendDataSourceHealthContributorAutoConfiguration.java new file mode 100644 index 0000000..ebe5c0d --- /dev/null +++ b/src/main/java/com/chinaunicom/mall/ebtp/extend/ExtendDataSourceHealthContributorAutoConfiguration.java @@ -0,0 +1,27 @@ +package com.chinaunicom.mall.ebtp.extend; + +import java.util.Map; + +import javax.sql.DataSource; + +import org.springframework.beans.factory.ObjectProvider; +import org.springframework.boot.actuate.autoconfigure.jdbc.DataSourceHealthContributorAutoConfiguration; +import org.springframework.boot.actuate.health.AbstractHealthIndicator; +import org.springframework.boot.actuate.jdbc.DataSourceHealthIndicator; +import org.springframework.boot.jdbc.metadata.DataSourcePoolMetadataProvider; +import org.springframework.context.annotation.Configuration; + +@Configuration +public class ExtendDataSourceHealthContributorAutoConfiguration extends DataSourceHealthContributorAutoConfiguration { + + public ExtendDataSourceHealthContributorAutoConfiguration(Map dataSources, + ObjectProvider metadataProviders) { + super(dataSources, metadataProviders); + } + + @Override + protected AbstractHealthIndicator createIndicator(DataSource source) { + return new DataSourceHealthIndicator(source, "select 1"); + } + +}