SpringBoot自定义方式整合Druid数据源

SpringBoot自定义方式整合Druid数据源

1、导入Druid数据源

导入依赖：

<dependency>
    <groupId>com.alibaba</groupId>
    <artifactId>druid</artifactId>
    <version>1.2.6</version>
</dependency>

然后我们看他默认给我们导入的数据源HikariDataSource：
matchIfMissing=true意思是：如果我们没有配置spring.datasource.type的话，那么就默认配置com.zaxxer.hikari.HikariDataSource为我们的数据源。
而这个数据源的导入呢，有两个前提，一个是要有HikariDataSource类，另一个是没有返回DataSource的Bean。

所以，我们要想把他替换掉，就是写一个返回DataSource的Bean就可以了。（说一下，DruidDataSource的爸爸DruidAbstractDataSource类实现了DataSource接口，因此，DruidDataSource也间接地实现了DataSource接口，算起来应该能算他的外孙吧。可以用多态的思想，让DataSource装一下他的外孙，返回DruidDataSource，能懂吧，相当于使用 DataSource source = new DruidDataSource();）
来，我们写我们的配置类：

package com.example.boot.config;

import com.alibaba.druid.pool.DruidDataSource;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.sql.DataSource;

@Configuration
public class MyDataSourceConfig {
    /**
     *  @ConfigurationProperties 注解让这个Bean与配置文件进行绑定
     *  具体绑定到前缀为spring.datasource下的属性
     *  简单说，spring.datasource配了啥，他的setter就跟着配啥
     */
    @ConfigurationProperties("spring.datasource")
    @Bean
    public DataSource dataSource() throws SQLException {
        DruidDataSource dataSource = new DruidDataSource();
        return dataSource;
    }
}

我们可以写一个测试方法来测试一下：

package com.example.boot;

import org.junit.jupiter.api.Test;
import org.springframework.boot.test.context.SpringBootTest;

import javax.annotation.Resource;
import javax.sql.DataSource;

@SpringBootTest
class SpringbootMysqlApplicationTests {
    @Resource
    DataSource dataSource;

    @Test
    void testDataSource(){
        System.out.println("数据源的类型："+dataSource.getClass());
    }
}

运行结果：

2、配置Druid的监控页，开启SQL监控

首先，我们看看官网在web.xml下是怎么配的。

其实，这么写还不够，我们还得给他的属性filter添加一个字符串：stat
所以，我们修改我们的代码为：

package com.example.boot.config;

import com.alibaba.druid.pool.DruidDataSource;
import com.alibaba.druid.support.http.StatViewServlet;
import com.alibaba.druid.support.http.WebStatFilter;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.boot.web.servlet.ServletRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.sql.DataSource;
import java.sql.SQLException;
import java.util.Arrays;

@Configuration
public class MyDataSourceConfig {
    /**
     *  @ConfigurationProperties 注解让这个Bean与配置文件进行绑定
     *  具体绑定到前缀为spring.datasource下的属性
     *  简单说，spring.datasource配了啥，他的setter就跟着配啥
     */
    @ConfigurationProperties("spring.datasource")
    @Bean
    public DataSource dataSource() throws SQLException {
        DruidDataSource dataSource = new DruidDataSource();
        //开启监控功能stat
        dataSource.setFilters("stat");
        return dataSource;
    }

    //配置Druid的监控页功能
    @Bean
    public ServletRegistrationBean statViewServlet(){
        StatViewServlet servlet = new StatViewServlet();
        ServletRegistrationBean<StatViewServlet> servletRegistrationBean = new ServletRegistrationBean<>(servlet, "/druid/*");
        return servletRegistrationBean;
    }
}

编译一下，然后我们可以打开我们Druid的监控页，网址localhost:8080/druid。

然后，我们再测试一下使用jdbc的方式获取一下信息：

package com.example.boot.controller;

import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.annotation.Resource;

@RestController
public class JdbcController {
    @Resource
    JdbcTemplate jdbcTemplate;

    @GetMapping("/getCount")
    public int getCount(){
        String sql = "select count(*) from student";
        Integer number = jdbcTemplate.queryForObject(sql, Integer.class);
        return number;
    }
}

我们再看看Druid的监控：
他告诉了我们，那个网页我刷新了10次，执行时间用了33秒，最慢的一次用了15秒。读取的行数有10行。（为什么读取了10行呢，可能是因为之前我删掉了部分记录，他其实内部没有真的删掉把，然后Druid能读到，就是我们mysql后台自己看不到）

3、开启防火墙

如果我们要开启他的防火墙的话：（官方在web.xml中是这么配置的）

所以，我们在Springboot中就这么写：

    @Bean
    public DataSource dataSource() throws SQLException {
        DruidDataSource dataSource = new DruidDataSource();
        //开启监控功能stat和防火墙wall
        dataSource.setFilters("stat,wall");
        return dataSource;
    }

刷新http://localhost:8080/getCount页面，可以看到防火墙的信息如下：

4、开启Web应用模块

如果要开启他的web应用，官方给出的在web.xml中的配置如下：

在SpringBoot中呢，我们就得再写一个Bean：

    @Bean
    public FilterRegistrationBean webStatFilter(){
        WebStatFilter webStatFilter = new WebStatFilter();
        FilterRegistrationBean<WebStatFilter> filterRegistrationBean = new FilterRegistrationBean(webStatFilter);
        //拦截所有的请求
        filterRegistrationBean.setUrlPatterns(Arrays.asList("/*"));
        //除了这几个请求
        filterRegistrationBean.addInitParameter("exclusions","*.js,*.gif,*.jpg,*.png,*.css,*.ico,/druid/*");
        return filterRegistrationBean;
    }

刷新几次http://localhost:8080/getCount请求。

5、配置监控页面的账号密码

对于监控页面的访问，我们也不能让别人随随便便就登录进去，因此，我们还可以设置登录的账号、密码等信息。
首先可以看出来，这是我们的Servlet，而且，刚才已经配过了，现在要加账号密码，只需要加两个初始化参数（init-param）即可。一个叫loginUsername，一个叫loginPasword，当然了，具体的账号密码信息我们是可以自己配置的。

所以，我们修改刚才Servlet的那个Bean如下：

    //配置Druid的监控页功能
    @Bean
    public ServletRegistrationBean statViewServlet(){
        StatViewServlet servlet = new StatViewServlet();
        ServletRegistrationBean<StatViewServlet> servletRegistrationBean = new ServletRegistrationBean<>(servlet, "/druid/*");
        //添加账号密码
        servletRegistrationBean.addInitParameter("loginUsername","Keeling");
        servletRegistrationBean.addInitParameter("loginPassword","123456");
        return servletRegistrationBean;
    }

编译后，我们再进入，可以看到需要我们的账号密码才行了：