Mybatis中自定義實例化SqlSessionFactoryBean

現在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就是一個很牛逼的架構師了。這只是萬里長征的第一步，你需要學習的技能還非常多。希望大家使用框架的時候，也要學着去深入瞭解它的實現原理，當然這不是爲了重複造輪子，而是爲了你能更加的瞭解它，並真正的掌握它。

想要更多幹貨、技術猛料的孩子，快點拿起手機掃碼關注我，我在這裏等你哦~

林老師帶你學編程：https://wolzq.com