Блог

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

Ваш код не показывает, что QueryDescription делает класс, но я могу догадаться.

При работе с предложением where вы ищете три типа узлов:

• BinaryLogicalOperatorNode — В нем есть операторы AND , OR, IS, которые разделяют отдельные предложения в предложении WHERE .
• BinaryOperatorNode — Это имеет отдельные >, < и другие операции.
• Операторы констант и столбцов.

В вашем примере вы должны посетить a BinaryOperatorNode с типом LT , и два дочерних элемента будут ConstantNode иметь значение 5, а ColumnReference узел — значение «d».

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

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

Один узел, который я бы добавил в список проверок в дескрипторах InListOperatorNode which 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(); 
  

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