JPA查询如何基于部分复合主键字段实现高效数据筛选

以下是两种性能最优的实现方案，都可以实现一次数据库请求查询多组name+location匹配的记录：

方案1：JPQL多列IN查询（首选） #

主流关系型数据库均支持多列IN语法，直接通过@Query自定义JPQL即可实现，写法简单高效：

仓库接口定义 #

interface EmployeeJpaRepository extends JpaRepository<Employee, EmployeeId> {
    @Query("SELECT e FROM Employee e WHERE (e.id.name, e.id.location) IN :nameLocationPairs")
    List<Employee> findAllByNameLocationPairs(List<Object[]> nameLocationPairs);
}

调用示例 #

// 构造所有需要查询的(name, location)组合
List<Object[]> queryPairs = new ArrayList<>();
queryPairs.add(new Object[]{"张三", "北京"});
queryPairs.add(new Object[]{"李四", "上海"});
queryPairs.add(new Object[]{"王五", "深圳"});

// 仅发起一次数据库请求，返回所有匹配的记录
List<Employee> resultList = employeeRepository.findAllByNameLocationPairs(queryPairs);

注意：此方案兼容MySQL、PostgreSQL、Oracle、SQL Server 2008+等绝大多数主流数据库

方案2：JPA Specification动态拼接（兼容性更强） #

如果数据库不支持多列IN语法，或者需要动态拼接查询条件，可以继承JpaSpecificationExecutor实现：

仓库接口定义 #

interface EmployeeJpaRepository extends JpaRepository<Employee, EmployeeId>, JpaSpecificationExecutor<Employee> {
}

查询逻辑实现 #

// 待查询的(name, location)组合列表
List<Map<String, String>> queryPairs = List.of(
    Map.of("name", "张三", "location", "北京"),
    Map.of("name", "李四", "location", "上海")
);

// 构造查询规则
Specification<Employee> spec = (root, query, criteriaBuilder) -> {
    List<Predicate> pairPredicates = new ArrayList<>();
    for (Map<String, String> pair : queryPairs) {
        pairPredicates.add(criteriaBuilder.and(
            criteriaBuilder.equal(root.get("id").get("name"), pair.get("name")),
            criteriaBuilder.equal(root.get("id").get("location"), pair.get("location"))
        ));
    }
    return criteriaBuilder.or(pairPredicates.toArray(new Predicate[0]));
};

// 一次数据库请求返回所有匹配结果
List<Employee> resultList = employeeRepository.findAll(spec);

批量查询优化建议 #

如果单次查询的name+location组合数量超过1000组，建议按每500~1000组拆分批量查询，避免SQL长度超出数据库限制导致报错。

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