FoundationDB SQL 解析器获取 WHERE 子句

Question

我正在使用 foundationDB SQL 解析器 (https://github.com/FoundationDB/sql-parser) 来解析 Java 中的查询，但我对解析器使用的访问者设计模式不是很熟悉使用查询。
我想像这样向解析器发送查询： "SELECT a, b FROM c WHERE d > 5" 并得到结果：

1. SELECT 子句中的所有字段名称（已完成）
2. FROM子句中的表名（已完成）
3. WHERE子句中的列名、操作数和表达式

这就是我正在实现的代码：

    @Override
public QueryDescription parse() throws StandardException {

    SQLParser parser = new SQLParser();

    StatementNode stmt = parser.parseStatement(sql);

    Visitor v = new Visitor() {

        @Override
        public boolean visitChildrenFirst(Visitable arg0) {
            // TODO Auto-generated method stub
            return false;
        }

        @Override
        public Visitable visit(Visitable arg0) throws StandardException {

            // Temporary stores the QueryDescription parameters
            StatementEnum se = null;
            String fromTable = null;
            String[] fields = null;

            if(arg0 instanceof CursorNode) {
                CursorNode cn = (CursorNode) arg0;

                // print out what statement is been declared in sql query
                System.out.println("Statement: " + cn.statementToString());

                // temporarly stores the statement
                String statement = cn.statementToString();

                // creates the right StatementEnum
                if(statement == "CREATE TABLE") {
                    se = StatementEnum.CREATE_TABLE;
                } else if(statement == "INSERT") {
                    se = StatementEnum.INSERT;
                } else if(statement == "SELECT") {
                    se = StatementEnum.SELECT;
                } else if(statement == "DROP TABLE") {
                    se = StatementEnum.DROP_TABLE;
                }
            } 

            description = new QueryDescription(se, fromTable, fields);

            return arg0;
        }

        @Override
        public boolean stopTraversal() { return false; }

        @Override
        public boolean skipChildren(Visitable arg0) throws StandardException { return false; }
    };  
    stmt.accept(v);

    // TODO remove, only for debug purpose
    stmt.treePrint();

    return description;
}

这就是 QueryDescription 类代码：

public class QueryDescription {

    /* Member variables: */
    private QueryTypeEnum queryType;
    private StatementEnum statement;
    private String fromTable;
    private String[] fields;


    /* Constructors: */
    /**
     * 
     * @param statement
     * @param fromTable
     * @param fields
     */
    public QueryDescription(StatementEnum statement, String fromTable, String[] fields) {
        this.statement = statement;
        this.fromTable = fromTable;
        this.fields = fields;
    }


    /* Methods: */
    /**
     * Analyze which type of query is the one passed by parameter and assigns the right queryTypeEnum
     */
    public void assignType() {

        switch(statement) {

            case CREATE_TABLE:
                break;

            case SELECT:
                if(fields[0] == "allFields")
                    queryType = QueryTypeEnum.DUMP;
                else {
                    // TODO risolvere questione del WHERE
                    queryType = QueryTypeEnum.SELECT_FROM;
                }
                break;

            case UPDATE:
                break;

            case INSERT:
                break;

            case DROP_TABLE:
                break;
        }
    }


    /* Getters and Setter: */
    /**
     * 
     * @return the queryType
     */
    public QueryTypeEnum getQueryType() {
        return queryType;
    }

    /**
     * 
     * @return the statement
     */
    public StatementEnum getStatement() {
        return statement;
    }

    /**
     * 
     * @return the from table
     */
    public String getFromTable() {
        return fromTable;
    }

    /**
     * 
     * @return the fields
     */
    public String[] getFields() {
        return fields;
    }
}

Answer 1

您的代码没有显示QueryDescription 类的作用，但我可以猜到。

在处理 where 子句时，您正在寻找三种类型的节点：

• BinaryLogicalOperatorNode - 这有 AND、OR、IS 运算符，用于分隔 WHERE 子句中的各个子句。
• BinaryOperatorNode - 这有单独的 >、
• 常量和列运算符。

在您的示例中，您将访问一个BinaryOperatorNode，其类型为LT，两个子节点将是ConstantNode，其值为5，以及一个ColumnReference 节点，其值为“d ”。

注意：解析器不会将列引用连接到表列表中的表。这是查询处理中的一个单独步骤。因此，您将拥有一个 ColumnReference 节点，但没有指向该列引用的表的链接。原因是解析器没有足够的信息将“d”列正确链接到正确的表。

这足以处理您在示例中给出的简单查询。显然查询会变得更加复杂。

我将添加到处理WHERE d IN (1,2,3,4) 的InListOperatorNode 中的检查列表中的一个节点。

编辑添加：

请记住，Vistor#visit() 方法会为查询解析器创建的树中的每个节点调用。访问方法需要检查并正确设置变量，用于FromBaseTable、ConstantNode、ColumnReference。

Visitor v = new Visitor() {
    List<String> fromTable = new ArrayList<String>();
    List<String> fields = new ArrayList<String>();

    // other Visitor methods go here, not copied for conciseness. 

    @Override
    public Visitable visit(Visitable arg0) throws StandardException {
        // other code from your visit() method goes here
        // 
        if (arg0 instanceof FromBaseTable) {
           FromBaseTable table = (FromBaseTable)arg0;
           fromTable.append(table.getTableName());
        } else if (arg0 instanceof ColumnReference) {
           ColumnReference column = (ColumnReference) arg0;
           fields.append(column.getColumnName())
        }
        // Remove the call to create QueryDescription
    }
    public QueryDescription getQueryDescription() {
        return new QueryDescription(se, fromTable, fields)
    }
}

然后在你调用的主代码行中：

stmt.accept(v);
QueryDescription description = v.getQueryDescription(); 

现在，如果您对查询的其他部分感兴趣，则需要将这些节点添加到访问方法中，并捕获您感兴趣的部分节点（名称、值等）。