生成修复设置实体父类 Class 未读包路径 BUG

yes00101 · Feb 26, 2019 · 4a1c122 · 4a1c122
1 parent 164ef83
commit 4a1c122
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/...lus-generator/src/main/java/com/baomidou/mybatisplus/generator/config/StrategyConfig.java b/...lus-generator/src/main/java/com/baomidou/mybatisplus/generator/config/StrategyConfig.java
@@ -19,6 +19,7 @@
 import java.util.List;
 import java.util.stream.Collectors;
 
+import com.baomidou.mybatisplus.core.toolkit.ClassUtils;
 import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
 import com.baomidou.mybatisplus.core.toolkit.StringUtils;
 import com.baomidou.mybatisplus.core.toolkit.TableInfoHelper;
@@ -264,7 +265,13 @@ public StrategyConfig setSuperEntityClass(Class<?> clazz, NamingStrategy columnN
         if (null != columnNaming) {
             this.columnNaming = columnNaming;
         }
-        this.superEntityClass = clazz.getSimpleName();
+        String pkg = ClassUtils.getPackageName(clazz);
+        if (StringUtils.isNotEmpty(pkg)) {
+            pkg += "." + clazz.getSimpleName();
+        } else {
+            pkg = clazz.getSimpleName();
+        }
+        this.superEntityClass = pkg;
         convertSuperEntityColumns(clazz);
         return this;
     }