优化代码

RuoYi

2022-02-12 98c62a9f5d38d106749609389577b52ddac5ae8a

 src/main/java/com/ruoyi/common/utils/sql/SqlUtil.java

@@ -1,6 +1,6 @@
package com.ruoyi.common.utils.sql;



import com.ruoyi.common.exception.BaseException;

import com.ruoyi.common.exception.UtilException;

import com.ruoyi.common.utils.StringUtils;



/**

@@ -10,6 +10,11 @@
 */

public class SqlUtil

{

    /**

     * 定义常用的 sql关键字

     */

    public static String SQL_REGEX = "select |insert |delete |update |drop |count |exec |chr |mid |master |truncate |char |and |declare ";



    /**

     * 仅支持字母、数字、下划线、空格、逗号、小数点（支持多个字段排序）

     */

@@ -22,7 +27,7 @@
    {

        if (StringUtils.isNotEmpty(value) && !isValidOrderBySql(value))

        {

            throw new BaseException("参数不符合规范，不能进行查询");

            throw new UtilException("参数不符合规范，不能进行查询");

        }

        return value;

    }

@@ -34,4 +39,23 @@
    {

        return value.matches(SQL_PATTERN);

    }



    /**

     * SQL关键字检查

     */

    public static void filterKeyword(String value)

    {

        if (StringUtils.isEmpty(value))

        {

            return;

        }

        String[] sqlKeywords = StringUtils.split(SQL_REGEX, "\\|");

        for (String sqlKeyword : sqlKeywords)

        {

            if (StringUtils.indexOfIgnoreCase(value, sqlKeyword) > -1)

            {

                throw new UtilException("参数存在SQL注入风险");

            }

        }

    }

}