001package gudusoft.gsqlparser.dlineage.statistics;
002
003import gudusoft.gsqlparser.nodes.TParseTreeVisitor;
004import gudusoft.gsqlparser.nodes.TTable;
005
006import java.util.HashSet;
007import java.util.Set;
008
009/**
010 * 表引用访问者
011 */
012public class TableReferenceVisitor extends TParseTreeVisitor {
013    private Set<String> tableReferences = new HashSet<>();
014
015    public void preVisit(TTable node) {
016        if (node.getTableName() != null) {
017            tableReferences.add(node.getTableName().toString());
018        }
019    }
020
021    public Set<String> getTableReferences() {
022        return tableReferences;
023    }
024    
025    public void reset() {
026        tableReferences.clear();
027    }
028}