From 52716a2f37f872cdee4cd9fd9f9839b663be0434 Mon Sep 17 00:00:00 2001
From: zhangping17 <zhangping17@xiaomi.com>
Date: Mon, 7 Aug 2023 19:42:37 +0800
Subject: [PATCH] mybatis for ann

---
 jcommon/docean-plugin/docean-plugin-mybatis/pom.xml            | 2 +-
 .../com/xiaomi/youpin/docean/plugin/mybatis/MybatisPlugin.java | 3 +++
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/jcommon/docean-plugin/docean-plugin-mybatis/pom.xml b/jcommon/docean-plugin/docean-plugin-mybatis/pom.xml
index 3fb6d455b..470f8a051 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.5-SNAPSHOT</version>
+    <version>1.4.6-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 cef6e3e33..25e03d5bb 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
@@ -119,6 +119,9 @@ private void addDAO(Ioc ioc, String beanName, String mapperLocation, boolean one
                 }
                 SqlSessionFactory factory = bean.buildSqlSessionFactory();
                 ioc.putBean("mybatis_" + beanName + config.getName(), factory);
+                if (System.getenv("mybatis_mapper_annotation") != null) {
+                    factory.getConfiguration().addMappers(System.getenv("mybatis_mapper_annotation"));
+                }
                 Collection<Class<?>> mappers = factory.getConfiguration().getMapperRegistry().getMappers();
 
                 List<String> mapperNameList = mappers.stream().map(it -> it.getName()).collect(Collectors.toList());