From 6e8c95d31d85b53857e4e3ce12ea303eb0ea42eb Mon Sep 17 00:00:00 2001
From: java110 <928255095@qq.com>
Date: 星期四, 19 八月 2021 00:47:48 +0800
Subject: [PATCH] 优化代码

---
 java110-db/src/main/java/com/java110/db/MyBatisConfig.java |   42 ++++++++++++++++++++++++++++++++----------
 1 files changed, 32 insertions(+), 10 deletions(-)

diff --git a/java110-db/src/main/java/com/java110/db/MyBatisConfig.java b/java110-db/src/main/java/com/java110/db/MyBatisConfig.java
old mode 100644
new mode 100755
index fc67a1f..e153d3a
--- a/java110-db/src/main/java/com/java110/db/MyBatisConfig.java
+++ b/java110-db/src/main/java/com/java110/db/MyBatisConfig.java
@@ -1,13 +1,14 @@
 package com.java110.db;
 
-import com.alibaba.druid.pool.DruidDataSource;
+import com.java110.config.properties.code.Java110Properties;
+import org.apache.ibatis.plugin.Interceptor;
 import org.apache.ibatis.session.SqlSessionFactory;
 import org.mybatis.spring.SqlSessionFactoryBean;
 import org.mybatis.spring.SqlSessionTemplate;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.boot.context.properties.ConfigurationProperties;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
+import org.springframework.core.io.Resource;
 import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
 import org.springframework.core.io.support.ResourcePatternResolver;
 import org.springframework.jdbc.datasource.DataSourceTransactionManager;
@@ -16,6 +17,9 @@
 import org.springframework.transaction.annotation.TransactionManagementConfigurer;
 
 import javax.sql.DataSource;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
 
 /**
  * MyBatis鍩虹閰嶇疆
@@ -28,14 +32,17 @@
 public class MyBatisConfig implements TransactionManagementConfigurer {
 
     @Autowired
-    DataSource dataSource;
+    private Java110Properties java110Properties;
 
-    @Bean(name = "dataSource")
-    @ConfigurationProperties(prefix = "spring.datasource")
-    public DataSource dataSource() {
-        DruidDataSource druidDataSource = new DruidDataSource();
-        return druidDataSource;
-    }
+    @Autowired
+    DataSource dataSource;
+//
+//    @Bean(name = "dataSource")
+//    @ConfigurationProperties(prefix = "spring.datasource")
+//    public DataSource dataSource() {
+//        DruidDataSource druidDataSource = new DruidDataSource();
+//        return druidDataSource;
+//    }
 
 
     @Bean(name = "sqlSessionFactory")
@@ -43,11 +50,19 @@
         SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
         bean.setDataSource(dataSource);
         bean.setTypeAliasesPackage("tk.mybatis.springboot.model");
+        bean.setPlugins(new Interceptor[]{new Java110MybatisInterceptor()});
 
         //娣诲姞XML鐩綍
         ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
         try {
-            bean.setMapperLocations(resolver.getResources("classpath:mapper/*/*.xml"));
+            // bean.setMapperLocations(resolver.getResources("classpath:mapper/*/*.xml"));
+            Resource[] resources = null;
+            List<Resource> resourceList = new ArrayList<Resource>();
+            for (String path : java110Properties.getMappingPath().split(",")) {
+                resources = resolver.getResources(path);
+                resourceList.addAll(Arrays.asList(resources));
+            }
+            bean.setMapperLocations(resourceList.toArray(new Resource[resourceList.size()]));
             return bean.getObject();
         } catch (Exception e) {
             e.printStackTrace();
@@ -67,4 +82,11 @@
     }
 
 
+    public Java110Properties getJava110Properties() {
+        return java110Properties;
+    }
+
+    public void setJava110Properties(Java110Properties java110Properties) {
+        this.java110Properties = java110Properties;
+    }
 }

--
Gitblit v1.8.0