From 69dab0ec972f942bca363dd5e29e2aa25d8c9568 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=97=A0=E9=99=90=E8=B6=85=E9=A2=91?= <102959041@qq.com>
Date: Sat, 20 Apr 2019 02:13:11 +0800
Subject: [PATCH] init

---
 mybatis/src/main/java/config/MyBatisConfig.java | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/mybatis/src/main/java/config/MyBatisConfig.java b/mybatis/src/main/java/config/MyBatisConfig.java
index e93da47..a9e5a90 100644
--- a/mybatis/src/main/java/config/MyBatisConfig.java
+++ b/mybatis/src/main/java/config/MyBatisConfig.java
@@ -7,6 +7,7 @@ import model.DataSourceModel;
 import org.apache.commons.io.FileUtils;
 import org.apache.ibatis.session.Configuration;
 import org.mybatis.spring.SqlSessionFactoryBean;
+import org.mybatis.spring.SqlSessionTemplate;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.support.DefaultListableBeanFactory;
 import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
@@ -86,6 +87,10 @@ public class MyBatisConfig extends InitConfig {
             try {
                 if(factoryBean.getObject()!=null){
                     beanFactory.registerSingleton(dataSourceModel.getFactoryBean(), factoryBean.getObject());
+                    SqlSessionTemplate sqlSessionTemplate=new SqlSessionTemplate(factoryBean.getObject());
+                    sqlSessionTemplate.getConfiguration().getMapperRegistry().getMappers().forEach(c->{
+                        beanFactory.registerSingleton(dataSourceModel.getFactoryBean()+c.getSimpleName(),sqlSessionTemplate.getMapper(c));
+                    });
                 }else{
                     LOG.error("初始化数据库配置" + JSON.toJSONString(dataSourceModel) + "失败");
                 }