Анализатор SQL FoundationDB для получения предложения WHERE

Question

Я использую парсер SQL FoundationDB ( https://github.com/FoundationDB/sql-parser) для разбора запроса внутри Java, но я не очень знаком с шаблоном проектирования посетителя, который используется парсером для потреблять запрос.
Я хотел бы отправить запрос синтаксическому анализатору следующим образом: "ВЫБЕРИТЕ a, b ОТ C, ГДЕ 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".

Примечание: синтаксический анализатор не связывает ссылки Column с таблицами в списке таблиц. Это отдельный шаг в обработке запроса. Так что у вас будет ColumnReference узел, но нет ссылки на таблицу, на которую ссылается столбец. Причина в том, что у синтаксического анализатора недостаточно информации, чтобы правильно связать столбец "d" с правильной таблицей.

Этого достаточно для обработки простого запроса, который вы дали в примере. Очевидно, что запросы могут стать намного более сложными.

Один узел, который я бы добавил в список проверок в InListOperatorNode какие ручки WHERE d IN (1,2,3,4),

РЕДАКТИРОВАТЬ, чтобы добавить:

Имейте в виду 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(); 

Теперь, если у вас есть другие части интересующего вас запроса, вам нужно добавить эти узлы в метод посещения и захватить интересующую вас часть узла (имена, значения и т. Д.).