使用Druid的SqlParser解析sql语句

谢谢蟹黄堡

于 2024-02-05 02:00:55 发布

阅读量733

点赞数 8

文章标签： sql 数据库

本文链接：https://blog.csdn.net/qq_38978882/article/details/132846866

版权

解析sql
public Map<Integer, String> parseSql(String sql, DbType dbType) {
        List<SQLStatement> statementList = SQLUtils.parseStatements(sql, dbType);
        Map<Integer, String> sqlMap = new HashMap<Integer,String>();
        for (SQLStatement stmt : statementList) {
            if (stmt instanceof SQLInsertStatement) {
                SQLInsertStatement insertStatement = (SQLInsertStatement) stmt;
                List<SQLInsertStatement.ValuesClause> valuesClauses = insertStatement.getValuesList();
                for (int i = 0; i < insertStatement.getColumns().size(); i++) {
                    String columnName = insertStatement.getColumns().get(i).toString();
                    SQLExpr columnValue = valuesClauses.get(0).getValues().get(i);
                    sqlMap.put(i,columnName);
                }
            }
        }
        return sqlMap;
    }
变更sql
public static void main(String[] args) {
        String sql ="INSERT INTO student  ( user_name, email )  VALUES  ( ?,? )";

        List<SQLStatement> stmtList = SQLUtils.parseStatements(sql, DbType.mysql);

        for (SQLStatement stmt : stmtList) {
            if (stmt instanceof SQLInsertStatement) {
                SQLInsertStatement insert = (SQLInsertStatement) stmt;
                // 添加字段和值

                insert.addColumn(SQLUtils.toSQLExpr("user_name_1", DbType.mysql));
                // 创建新的值子句
                SQLInsertStatement.ValuesClause values = insert.getValues();

                values.addValue(SQLUtils.toSQLExpr("？", DbType.mysql));

                // 输出修改后的SQL语句
                String modifiedSql = SQLUtils.toMySqlString(insert);

                System.out.println(modifiedSql);

            }
        }
    }