基于druid语法树解析SQL语句增加limit的一种方法

import org.apache.calcite.sql.SqlInsert;
import org.apache.calcite.sql.SqlNode;
import org.apache.calcite.sql.parser.SqlParseException;
import org.apache.calcite.sql.parser.SqlParser;
import org.apache.calcite.tools.FrameworkConfig;
import org.apache.calcite.tools.Frameworks;
import org.apache.calcite.tools.RelConversionException;
import org.apache.calcite.tools.RelConversionException;
import org.apache.calcite.tools.RelRunner;
 
import java.util.HashMap;
import java.util.Map;
 
public class SqlLimitInjection {
 
    public static void main(String[] args) throws Exception {
        // 初始化Calcite框架配置
        FrameworkConfig calciteFrameworkConfig = Frameworks.newConfigBuilder()
                .defaultSchema(DruidSchema.createDruidSchema())
                .parserConfig(SqlParser.configBuilder().setLex(Lex.JAVA).build())
                .build();
 
        // 创建SQL解析器
        SqlParser.Config sqlParserConfig = SqlParser.configBuilder().setLex(Lex.JAVA).build();
        SqlParser sqlParser = SqlParser.create(QUERY_WITH_LIMIT, sqlParserConfig);
 
        // 解析SQL语句
        SqlNode sqlNode = sqlParser.parseStmt();
 
        // 检查是否包含LIMIT子句，并注入自定义的LIMIT值
        if (sqlNode instanceof SqlInsert) {
            SqlInsert sqlInsert = (SqlInsert) sqlNode;
            if (sqlInsert.getSelect() != null && sqlInsert.getSelect().getFetch() == null) {
                // 设置自定义的LIMIT值
                int customLimit = 1000;
                sqlInsert.getSelect().setFetch(SqlParserPos.ZERO, customLimit);
            }
        }
 
        // 将SqlNode转换为RelNode
        RelRunner relRunner = new RelRunner(calciteFrameworkConfig.getConfig(), null, null);
        relRunner.convertStatement(sqlNode);
 
        // 打印修改后的SQL语句
        System.out.println(sqlNode.toSqlString(SqlDialect.DatabaseProduct.DRUID.getDialect()));
    }
 
    private static final String QUERY_WITH_LIMIT = "INSERT INTO druid_table SELECT * FROM another_table LIMIT 10";
}

这个代码示例展示了如何使用Apache Calcite框架来解析一个包含LIMIT子句的SQL语句，并且如何在没有LIMIT子句的情况下注入自定义的LIMIT值。代码首先初始化了Calcite框架的配置，然后创建了一个SQL解析器，并解析了一个包含LIMIT子句的SQL语句。接着检查该SQL语句是否包含LIMIT子句，并且没有被指定FETCH（即没有被限制）。如果满足这些条件，