SqlForamt,sql组装工具类

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @author: wenyong.wang@kuwo.cn
 * @version: v1.0
 * @description: cn.kuwo.added.service.ui
 * @date:2023/2/20
 */
public class SqlForamt {

    private static final String sqlFormat = "select %s from %s where 1=1 %s %s %s %s %s";

    private StringBuffer columns = new StringBuffer();

    private StringBuffer table = new StringBuffer();

    private StringBuffer condition = new StringBuffer();

    private StringBuffer groupBy = new StringBuffer();

    private StringBuffer orderBy = new StringBuffer();

    private StringBuffer limit = new StringBuffer();

    private StringBuffer ext = new StringBuffer();

    public String getSQL(){
        return String.format(sqlFormat, columns, table, condition, ext, groupBy, orderBy, limit);
    }

    public SqlForamt setTable(String tableName) {
        table.append(" ").append(tableName).append(" ");
        return this;
    }

    public SqlForamt setColumns(List<String> columnNames) {
        Boolean first = true;
        for (String columnName : columnNames) {
            if (!first) {
                condition.append(",");
            }
            columns.append(" ").append(columnName).append(",");
            first = false;
        }
        return this;
    }

    public SqlForamt setWhereExtSql(String value) {
        ext.append(value);
        return this;
    }

    public SqlForamt setWhereEqualSQL(String column, String value) {
        condition.append(" and ").append(column).append(" = ").append(value);
        return this;
    }

    public SqlForamt setWhereGreaterSQL(String column, String value) {
        condition.append(" and ").append(column).append(" > ").append(value);
        return this;
    }

    public SqlForamt setWhereLessSQL(String column, String value) {
        condition.append(" and ").append(column).append(" < ").append(value);
        return this;
    }

    public SqlForamt setWhereInSQL(String column, List<String> value) {
        condition.append(" and ").append(column).append(" in ").append(" ( ");
        Boolean first = true;
        for (Object o : value) {
            if (!first) {
                condition.append(",");
            }
            condition.append("'").append(o).append("'");
            first = false;
        }
        condition.append(" ) ");
        return this;
    }

    public SqlForamt setGroupBySQL(String column){
        orderBy.append(" group by ").append(column).append(" ");
        return this;
    }

    public SqlForamt setOrderBySQL(String column, String condition){
        orderBy.append(" ").append(column).append(" ").append(condition);
        return this;
    }

    public SqlForamt setLimitSQL(Integer offset, Integer size){
        orderBy.append(" limit ").append(offset).append(",").append(size);
        return this;
    }

    public static List<Map<String, Object>> converListMap(List<String> columns, List<Object> values) {
        List<Map<String, Object>> list = new ArrayList<>();
        if (columns == null || columns.isEmpty() || values == null || values.isEmpty()) {
            return list;
        }
        if (columns.size() != values.size()) {
            return list;
        }
        for (Object valueobj : values) {
            Map<String, Object> map = new HashMap<>();
            Object[] valuearr = (Object[]) valueobj;
            for (int i = 0; i < valuearr.length; i++) {
                map.put(columns.get(i), valuearr[i]);
            }
            list.add(map);
        }
        return list;
    }

    public static Map<String, Object> converMap(List<String> columns, List<Object> values) {
        if (columns == null || columns.isEmpty() || values == null || values.isEmpty()) {
            return new HashMap<>();
        }
        if (columns.size() != values.size()) {
            return new HashMap<>();
        }
        for (Object valueobj : values) {
            Map<String, Object> map = new HashMap<>();
            Object[] valuearr = (Object[]) valueobj;
            for (int i = 0; i < valuearr.length; i++) {
                map.put(columns.get(i), valuearr[i]);
            }
            return map;
        }
        return new HashMap<>();
    }

}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
防止SQL注入是开发Web应用程序时的一个重要安全措施。下面是一个简单的示例工具类,用于防止SQL注入攻击: ```java import java.sql.PreparedStatement; import java.sql.SQLException; public class SQLInjectionProtection { public static PreparedStatement prepareStatement(String query, Object... params) throws SQLException { // 创建PreparedStatement对象 PreparedStatement statement = connection.prepareStatement(query); // 为参数设置值 for (int i = 0; i < params.length; i++) { if (params[i] instanceof String) { // 对String类型的参数进行过滤 statement.setString(i + 1, sanitizeString((String) params[i])); } else { // 其他类型的参数直接设置值 statement.setObject(i + 1, params[i]); } } return statement; } private static String sanitizeString(String input) { // 对输入字符串进行过滤,替换特殊字符 String sanitized = input.replaceAll("['\"\\\\]", ""); return sanitized; } } ``` 这个工具类的核心方法是prepareStatement(),它接受一个SQL查询和参数,并返回一个已经处理好的PreparedStatement对象。在设置参数值时,对于String类型的参数,会调用sanitizeString()方法对其进行过滤,去除可能导致注入的特殊字符。 虽然这个工具类提供了一定的防护,但并不能保证完全防止SQL注入攻击。因此,还需要在开发过程中注意其他具体的防护措施,例如使用参数化查询、严格控制用户输入等。此外,及时更新数据库驱动程序和数据库软件也是保持安全的好方法。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值