init

bashi-common/src/main/java/com/bashi/common/utils/sql/SqlUtil.java

@@ -0,0 +1,37 @@
+package com.bashi.common.utils.sql;
+
+import cn.hutool.core.lang.Validator;
+import com.bashi.common.exception.BaseException;
+
+/**
+ * sql操作工具类
+ *
+ * @author duteliang
+ */
+public class SqlUtil
+{
+    /**
+     * 仅支持字母、数字、下划线、空格、逗号、小数点（支持多个字段排序）
+     */
+    public static String SQL_PATTERN = "[a-zA-Z0-9_\\ \\,\\.]+";
+
+    /**
+     * 检查字符，防止注入绕过
+     */
+    public static String escapeOrderBySql(String value)
+    {
+        if (Validator.isNotEmpty(value) && !isValidOrderBySql(value))
+        {
+            throw new BaseException("参数不符合规范，不能进行查询");
+        }
+        return value;
+    }
+
+    /**
+     * 验证 order by 语法是否符合规范
+     */
+    public static boolean isValidOrderBySql(String value)
+    {
+        return value.matches(SQL_PATTERN);
+    }
+}