Method selectGroup

src/ResourceGroup/ResourceGroupSelectStrategy.cpp:23–112

}

IResourceGroup* UserTableSelectStrategy::selectGroup(const Context & query_context, const IAST *ast)
{
    auto & select_cases = resource_group_manager->getSelectCases();
    const ClientInfo & client_info = query_context.getClientInfo();

    const auto get_user_group = [&select_cases, &client_info]() -> IResourceGroup * {
        for (const auto & [_, select_case] : select_cases)
        {
            if ((select_case.user == nullptr || std::regex_match(client_info.initial_user, *(select_case.user))))
                return select_case.group;
        }
        return nullptr;
    };

    IResourceGroup * user_group = get_user_group();
    LOG_DEBUG(logger, "user {} user_group {} children_size {}", 
            client_info.initial_user, user_group ? user_group->getName() : "not found", 
            user_group ? user_group->getChildren().size() : 0);
    if (user_group && !user_group->getChildren().empty())
    {
        size_t group_idx = 0;

        std::vector<IResourceGroup *> groups;
        const auto & children = user_group->getChildren();
        for (const auto & item: children)
        {
            groups.emplace_back(item.second);
        }

        const ASTSelectQuery * select_query = nullptr;
        if (ast->as<ASTSelectQuery>())
        {
            select_query = ast->as<ASTSelectQuery>();
        } 
        else if (ast->as<ASTSelectWithUnionQuery>())
        {
            const ASTSelectWithUnionQuery * union_ast = ast->as<ASTSelectWithUnionQuery>();
            if (union_ast->list_of_selects->children.size() == 1)
            {
                const ASTPtr first_select_ast = union_ast->list_of_selects->children.at(0);
                select_query = dynamic_cast<ASTSelectQuery *>(first_select_ast.get());
            }
        }

        std::optional<DatabaseAndTableWithAlias> database_and_table;
        if (select_query)
        {
            database_and_table = getDatabaseAndTable(*select_query, 0);
        }

        LOG_DEBUG(logger, "user_query {} user_group {} table {}", 
            select_query ? "select" : "not_select", 
            user_group->getName(), 
            database_and_table ? database_and_table->table : "empty");

        if (select_query && database_and_table.has_value())
        {
            auto table = database_and_table->table;