JPA中@ElementCollection集合字段countBy查询异常的解决疑问

想要统计被@ElementCollection注解标注的Set<String>类型字段companyIds中包含指定字符串的User实体数量。目前在JpaRepository<User, String>中，通过List<User> findByCompanyIds(String companyId)获取结果后调用.size()可以正常实现统计，但使用long countByCompanyIds(String companyId)或long countByCompanyIdsContaining(String companyId)时均抛出QueryException异常。询问是否需要编写自定义查询，还是count查询的方式有误？

实体类代码 #

@Entity
@Table(name = "user")
public class User {
    @Id
    @Column(name = "id", nullable = false)
    private String id;

    @ElementCollection(fetch = FetchType.EAGER)
    @CollectionTable(name = "companyId",
            joinColumns = @JoinColumn(name = "id"),
            indexes = {
                    @Index(name = "company_ids_index", columnList = "company_ids")
            }
    )
    @Column(name = "company_ids")
    private Set<String> companyIds;

    public User() {
        companyIds = new HashSet<>();
    }
}

Repository代码 #

public interface UserDao extends JpaRepository<User, String> {
    /* 可以正常工作 */
    @EntityGraph(attributePaths = {"companyIds"})
    List<User> findByCompanyIds(String companyId);

    /* 无法工作 */
    @EntityGraph(attributePaths = {"companyIds"})
    long countByCompanyIds(String companyId);

    /* 无法工作（由@Nick提出的方案） */
    @EntityGraph(attributePaths = {"companyIds"})
    long countByCompanyIdsContaining(String companyId);
}

异常栈信息 #

org.springframework.dao.InvalidDataAccessApiUsageException: 
 org.hibernate.QueryException: query specified join fetching, but the owner of the fetched association was not present in the select list 
 [FromElement{explicit,collection join,fetch join,fetch non-lazy properties,classAlias=generatedAlias1,role=com.test.User.companyIds,
 tableName={none},tableAlias=companyids1_,origin=user user0_,columns={,className=null}}] [select count(generatedAlias0) 
 from com.test.User as generatedAlias0 left join generatedAlias0.companyIds as generatedAlias1 where generatedAlias1=:param0]

问题原因 #

异常提示明确：查询指定了join fetch，但被抓取关联的所有者（User实体）不在查询的select列表中。核心问题是给count查询添加了@EntityGraph(attributePaths = {"companyIds"})注解——count查询仅需统计符合条件的User数量，不需要抓取关联的companyIds集合，该注解会强制Hibernate执行join fetch操作，导致SQL逻辑冲突。

解决方法 #

1. 移除count方法上的@EntityGraph注解
   count查询无需加载关联集合，直接移除注解即可。修改后的Repository代码如下：

public interface UserDao extends JpaRepository<User, String> {
    /* 可以正常工作 */
    @EntityGraph(attributePaths = {"companyIds"})
    List<User> findByCompanyIds(String companyId);

    /* 修复后可正常工作 */
    long countByCompanyIds(String companyId);
}

此时Spring Data JPA会生成正确的count查询SQL，不会触发异常。

2. 自定义JPQL查询（可选）
   若想更直观地控制查询逻辑，可使用@Query注解手动编写JPQL：

@Query("SELECT COUNT(u) FROM User u JOIN u.companyIds c WHERE c = :companyId")
long countByCompanyId(@Param("companyId") String companyId);

这种方式能避免Spring Data JPA自动生成查询时的潜在问题。

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