如何使用JPA Specification实现针对指定字段的Distinct查询

我明白你遇到的问题了——当你想用JPA Specification查询去重的name字段时，结果却生成了包含所有实体字段的SQL，distinct作用在了整个实体上而不是只name。这背后的原因其实和你使用的findAll方法返回类型有关，下面我来一步步帮你解决这个问题：

问题根源 #

你当前的代码使用hcpRepository.findAll(Specification<HCP>)，这个方法的返回类型是List<HCP>。JPA需要构建完整的HCP实体对象，而HCP带有@Id注解的id字段，所以Hibernate必须查询所有实体字段（包括id、area）来创建对象。这就导致你在Specification中设置的query.select(root.get("name"))被覆盖，distinct最终作用在整行数据上，而不是单独的name字段。

解决方案 #

要实现只查询去重的name字段，我们需要改变查询的返回类型（从HCP实体改为String），并构建对应的CriteriaQuery。这里有三种常用的实现方式：

方式一：自定义Repository方法（推荐） #

首先在你的HcpRepository中添加一个自定义方法，利用EntityManager手动构建返回String类型的查询：

public interface HcpRepository extends JpaRepository<HCP, String>, JpaSpecificationExecutor<HCP> {

    default List<String> findDistinctNames(Specification<HCP> spec) {
        EntityManager em = getEntityManager();
        CriteriaBuilder cb = em.getCriteriaBuilder();
        // 创建返回String类型的CriteriaQuery
        CriteriaQuery<String> cq = cb.createQuery(String.class);
        Root<HCP> root = cq.from(HCP.class);
        
        // 指定查询name字段并开启distinct
        cq.select(root.get("name")).distinct(true);
        
        // 应用Specification中的查询条件
        Predicate predicate = spec.toPredicate(root, cq, cb);
        if (predicate != null) {
            cq.where(predicate);
        }
        
        return em.createQuery(cq).getResultList();
    }
}

然后调用这个方法即可得到预期的结果：

List<String> distinctNames = hcpRepository.findDistinctNames((root, query, cb) -> 
    cb.equal(root.get("area"), "Dhaka")
);

这个方法会生成你期望的SQL：

select distinct hcp0_.name as name3_0_ from hcp hcp0_ where hcp0_.area=?

方式二：直接使用EntityManager构建查询 #

如果你不想修改Repository接口，也可以直接在业务代码中注入EntityManager来构建查询：

@Autowired
private EntityManager em;

// ...

public List<String> getDistinctHcpNamesInDhaka() {
    CriteriaBuilder cb = em.getCriteriaBuilder();
    CriteriaQuery<String> cq = cb.createQuery(String.class);
    Root<HCP> root = cq.from(HCP.class);
    
    cq.select(root.get("name")).distinct(true);
    cq.where(cb.equal(root.get("area"), "Dhaka"));
    
    return em.createQuery(cq).getResultList();
}

方式三：使用Spring Data JPA投影 #

另外，你也可以通过投影接口来实现类似需求，适合需要提取多个字段的场景：

1. 定义一个投影接口，只包含你需要的字段：

public interface HcpNameProjection {
    String getName();
}

2. 在Repository中添加方法：

List<HcpNameProjection> findDistinctByArea(String area);

调用这个方法后，你可以通过流操作提取name字段：

List<String> distinctNames = hcpRepository.findDistinctByArea("Dhaka")
    .stream()
    .map(HcpNameProjection::getName)
    .collect(Collectors.toList());

这种方式生成的SQL同样只会查询name字段并应用distinct。

内容的提问来源于stack exchange，提问作者Hafizur Rahman Arfin