Spring Boot WebFlux中关联User与Role实体避免多次查询的问题

你当前的代码会触发N+1查询（N为用户数量），每个用户都单独发起一次角色查询，这是明显的性能瓶颈。正确的思路是先批量获取所有用户，再一次性查询所有关联的角色，最后在内存中完成用户与角色的匹配，全程只需要两次数据库查询。

直接上优化后的代码：

// 第一步：查询所有用户
Flux<User> usersFlux = userRepository.findAll();

return usersFlux
    // 将User实体转为UserDto
    .map(userMapper::userToUserDto)
    // 收集为List，方便后续批量处理
    .collectList()
    .flatMapMany(userDtos -> {
        // 提取所有非空的角色ID，去重避免重复查询
        List<Long> roleIds = userDtos.stream()
            .map(UserDto::getRoleId)
            .filter(Objects::nonNull)
            .distinct()
            .collect(Collectors.toList());

        // 如果没有需要查询的角色ID，直接返回用户列表
        if (roleIds.isEmpty()) {
            return Flux.fromIterable(userDtos);
        }

        // 第二步：批量查询所有角色，转为以角色ID为key的Map，方便快速匹配
        return roleRepository.findByIdIn(roleIds)
            .collectMap(RoleDto::getId)
            .flatMapMany(roleMap -> {
                // 第三步：为每个用户匹配对应的角色
                userDtos.forEach(userDto -> {
                    userDto.setRole(roleMap.get(userDto.getRoleId()));
                });
                return Flux.fromIterable(userDtos);
            });
    });

优化点说明 #

1. 彻底消除N+1查询：仅执行两次数据库操作——一次查询所有用户，一次批量查询所有关联角色，无论用户数量多少，角色查询只执行一次。
2. 内存关联效率更高：将查询到的角色转为Map<Long, RoleDto>，通过角色ID直接匹配用户，时间复杂度为O(N)，比循环匹配更高效。
3. 边界情况处理：过滤空角色ID，同时处理无角色关联的场景，避免无效查询。

原代码的问题所在 #

你原代码中的flatMap是对每个用户单独处理，每个用户都会触发一次角色查询，等于用户数量有多少，角色SQL就会执行多少次，这种N+1查询在用户量较大时会导致性能急剧下降。

内容的提问来源于stack exchange，提问作者Kuldeep KV