druid解析HiveSQL获取输入输出表

通过map保存封装调用：

private static List<SQLStatement> getSqlStatements(String sqlStr, Map<String, Object> map) {
        List<String> inputList = new ArrayList<>();
        List<String> outputList = new ArrayList<>();
        List<SQLTableSource> sourceTables = new ArrayList<>(); // 存储来源表的列表
        Map<SQLTableSource, String> sourceTableMap = new HashMap<>(); // 存储来源表与对应别名或表名的映射关系

        sqlStr = SQLUtils.format(sqlStr, DbType.hive);

        List<SQLStatement> stmtList = SQLUtils.parseStatements(sqlStr, DbType.hive);
        for (SQLStatement stmt : stmtList) {

            if (!(stmt instanceof SQLInsertStatement)) {

                if (stmt instanceof SQLSelectStatement) {
                    SQLSelectStatement select = (SQLSelectStatement) stmt;
                    SQLSelect source = select.getSelect();

                    Map<String, Object> sourcemap = getSelectTableNames(source);
                    sourceTables = (List<SQLTableSource>) sourcemap.get("sourceTables");
                    sourceTableMap = (Map<SQLTableSource, String>) sourcemap.get("sourceTableMap");

                }
                continue;
            }
            SQLInsertStatement insert = (SQLInsertStatement) stmt;
            String targetTable = insert.getTableName().toString(); // 获取目标表名
            SQLSelect source = insert.getQuery(); // 找出来源表查询

            Map<String, Object> sourcemap = getSelectTableNames(source);
            sourceTables = (List<SQLTableSource>) sourcemap.get("sourceTables");
            sourceTableMap = (Map<SQLTableSource, String>) sourcemap.get("sourceTableMap");

            outputList.add(targetTable);

          
        }

        for (SQLTableSource sourceTable : sourceTables) {
            String sourceTableName = sourceTableMap.get(sourceTable);
            inputList.add(sourceTableName);
        }

        map.put("input", inputList);
        map.put("output", outputList);

        return stmtList;
    }

private static void getSources(SQLTableSource tableSource, List<SQLTableSource> sources, Set<String> aliases) {
        if (tableSource == null) {
            return;
        }
        if (tableSource instanceof SQLJoinTableSource) {
            SQLJoinTableSource join = (SQLJoinTableSource) tableSource;

            getSources(join.getLeft(), sources, aliases);
            getSources(join.getRight(), sources, aliases);
        } else {
            if (!sources.contains(tableSource)) {
                sources.add(tableSource);
                String alias = tableSource.getAlias();
                if (alias != null) {
                    aliases.add(alias);
                }
            }
        }
    }

private static Map<String, Object> getSelectTableNames(SQLSelect source) {

        Map<String, Object> map = new HashMap<>();
        List<SQLTableSource> sourceTables = new ArrayList<>(); // 存储来源表的列表
        Set<String> sourceTableAliases = new HashSet<>(); // 存储来源表别名的集合
        Map<SQLTableSource, String> sourceTableMap = new HashMap<>(); // 存储来源表与对应别名或表名的映射关系

        if (source != null) {

            getSources(source.getQueryBlock().getFrom(), sourceTables, sourceTableAliases); // 获取主表及其关联的所有来源表
        }

        for (SQLTableSource sourceTable : sourceTables) {
            if (sourceTable instanceof SQLExprTableSource) { // 如果是基本表，则直接获取表名
                SQLExprTableSource exprTableSource = (SQLExprTableSource) sourceTable;
                String sourceTableName = exprTableSource.getName().toString();
                sourceTableMap.put(sourceTable, sourceTableName);
            } else if (sourceTable instanceof SQLSubqueryTableSource) { // 如果是子查询，则递归解析
                SQLSubqueryTableSource subqueryTableSource = (SQLSubqueryTableSource) sourceTable;
                SQLSelect subquery = subqueryTableSource.getSelect();
                List<SQLTableSource> subquerySources = new ArrayList<>();
                Set<String> subqueryAliases = new HashSet<>();
                getSources(subquery.getQueryBlock().getFrom(), subquerySources, subqueryAliases);
                for (SQLTableSource subquerySource : subquerySources) {
                    if (!sourceTables.contains(subquerySource)) {
                        sourceTables.add(subquerySource); // 将子查询中的来源表加入列表中
                    }
                    String alias =
                            subqueryAliases.contains(subquerySource.getAlias()) ? subquerySource.getAlias() : null; // 子查询中的来源表可能存在别名，进行处理
                    if (alias == null && subquerySource instanceof SQLExprTableSource) { // 如果没有别名，则使用表名作为键
                        SQLExprTableSource exprTableSource = (SQLExprTableSource) subquerySource;
                        alias = exprTableSource.getName().getSimpleName();
                    }
                    sourceTableMap.put(subquerySource, alias);
                }
            } else {
                throw new UnsupportedOperationException("Unsupported table type: " + sourceTable.getClass());
            }
        }

        map.put("sourceTables", sourceTables);
        map.put("sourceTableMap", sourceTableMap);

        return map;
    }