【Flink SQL】java解析和校验 Flink SQL语句

我们有时候需要验证自己的Flink SQL节点语句和上下文语句是否 正确，可以通过tableEnv获取的 flink sql parser 解析 sql。

 //sql为你需要验证的语句，多个语句以“;”分割
SqlParser parser = SqlParser.create(sql, SqlParser.configBuilder()
                .setParserFactory(FlinkSqlParserImpl.FACTORY)
                .setQuoting(BACK_TICK)
                .setUnquotedCasing(Casing.TO_UPPER)
                .setQuotedCasing(Casing.UNCHANGED)
                .setConformance(FlinkSqlConformance.DEFAULT)
                .build()
        );
        SqlNodeList sqlNodeList = parser.parseStmtList();
        List<SqlNode> sqlNodes = sqlNodeList.getList();
        StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();

        EnvironmentSettings settings = EnvironmentSettings.newInstance()
                .inStreamingMode()
                .build();

        StreamTableEnvironment tableEnv = StreamTableEnvironment.create(env, settings);
        // 通过 table env 获取 Planner 和 CatalogManager
        CatalogManager catalogManager = ((TableEnvironmentImpl) tableEnv).getCatalogManager();
        StreamPlanner planner = (StreamPlanner) ((TableEnvironmentImpl) tableEnv).getPlanner();

        //创建 flink planner 实例
        FlinkPlannerImpl flinkPlanner = planner.createFlinkPlanner();

         // 校验单条 sqlNode
        SqlToOperationConverter.convert(flinkPlanner, catalogManager, sqlNodes.get(0)).isPresent();



        //校验上下文的sql,如果你的sql存在修改和删除，你还需要对删除和修改操作进行判断
        sqlNodes.forEach(nodes -> {
                    // 单条 sql 校验
                    Operation operation = SqlToOperationConverter.convert(flinkPlanner, catalogManager, nodes)
                            .orElseThrow(() -> new TableException("Unsupported query: " + nodes));
                    if(operation instanceof CreateTableOperation){
                        CreateTableOperation createTableOperation = (CreateTableOperation) operation;
                        //创建临时表
                        if(createTableOperation.isTemporary()){
                            catalogManager.createTemporaryTable(
                                    createTableOperation.getCatalogTable(),
                                    createTableOperation.getTableIdentifier(),
                                    createTableOperation.isIgnoreIfExists());
                        }else{
                            catalogManager.createTable(
                                    createTableOperation.getCatalogTable(),
                                    createTableOperation.getTableIdentifier(),
                                    createTableOperation.isIgnoreIfExists());
                        }
                    }
                }
        );