Druid连接池工具公式化SQL附踩坑记录

1. 需求

使用Druid连接池工具格式化sql用于回显时候美观展示

2. 代码示例

2.1 依赖

      com.alibaba
      druid
      1.2.6

2.2 ParseUtils

import com.alibaba.druid.DbType;
import com.alibaba.druid.sql.ast.SQLStatement;
import com.alibaba.druid.sql.dialect.postgresql.visitor.PGOutputVisitor;
import com.alibaba.druid.sql.parser.SQLParserUtils;
import com.alibaba.druid.sql.parser.SQLStatementParser;
import com.alibaba.druid.sql.visitor.SQLASTOutputVisitor;

import java.util.List;

/**
 * @author pp_lan
 * @date 2024/1/15
 */
public class ParseUtils {

    private ParseUtils() {
    }

    public static String format(String sql, String dbTypeStr) {
        DbType dbType = DbType.of(dbTypeStr);
        if (dbType == null) {
            throw new RuntimeException("不支持的数据库类型");
        }
        List statementList = toStatementList(sql, dbType);
        String result = sqlToString(statementList, dbTypeStr, null);
        return result;
    }

    public static List toStatementList(String sql, DbType dbType) {
        SQLStatementParser parser = SQLParserUtils.createSQLStatementParser(sql, dbType);
        return parser.parseStatementList();
    }

    public static String sqlToString(List statementList, String dbType, List