Spring Data JPA如何查询含嵌套列表的ChatDTO集合？

调整JPQL查询语句 #

使用Hibernate的COLLECT聚合函数将关联的用户ID和消息聚合为列表，配合分组查询确保每个聊天只返回一条记录，匹配ChatDTO的构造器参数：

@Query("SELECT new com.example.demo.dto.ChatDTO(gc.id, gc.nome, gc.dataCreazione, " +
       "COLLECT(DISTINCT ca.fkAccount), COLLECT(cm)) " +
       "FROM GroupChatEntity gc " +
       "JOIN ChatAccountEntity ca ON ca.fkChat = gc.id " +
       "LEFT JOIN ChatMessaggioEntity cm ON cm.fkChat = gc.id " +
       "WHERE ca.fkAccount = :idAccount " +
       "GROUP BY gc.id, gc.nome, gc.dataCreazione")
List<ChatDTO> findChatsByAccountId(@Param("idAccount") Long idAccount);

关键说明 #

1. 关联与过滤：直接通过JOIN ChatAccountEntity关联用户参与的聊天，并用WHERE ca.fkAccount = :idAccount过滤当前登录用户，替代原查询的子查询IN，逻辑更清晰。
2. 聚合列表：
   • COLLECT(DISTINCT ca.fkAccount)：将当前聊天的所有参与用户ID聚合为List<Long>，DISTINCT避免因笛卡尔积产生重复ID。
   • COLLECT(cm)：将当前聊天的所有消息聚合为List<ChatMessaggioEntity>。
3. 分组查询：通过GROUP BY gc.id, gc.nome, gc.dataCreazione确保每个聊天仅生成一条结果，对应一个ChatDTO实例。

确保DTO构造器匹配 #

你的ChatDTO需要提供对应参数的构造器，注意将utentiIds转换为ArrayList（如果属性定义是ArrayList<Long>）：

public ChatDTO(Long id, String nome, Date dataCreazione, List<Long> utentiIds, List<ChatMessaggioEntity> messaggi) {
    this.id = id;
    this.nome = nome;
    this.dataCreazione = dataCreazione;
    this.utentiIds = new ArrayList<>(utentiIds);
    this.messaggi = messaggi;
}

标准JPA兼容方案（非Hibernate） #

如果使用标准JPA而非Hibernate，COLLECT函数不支持，针对MySQL可以用GROUP_CONCAT聚合用户ID，再在DTO中拆分字符串为列表：

@Query("SELECT new com.example.demo.dto.ChatDTO(gc.id, gc.nome, gc.dataCreazione, " +
       "GROUP_CONCAT(DISTINCT ca.fkAccount SEPARATOR ','), COLLECT(cm)) " +
       "FROM GroupChatEntity gc " +
       "JOIN ChatAccountEntity ca ON ca.fkChat = gc.id " +
       "LEFT JOIN ChatMessaggioEntity cm ON cm.fkChat = gc.id " +
       "WHERE ca.fkAccount = :idAccount " +
       "GROUP BY gc.id, gc.nome, gc.dataCreazione")
List<ChatDTO> findChatsByAccountId(@Param("idAccount") Long idAccount);

同时修改ChatDTO构造器，将字符串拆分为ArrayList<Long>：

public ChatDTO(Long id, String nome, Date dataCreazione, String utentiIdsStr, List<ChatMessaggioEntity> messaggi) {
    this.id = id;
    this.nome = nome;
    this.dataCreazione = dataCreazione;
    this.utentiIds = Arrays.stream(utentiIdsStr.split(","))
                           .map(Long::parseLong)
                           .collect(Collectors.toCollection(ArrayList::new));
    this.messaggi = messaggi;
}

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