浏览代码

Update 优化数据库类型获取和判断逻辑,增强代码可读性

Yue 6 月之前
父节点
当前提交
2a18de8179

+ 31 - 2
SERVER/VberAdminPlusV3/vber-common/vber-common-mybatis/src/main/java/com/vber/common/mybatis/enums/DataBaseType.java

@@ -46,13 +46,42 @@ public enum DataBaseType {
      */
     public static DataBaseType find(String databaseProductName) {
         if (StringUtils.isBlank(databaseProductName)) {
-            return null;
+            return MY_SQL;
         }
         for (DataBaseType type : values()) {
             if (type.getType().equals(databaseProductName)) {
                 return type;
             }
         }
-        return null;
+        return MY_SQL;
     }
+
+    /**
+     * 判断是否为 MySQL 类型
+     */
+    public boolean isMySql() {
+        return this == MY_SQL;
+    }
+
+    /**
+     * 判断是否为 Oracle 类型
+     */
+    public boolean isOracle() {
+        return this == ORACLE;
+    }
+
+    /**
+     * 判断是否为 PostgreSQL 类型
+     */
+    public boolean isPostgreSql() {
+        return this == POSTGRE_SQL;
+    }
+
+    /**
+     * 判断是否为 SQL Server 类型
+     */
+    public boolean isSqlServer() {
+        return this == SQL_SERVER;
+    }
+
 }

+ 29 - 29
SERVER/VberAdminPlusV3/vber-common/vber-common-mybatis/src/main/java/com/vber/common/mybatis/helper/DataBaseHelper.java

@@ -26,7 +26,14 @@ public class DataBaseHelper {
     private static final DynamicRoutingDataSource DS = SpringUtils.getBean(DynamicRoutingDataSource.class);
 
     /**
-     * 获取当前数据库类型
+     * 获取当前数据源对应的数据库类型
+     * <p>
+     * 通过 DynamicRoutingDataSource 获取当前线程绑定的数据源,
+     * 然后从数据源获取数据库连接,利用连接的元数据获取数据库产品名称,
+     * 最后调用 DataBaseType.find 方法将数据库名称转换为对应的枚举类型
+     *
+     * @return 当前数据库对应的 DataBaseType 枚举,找不到时默认返回 MY_SQL
+     * @throws ServiceException 当获取数据库连接或元数据出现异常时抛出业务异常
      */
     public static DataBaseType getDataBaseType() {
         DataSource dataSource = DS.determineDataSource();
@@ -39,37 +46,30 @@ public class DataBaseHelper {
         }
     }
 
-    public static boolean isMySql() {
-        return DataBaseType.MY_SQL == getDataBaseType();
-    }
-
-    public static boolean isOracle() {
-        return DataBaseType.ORACLE == getDataBaseType();
-    }
-
-    public static boolean isPostgerSql() {
-        return DataBaseType.POSTGRE_SQL == getDataBaseType();
-    }
-
-    public static boolean isSqlServer() {
-        return DataBaseType.SQL_SERVER == getDataBaseType();
-    }
-
+    /**
+     * 根据当前数据库类型,生成兼容的 FIND_IN_SET 语句片段
+     * <p>
+     * 用于判断指定值是否存在于逗号分隔的字符串列中,SQL写法根据不同数据库方言自动切换:
+     * - Oracle 使用 instr 函数
+     * - PostgreSQL 使用 strpos 函数
+     * - SQL Server 使用 charindex 函数
+     * - 其他默认使用 MySQL 的 find_in_set 函数
+     *
+     * @param var1 要查找的值(支持任意类型,内部会转换成字符串)
+     * @param var2 存储逗号分隔值的数据库列名
+     * @return 适用于当前数据库的 SQL 条件字符串,通常用于 where 或 apply 中拼接
+     */
     public static String findInSet(Object var1, String var2) {
-        DataBaseType dataBasyType = getDataBaseType();
         String var = Convert.toStr(var1);
-        if (dataBasyType == DataBaseType.SQL_SERVER) {
-            // charindex(',100,' , ',0,100,101,') <> 0
-            return "charindex(',%s,' , ','+%s+',') <> 0".formatted(var, var2);
-        } else if (dataBasyType == DataBaseType.POSTGRE_SQL) {
+        return switch (getDataBaseType()) {
+            case ORACLE -> "instr(','||%s||',' , ',%s,') <> 0".formatted(var2, var);
             // (select strpos(',0,100,101,' , ',100,')) <> 0
-            return "(select strpos(','||%s||',' , ',%s,')) <> 0".formatted(var2, var);
-        } else if (dataBasyType == DataBaseType.ORACLE) {
-            // instr(',0,100,101,' , ',100,') <> 0
-            return "instr(','||%s||',' , ',%s,') <> 0".formatted(var2, var);
-        }
-        // find_in_set(100 , '0,100,101')
-        return "find_in_set('%s' , %s) <> 0".formatted(var, var2);
+            case POSTGRE_SQL -> "(select strpos(','||%s||',' , ',%s,')) <> 0".formatted(var2, var);
+            // charindex(',100,' , ',0,100,101,') <> 0
+            case SQL_SERVER -> "charindex(',%s,' , ','+%s+',') <> 0".formatted(var, var2);
+            // find_in_set(100 , '0,100,101')
+            default -> "find_in_set('%s' , %s) <> 0".formatted(var, var2);
+        };
     }
 
     /**

+ 13 - 9
SERVER/VberAdminPlusV3/vber-modules/vber-generator/src/main/java/com/vber/generator/util/VelocityUtils.java

@@ -6,6 +6,8 @@ import cn.hutool.core.lang.Dict;
 import com.vber.common.core.utils.DateUtils;
 import com.vber.common.core.utils.StringUtils;
 import com.vber.common.json.utils.JsonUtils;
+import com.vber.common.mybatis.enums.DataBaseType;
+import com.vber.common.mybatis.helper.DataBaseHelper;
 import com.vber.generator.constant.GenConstants;
 import com.vber.generator.domain.GenTable;
 import com.vber.generator.domain.GenTableColumn;
@@ -130,15 +132,17 @@ public class VelocityUtils {
         templates.add("vm/java/serviceImpl.java.vm");
         templates.add("vm/java/controller.java.vm");
         templates.add("vm/xml/mapper.xml.vm");
-//        if (DataBaseHelper.isOracle()) {
-//            templates.add("vm/sql/oracle/sql.vm");
-//        } else if (DataBaseHelper.isPostgerSql()) {
-//            templates.add("vm/sql/postgres/sql.vm");
-//        } else if (DataBaseHelper.isSqlServer()) {
-//            templates.add("vm/sql/sqlserver/sql.vm");
-//        } else {
-//            templates.add("vm/sql/sql.vm");
-//        }
+
+        DataBaseType dataBaseType = DataBaseHelper.getDataBaseType();
+        if (dataBaseType.isOracle()) {
+            templates.add("vm/sql/oracle/sql.vm");
+        } else if (dataBaseType.isPostgreSql()) {
+            templates.add("vm/sql/postgres/sql.vm");
+        } else if (dataBaseType.isSqlServer()) {
+            templates.add("vm/sql/sqlserver/sql.vm");
+        } else {
+            templates.add("vm/sql/sql.vm");
+        }
         templates.add("vm/ts/api.ts.vm");
         if (GenConstants.TPL_CRUD.equals(tplCategory)) {
             templates.add("vm/vue/view.vue.vm");