Function formatQuery

src/Client/ClientBaseHelpers.cpp:574–653

}

String formatQuery(String query)
{
    const unsigned max_parser_depth = DBMS_DEFAULT_MAX_PARSER_DEPTH;
    const unsigned max_parser_backtracks = DBMS_DEFAULT_MAX_PARSER_BACKTRACKS;

    ParserQuery parser(query.data() + query.size(), /*allow_settings_after_format_in_insert_=*/ false, /*implicit_select_=*/ false);

    String res;
    res.reserve(query.size());

    auto comments_callback = [&](std::string_view comment)
    {
        res += comment;
        res += '\n';
    };

    const char * begin = query.data();
    const char * pos = begin;
    const char * end = begin + query.size();

    skipSpacesAndComments(pos, end, comments_callback);

    size_t queries = 0;
    while (pos < end)
    {
        const char * query_start = pos;
        const ASTPtr ast = parseQueryAndMovePosition(parser, pos, end, "query in editor", /*allow_multi_statements=*/ true, /*max_query_size=*/ 0, max_parser_depth, max_parser_backtracks);

        std::string_view insert_query_payload;
        if (auto * insert_ast = ast->as<ASTInsertQuery>(); insert_ast && insert_ast->data)
        {
            if (Poco::toLower(insert_ast->format) == "values")
            {
                /// Reset format to default to have `INSERT INTO table VALUES` instead of `INSERT INTO table VALUES FORMAT Values`
                insert_ast->format.clear();

                /// We assume that data ends with a newline character (same as in other places)
                const char * this_query_end = find_first_symbols<'\n'>(insert_ast->data, end);
                insert_ast->end = this_query_end;
                pos = this_query_end;

                /// Remove semicolon from the INSERT query payload, since it will be added explicitly below
                if (*insert_ast->end == '\n')
                {
                    while (insert_ast->end > insert_ast->data && std::isspace(*insert_ast->end))
                        --insert_ast->end;
                }
            }
            else
                pos = insert_ast->end;

            /// No need to use getReadBufferFromASTInsertQuery() here, since it does extra things that we do not need here (i.e. handle INFILE)
            insert_query_payload = std::string_view(insert_ast->data, insert_ast->end);
        }

        bool multiline_query = std::string_view(query_start, pos).contains('\n');
        if (multiline_query)
            res += ast->formatWithSecretsMultiLine();