Mybatis中自定义实例化SqlSessionFactoryBean

朴灿烈づ我的快乐病毒、 2021-11-30 00:20 279阅读 0赞

现在SpringBoot基本成为开发的标配，如果你上司让你搭建一个SpringBoot，然后集成Mybatis+Druid，你可以能百度几下，卡卡就搭建完毕了。

> 现在项目基本都会使用连接池技术，市面上的连接池有很多，比如：DBCP、c3p0、Druid等，今天我们重点介绍Druid连接池。

application.yml配置文件如下所示：

spring:
      #数据库配置
      datasource:
        druid:
          type: com.alibaba.druid.pool.DruidDataSource
          url: jdbc:mysql://127.0.0.1:3306/test?useUnicode=true&characterEncoding=UTF-8&zeroDateTimeBehavior=round&serverTimezone=GMT%2B8
          username: test
          password: test
          driver-class-name: com.mysql.jdbc.Driver
    #     获取连接时最大等待时间，单位毫秒
          max-wait: 60000
    #      最大连接池数量
          max-active: 80
    #      初始化时建立物理连接的个数
          initial-size: 20
    #      最小连接池数量
          min-idle: 40
    #Destory线程中如果检测到当前连接的最后活跃时间和当前时间的差值大于minEvictableIdleTimeMillis，则关闭当前连接。
          min-evictable-idle-time-millis: 600000
    #      testWhileIdle的判断依据，详细看testWhileIdle属性的说明
          time-between-eviction-runs-millis: 2000
    #用来检测连接是否有效的sql，要求是一个查询语句。
          validation-query: select 1
    #      申请连接的时候检测，如果空闲时间大于timeBetweenEvictionRunsMillis，执行validationQuery检测连接是否有效。
          test-while-idle: true
    #      申请连接时执行validationQuery检测连接是否有效，做了这个配置会降低性能。
          test-on-borrow: false
    #      归还连接时执行validationQuery检测连接是否有效，做了这个配置会降低性能
          test-on-return: false
    #      属性类型是字符串，通过别名的方式配置扩展插件
          filters: stat,wall
    #      开启慢sql，并设置时间
          filter:
            stat.log-slow-sql: true
            stat.slow-sql-millis: 2000
          web-stat-filter:
            enabled: true
            url-pattern: /*
          stat-view-servlet:
            enabled: true
            url-pattern: /druid/*

使用SpringBoot作为项目框架自然简单，没有很多的xml配置文件，不需要配置额外的tomcat，不管是开发还是部署都非常方便。但高度集成有时候也会带来一些麻烦。比如你上司要求你在mybatis中集成插件并可以识别common模块的mybatis.xml映射文件。

这个时候你可能首先会百度如何配置mybatis插件、如何配置多模块的mapper-locations，然后有很多博文会说在SqlSessionFactoryBean设置即可。你可能会这么设置：

@Autowired
    private SqlSessionFactoryBean sqlSessionFactoryBean;

但是结果不是那么尽人意，初始化的结果为null。这是因为高版本的mybatis实现机制做了一些修改，我们没办法通过@Autowired来实例化SqlSessionFactoryBean对象。所以我们必须自己来实例化SqlSessionFactoryBean对象，而实例化SqlSessionFactoryBean对象的关键就是设置DataSource数据源。我们可以通过如下代码来实例化过SqlSessionFactoryBean。

/**
     * mybatis配置
     * @author linzhiqinag
     */
    
    @Configuration
    public class MybatisConfig {
        private Logger logger = LoggerFactory.getLogger(MybatisConfig.class);
    
        @Value("${mybatis.mapper-locations}")
        private String mapperLocation;
    
        @Value("${common-mybatis.mapper-locations}")
        private String commonMapperLocation;
    
        @Value("${spring.datasource.druid.username}")
        private String username;
    
        @Value("${spring.datasource.druid.password}")
        private String password;
    
        @Value("${spring.datasource.druid.url}")
        private String dbUrl;
    
        @Value("${spring.datasource.druid.initial-size}")
        private int initialSize;
    
        @Value("${spring.datasource.druid.min-idle}")
        private int minIdle;
    
        @Value("${spring.datasource.druid.max-active}")
        private int maxActive;
    
        @Value("${spring.datasource.druid.max-wait}")
        private long maxWait;
    
        @Value("${spring.datasource.druid.driver-class-name}")
        private String driverClassName;
    
        @Value("${spring.datasource.druid.min-evictable-idle-time-millis}")
        private long minEvictableIdleTimeMillis;
    
        @Value("${spring.datasource.druid.time-between-eviction-runs-millis}")
        private long timeBetweenEvictionRunsMillis;
    
        @Value("${spring.datasource.druid.validation-query}")
        private String validationQuery;
    
        @Value("${spring.datasource.druid.test-while-idle}")
        private boolean testWhileIdle;
    
        @Value("${spring.datasource.druid.test-on-borrow}")
        private boolean testOnBorrow;
    
        @Value("${spring.datasource.druid.test-on-return}")
        private boolean testOnReturn;
    
        @Value("${spring.datasource.druid.filter.stat.log-slow-sql}")
        private boolean logSlowSql;
    
        @Value("${spring.datasource.druid.filter.stat.slow-sql-millis}")
        private long slowSqlMillis;
    
        @Bean
        public DruidDataSource dataSource() {
            DruidDataSource druidDataSource = new DruidDataSource();
            try {
                druidDataSource.setUsername(username);
                druidDataSource.setPassword(password);
                druidDataSource.setUrl(dbUrl);
                druidDataSource.setFilters("stat,wall");
                druidDataSource.setInitialSize(initialSize);
                druidDataSource.setMinIdle(minIdle);
                druidDataSource.setMaxActive(maxActive);
                druidDataSource.setMaxWait(maxWait);
                druidDataSource.setMinEvictableIdleTimeMillis(minEvictableIdleTimeMillis);
                druidDataSource.setTimeBetweenEvictionRunsMillis(timeBetweenEvictionRunsMillis);
                druidDataSource.setUseGlobalDataSourceStat(true);
                druidDataSource.setDriverClassName(driverClassName);
                druidDataSource.setValidationQuery(validationQuery);
                druidDataSource.setTestWhileIdle(testWhileIdle);
                druidDataSource.setTestOnBorrow(testOnBorrow);
                druidDataSource.setTestOnReturn(testOnReturn);
                // 设置需要的过滤
                List<Filter> statFilters =new ArrayList<>();
                StatFilter statFilter = new StatFilter();
                statFilter.setLogSlowSql(logSlowSql);
                statFilter.setSlowSqlMillis(slowSqlMillis);
                statFilters.add(statFilter);
                // 设置慢SQL
                druidDataSource.setProxyFilters(statFilters);
             } catch (Exception e) {
                e.printStackTrace();
            }
            return druidDataSource;
        }
    
        @Bean
        public SqlSessionFactoryBean mysqlSessionFactory(DataSource dataSource) throws Exception {
            SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
            sqlSessionFactoryBean.setDataSource(dataSource);
            PathMatchingResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
            Resource[] resources1 = resolver.getResources(mapperLocation);
            Resource[] resources2 = resolver.getResources(commonMapperLocation);
            Resource[] resources = new Resource[resources1.length+resources2.length];
            for (int i=0;i<resources1.length;i++) {
                resources[i] = resources1[i];
            }
            int initSize = resources1.length;
            for (int i=0;i<resources2.length;i++) {
                resources[initSize+i] = resources2[i];
            }
            sqlSessionFactoryBean.setMapperLocations(resources);
            sqlSessionFactoryBean.setPlugins(new Interceptor[]{new CatMybatisInterceptor(dbUrl)});
            return sqlSessionFactoryBean;
        }
    }

这样我们就可以得到SqlSessionFactoryBean对象了，然后我们就可以通过sqlSessionFactoryBean.setMapperLocations()来设置多模块xml映射，通过sqlSessionFactoryBean.setPlugins()来设置指定的插件了。

> 注意：
> 
> 这边需要注意的是，如果采用代码的方式实例化SqlSessionFactoryBean，那关于数据库相关的配置将会失效，所以在设置数据源的时候一定要设置全。

会使用框架不代表就是架构师，千万不要觉得会搭建springboot、spring cloud就是一个很牛逼的架构师了。这只是万里长征的第一步，你需要学习的技能还非常多。希望大家使用框架的时候，也要学着去深入了解它的实现原理，当然这不是为了重复造轮子，而是为了你能更加的了解它，并真正的掌握它。

想要更多干货、技术猛料的孩子，快点拿起手机扫码关注我，我在这里等你哦~

![aHR0cHM6Ly9tbWJpei5xcGljLmNuL21tYml6X2pwZy9zdTUxaWJiaWF5dnhzZEkwODVpY1JNWWljTDVFaWIwMVkxZ0drcTVNdWxMdXJ6T0QzU251V2oydzA5cjgwSnBTdm5xSTJtV2g0QUJ5RmhJTnZ1bHhwS2I2b1JBLzY0MD93eF9mbXQ9anBlZw][]

**林老师带你学编程**：[https://wolzq.com][https_wolzq.com]

[aHR0cHM6Ly9tbWJpei5xcGljLmNuL21tYml6X2pwZy9zdTUxaWJiaWF5dnhzZEkwODVpY1JNWWljTDVFaWIwMVkxZ0drcTVNdWxMdXJ6T0QzU251V2oydzA5cjgwSnBTdm5xSTJtV2g0QUJ5RmhJTnZ1bHhwS2I2b1JBLzY0MD93eF9mbXQ9anBlZw]: /images/20211129/3cf5faa2588b44b9b7841249bcd2a224.png
[https_wolzq.com]: https://wolzq.com/