如何在Spring JPA Repository中注解式使用Hibernate ResultTransformer？

问题背景 #

Vlad Mihalcea在《How to fetch a one-to-many DTO projection with JPA and Hibernate》一文中给出了基于EntityManager+Hibernate ResultTransformer的一对多DTO投影实现：

List<PostDTO> postDTOs = entityManager.createQuery("""
    select p.id as p_id,
           p.title as p_title,
           pc.id as pc_id,
           pc.review as pc_review
    from PostComment pc
    join pc.post p
    order by pc.id
    """)
.unwrap(org.hibernate.query.Query.class)
.setResultTransformer(new PostDTOResultTransformer())
.getResultList();

希望能在Spring Data JpaRepository中通过注解驱动的方式实现相同逻辑，避免直接依赖EntityManager，比如期望写法：

// 注：不存在@ResultTransformer注解，仅为示例
@ResultTransformer(PostDTOResultTransformer.class)
@Query("""
        select p.id as p_id,
               p.title as p_title,
               pc.id as pc_id,
               pc.review as pc_review
        from PostComment pc
        join pc.post p
        order by pc.id
        """)
List<PostDTO> findAllWithComments();

已知@SqlResultSetMapping仅支持构造函数映射，无法处理一对多集合聚合，询问是否有注解驱动方案，或是否必须依赖EntityManager。

可行方案 #

1. 自定义Repository扩展（推荐） #

通过Spring Data的自定义Repository机制，将EntityManager+ResultTransformer的逻辑封装在Repository实现中，对外仍通过标准Repository接口调用：

步骤1：定义自定义Repository接口

public interface PostRepositoryCustom {
    List<PostDTO> findAllWithComments();
}

步骤2：实现自定义接口

注入EntityManager，封装原有的查询逻辑：

public class PostRepositoryImpl implements PostRepositoryCustom {
    private final EntityManager entityManager;

    // 构造函数注入EntityManager
    public PostRepositoryImpl(EntityManager entityManager) {
        this.entityManager = entityManager;
    }

    @Override
    public List<PostDTO> findAllWithComments() {
        return entityManager.createQuery("""
                select p.id as p_id,
                       p.title as p_title,
                       pc.id as pc_id,
                       pc.review as pc_review
                from PostComment pc
                join pc.post p
                order by pc.id
                """)
                .unwrap(org.hibernate.query.Query.class)
                .setResultTransformer(new PostDTOResultTransformer())
                .getResultList();
    }
}

步骤3：让主Repository继承自定义接口

public interface PostRepository extends JpaRepository<Post, Long>, PostRepositoryCustom {
}

之后即可通过postRepository.findAllWithComments()直接调用，业务代码无需感知EntityManager的存在。

2. Tuple查询+自定义Converter（无Hibernate API依赖） #

如果想避免依赖Hibernate的ResultTransformer，可以先查询返回Tuple列表，再通过自定义Converter手动实现一对多聚合逻辑：

步骤1：在Repository中定义Tuple查询

@Query("""
        select p.id as p_id,
               p.title as p_title,
               pc.id as pc_id,
               pc.review as pc_review
        from PostComment pc
        join pc.post p
        order by pc.id
        """)
List<Tuple> findAllCommentPostTuples();

步骤2：实现Converter组件

@Component
public class TupleToPostDTOConverter {
    public List<PostDTO> convert(List<Tuple> tuples) {
        // 手动实现ResultTransformer的聚合逻辑，合并同一Post的多个Comment
        Map<Long, PostDTO> postMap = new LinkedHashMap<>();
        
        for (Tuple tuple : tuples) {
            Long postId = tuple.get("p_id", Long.class);
            // 不存在则创建新的PostDTO
            PostDTO postDto = postMap.computeIfAbsent(postId, id -> {
                PostDTO dto = new PostDTO();
                dto.setId(id);
                dto.setTitle(tuple.get("p_title", String.class));
                dto.setComments(new ArrayList<>());
                return dto;
            });
            // 构建CommentDTO并添加到对应Post的集合中
            PostCommentDTO commentDto = new PostCommentDTO();
            commentDto.setId(tuple.get("pc_id", Long.class));
            commentDto.setReview(tuple.get("pc_review", String.class));
            postDto.getComments().add(commentDto);
        }
        
        return new ArrayList<>(postMap.values());
    }
}

步骤3：业务代码中调用转换

List<Tuple> tuples = postRepository.findAllCommentPostTuples();
List<PostDTO> postDTOs = tupleToPostDTOConverter.convert(tuples);

关于注解驱动的直接实现 #

目前Spring Data JPA和JPA标准中，没有类似@ResultTransformer的注解可以直接在@Query上绑定Hibernate的ResultTransformer。@SqlResultSetMapping仅支持基于单行数据的构造函数映射、字段映射，无法处理一对多数据的聚合合并——这种逻辑需要手动处理行数据的分组，正是Hibernate ResultTransformer的核心作用。

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