瀏覽代碼

自定义方法名优化

hubin 5 年之前
父節點
當前提交
0dc279126e

+ 15 - 0
mybatis-plus-core/src/main/java/com/baomidou/mybatisplus/core/toolkit/ClassUtils.java

@@ -128,6 +128,21 @@ public final class ClassUtils {
         }
     }
 
+    /**
+     * 获取【 包名 + 类名 】完整路径字符串
+     *
+     * @param clazz
+     * @return
+     */
+    public static String getPackageAndSimpleName(Class<?> clazz) {
+        String pkg = getPackageName(clazz);
+        if (StringUtils.isNotEmpty(pkg)) {
+            pkg += "." + clazz.getSimpleName();
+        } else {
+            pkg = clazz.getSimpleName();
+        }
+        return pkg;
+    }
 
     /**
      * Determine the name of the package of the given class,

+ 5 - 7
mybatis-plus-generator/src/main/java/com/baomidou/mybatisplus/generator/config/StrategyConfig.java

@@ -271,17 +271,15 @@ public class StrategyConfig {
         if (null != columnNaming) {
             this.columnNaming = columnNaming;
         }
-        String pkg = ClassUtils.getPackageName(clazz);
-        if (StringUtils.isNotEmpty(pkg)) {
-            pkg += "." + clazz.getSimpleName();
-        } else {
-            pkg = clazz.getSimpleName();
-        }
-        this.superEntityClass = pkg;
+        this.superEntityClass = ClassUtils.getPackageAndSimpleName(clazz);
         convertSuperEntityColumns(clazz);
         return this;
     }
 
+    public void setSuperControllerClass(Class<?> clazz) {
+        this.superControllerClass = ClassUtils.getPackageAndSimpleName(clazz);
+    }
+
     /**
      * <p>
      * 父类 Class 反射属性转换为公共字段