com.neko233.skilltree.commons.sql.SqlTemplate233 Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of skilltree-commons-core Show documentation
There is a newer version: 0.3.6
Show newest version
package com.neko233.skilltree.commons.sql;


import com.neko233.skilltree.commons.core.utils.StringUtils233;
import com.neko233.skilltree.commons.json.JsonUtils233;

import java.util.HashMap;
import java.util.Map;

/**
 * Simple Handle SQL batch replace
 *
 * @author SolarisNeko on 2022-08-01
 **/
public class SqlTemplate233 {

    private static final String SQL_INJECT_REGEX = ".*(exec|insert|select|delete|drop|update|count|%|chr|mid|master|truncate|char|declare|;) .*";

    private final String sqlTemplate;
    // 是否需要检查 SQL value 如 SQL 注入
    private final boolean isNeedCheckSqlValue;
    private final Map kvMap = new HashMap<>();

    public SqlTemplate233(String sqlTemplate,
                          boolean isNeedCheckSqlValue) {
        this.sqlTemplate = sqlTemplate;
        this.isNeedCheckSqlValue = isNeedCheckSqlValue;
    }

    public static boolean isValueSqlInject(String sql) {
        return sql.toLowerCase().matches(SQL_INJECT_REGEX);
    }

    public static SqlTemplate233 builder(String sqlTemplate) {
        return builder(sqlTemplate, true);
    }

    public static SqlTemplate233 builder(String sqlTemplate,
                                         boolean isNeedCheck) {
        if (StringUtils233.isBlank(sqlTemplate)) {
            throw new RuntimeException("your sql template is blank !");
        }
        return new SqlTemplate233(sqlTemplate, isNeedCheck);
    }

    public SqlTemplate233 put(String key,
                              Object value) {
        kvMap.put(key, translateValue(value, false));
        return this;
    }

    /**
     * set kv
     *
     * @param key        key
     * @param value      value
     * @param isOriginal 是否原始值
     * @return this
     */
    public SqlTemplate233 put(String key,
                              Object value,
                              boolean isOriginal) {
        kvMap.put(key, translateValue(value, isOriginal));
        return this;
    }

    public SqlTemplate233 putAll(Map kvMap) {
        return putAll(kvMap, false);
    }

    /**
     * set kv
     *
     * @param kvMap      key-value map
     * @param isOriginal 是否原始值
     * @return this
     */
    public SqlTemplate233 putAll(Map kvMap,
                                 boolean isOriginal) {
        for (Map.Entry en : kvMap.entrySet()) {
            this.kvMap.put(en.getKey(), translateValue(en.getValue(), isOriginal));
        }
        return this;
    }

    public String translateValue(Object valueObj) {
        return translateValue(valueObj, false);
    }

    public String translateValue(Object valueObj,
                                 boolean isOriginal) {
        String value;
        if (isOriginal) {
            return String.valueOf(valueObj);
        }
        if (valueObj instanceof String) {
            value = "'" + String.valueOf(valueObj).replaceAll("'", "''") + "'";
        } else {
            value = String.valueOf(valueObj);
        }
        return value;
    }

    public SqlTemplate233 merge(String key,
                                Object value) {
        return merge(key, value, "");
    }

    public SqlTemplate233 merge(String key,
                                Object value,
                                String union) {
        kvMap.merge(key, translateValue(value), (v1, v2) -> v1 + union + v2);
        return this;
    }

    public String build() {
        return this.toString();
    }

    @Override
    public String toString() {
        String tempSql = sqlTemplate;
        // 替换全部
        for (Map.Entry kv : kvMap.entrySet()) {
            String value = kv.getValue();
            if (isNeedCheckSqlValue) {
                boolean isSqlInject = isValueSqlInject(value);
                if (isSqlInject) {
                    String msg = StringUtils233.format(
                            "[SqlTemplate] SQL Inject Error. Please check. sql Template = {}. kvMap = {}"
                            ,
                            sqlTemplate,
                            JsonUtils233.toJsonString(kvMap));
                    throw new IllegalArgumentException(msg);
                }
            }
            tempSql = tempSql.replaceAll(
                    "\\$\\{" + kv.getKey() + "\\}",
                    value
            );
        }
        return tempSql;
    }
}