如何使用Spring Data JPA选择关联@Entity的指定属性子集，避免冗余查询

针对你的需求，以下两种方案均无需引入第三方依赖，可精准控制查询字段，避免N+1问题： #

方案1：构造器表达式+数据库聚合（性能最优，适配PostgreSQL） #

该方案完全对齐你给出的原生SQL逻辑，单次查询即可返回所有需要的字段，无多余数据传输。

步骤1：定义DTO类 #

需提供全参数构造器，可使用Lombok简化代码：

// 扁平化返回可直接用该类，无需嵌套PageDTO
@Getter
@AllArgsConstructor
public class BookCustomDTO {
    // 仅声明需要返回的字段
    private String first;
    private List<String> pageContents;
}

步骤2：编写Repository查询方法 #

使用FUNCTION调用PostgreSQL原生的array_agg函数实现聚合，仅查询指定字段：

@Repository
public interface BookRepository extends JpaRepository<Book, Long> {
    @Query("SELECT new com.your.package.BookCustomDTO(" +
            "b.first, " +
            "FUNCTION('array_agg', p.content)" +
            ") FROM Book b LEFT JOIN Page p ON p.book.id = b.id " +
            // 按book.id分组避免同first值的不同book聚合错误
            "GROUP BY b.id, b.first")
    List<BookCustomDTO> queryAllBookWithPageContent();
}

方案2：扁平化查询+内存聚合（全数据库兼容） #

如果需要兼容多种数据库，不想使用数据库特定的聚合函数，可采用单次查询拉取所需字段后在业务层聚合，性能也远高于N+1查询：

步骤1：编写基础查询 #

仅查询需要的first和content字段：

@Repository
public interface BookRepository extends JpaRepository<Book, Long> {
    @Query("SELECT b.first, p.content FROM Book b LEFT JOIN Page p ON p.book.id = b.id")
    List<Object[]> queryRawBookPageFields();
}

步骤2：业务层聚合结果 #

@Service
public class BookService {
    @Autowired
    private BookRepository bookRepository;

    public List<BookCustomDTO> getBookWithPageContent() {
        List<Object[]> rawData = bookRepository.queryRawBookPageFields();
        // 实际业务建议按book.id分组，避免同first值的不同book聚合错误
        Map<String, List<String>> bookPageMap = rawData.stream()
                .filter(item -> item[1] != null)
                .collect(Collectors.groupingBy(
                        item -> (String) item[0],
                        Collectors.mapping(item -> (String) item[1], Collectors.toList())
                ));
        // 补充无关联Page的Book
        rawData.stream()
                .map(item -> (String) item[0])
                .distinct()
                .forEach(first -> bookPageMap.putIfAbsent(first, Collections.emptyList()));
        // 转换为DTO返回
        return bookPageMap.entrySet().stream()
                .map(entry -> new BookCustomDTO(entry.getKey(), entry.getValue()))
                .collect(Collectors.toList());
    }
}

多关联实体扩展说明 #

针对你提到的3-4个关联实体的场景，两种方案都可以轻松扩展：

• 关联字段直接在SELECT语句中显式声明即可，不会查询多余字段
• 多集合关联建议用方案2的内存聚合，避免数据库多层嵌套聚合的性能损耗
• 所有查询均为单次SQL执行，无N+1问题

内容的提问来源于stack exchange，提问作者Ivan Denchev