优化代码

makunet · Jan 13, 2023 · 46ada80 · 46ada80
1 parent e2f8f00
commit 46ada80
Showing 1 changed file with 12 additions and 10 deletions.
diff --git a/maku-generator-core/src/main/java/net/maku/generator/utils/GenUtils.java b/maku-generator-core/src/main/java/net/maku/generator/utils/GenUtils.java
@@ -1,5 +1,6 @@
 package net.maku.generator.utils;
 
+import cn.hutool.core.text.NamingCase;
 import cn.hutool.core.util.StrUtil;
 import com.baomidou.mybatisplus.core.toolkit.StringUtils;
 import lombok.extern.slf4j.Slf4j;
@@ -148,30 +149,31 @@ public static String getFunctionName(String tableName) {
 
         return functionName;
     }
-    
+
     /**
      * 表名转驼峰并移除前后缀
-     * @param upperFirst 首字母大写
-     * @param tableName 表名
+     *
+     * @param upperFirst   首字母大写
+     * @param tableName    表名
      * @param removePrefix 删除前缀
      * @param removeSuffix 删除后缀
      * @return java.lang.String
      * @date 2022/12/24 14:33
      */
-    public static String camelCase(boolean upperFirst,String tableName, String removePrefix, String removeSuffix){
+    public static String camelCase(boolean upperFirst, String tableName, String removePrefix, String removeSuffix) {
         String className = tableName;
         // 移除前缀
-        if (StrUtil.isNotBlank(removePrefix)){
-            className = StrUtil.removePrefix(tableName,removePrefix);
+        if (StrUtil.isNotBlank(removePrefix)) {
+            className = StrUtil.removePrefix(tableName, removePrefix);
         }
         // 移除后缀
-        if (StrUtil.isNotBlank(removeSuffix)){
-            className = StrUtil.removeSuffix(className,removeSuffix);
+        if (StrUtil.isNotBlank(removeSuffix)) {
+            className = StrUtil.removeSuffix(className, removeSuffix);
         }
         // 是否首字母大写
-        if (upperFirst){
+        if (upperFirst) {
             return NamingCase.toPascalCase(className);
-        }else {
+        } else {
             return NamingCase.toCamelCase(className);
         }
     }