From 2aeb8144a147abbb8af2bd76b164fbe0fb09c326 Mon Sep 17 00:00:00 2001
From: zhangping17 <zhangping17@xiaomi.com>
Date: Wed, 13 Sep 2023 16:52:27 +0800
Subject: [PATCH] fixes #712

---
 .../docean-plugin/docean-plugin-mybatis/pom.xml   |  2 +-
 .../docean/plugin/mybatis/MybatisPlugin.java      | 15 +++++++++++++++
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/jcommon/docean-plugin/docean-plugin-mybatis/pom.xml b/jcommon/docean-plugin/docean-plugin-mybatis/pom.xml
index c1a775cf4..c361e0e97 100644
--- a/jcommon/docean-plugin/docean-plugin-mybatis/pom.xml
+++ b/jcommon/docean-plugin/docean-plugin-mybatis/pom.xml
@@ -25,7 +25,7 @@
     </parent>
     <modelVersion>4.0.0</modelVersion>
     <artifactId>docean-plugin-mybatis</artifactId>
-    <version>1.4.7-SNAPSHOT</version>
+    <version>1.5.0-SNAPSHOT</version>
 
     <dependencies>
 
diff --git a/jcommon/docean-plugin/docean-plugin-mybatis/src/main/java/com/xiaomi/youpin/docean/plugin/mybatis/MybatisPlugin.java b/jcommon/docean-plugin/docean-plugin-mybatis/src/main/java/com/xiaomi/youpin/docean/plugin/mybatis/MybatisPlugin.java
index 25e03d5bb..ed1a2b14d 100644
--- a/jcommon/docean-plugin/docean-plugin-mybatis/src/main/java/com/xiaomi/youpin/docean/plugin/mybatis/MybatisPlugin.java
+++ b/jcommon/docean-plugin/docean-plugin-mybatis/src/main/java/com/xiaomi/youpin/docean/plugin/mybatis/MybatisPlugin.java
@@ -66,6 +66,10 @@ public class MybatisPlugin implements IPlugin {
 
     private boolean serverLess = false;
 
+    private static SqlSessionFactory sqlSessionFactory;
+
+    private static String mapperName = "_mapperName_";
+
 
     @SneakyThrows
     @Override
@@ -118,6 +122,7 @@ private void addDAO(Ioc ioc, String beanName, String mapperLocation, boolean one
                     Thread.currentThread().setContextClassLoader(ioc.getClassLoader());
                 }
                 SqlSessionFactory factory = bean.buildSqlSessionFactory();
+                MybatisPlugin.sqlSessionFactory = factory;
                 ioc.putBean("mybatis_" + beanName + config.getName(), factory);
                 if (System.getenv("mybatis_mapper_annotation") != null) {
                     factory.getConfiguration().addMappers(System.getenv("mybatis_mapper_annotation"));
@@ -134,10 +139,12 @@ private void addDAO(Ioc ioc, String beanName, String mapperLocation, boolean one
                             new SqlSessionInterceptor(factory)));
                     if (one) {
                         ioc.putBean(it.getName(), proxy);
+                        ioc.putBean(mapperName + it.getName(), it.getName());
                     } else {
                         String name = it.getSimpleName() + ":" + config.getName();
                         log.info("mybatis dao name:{} add", name);
                         ioc.putBean(name, proxy);
+                        ioc.putBean(mapperName + name, it.getName());
                     }
                 });
             } finally {
@@ -149,6 +156,14 @@ private void addDAO(Ioc ioc, String beanName, String mapperLocation, boolean one
         }
     }
 
+    public static SqlSessionFactory getSqlSessionFactory() {
+        return sqlSessionFactory;
+    }
+
+    public static String getMapperName() {
+        return mapperName;
+    }
+
     @Override
     public String version() {
         return "0.0.1:2022-03-29:goodjava@qq.com";