java - 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 - 这具有单独的 >、< 和其他操作。
• 常量和列运算符。

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

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

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

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

编辑添加:

请记住，查询解析器创建的树中的每个节点都会调用 Vistor#visit() 方法。 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(); 

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