Spring Data JPA如何按投票集合大小排序并实现分页查询？

方案一：自定义JPQL查询（推荐） #

直接在RestaurantRepository中编写带投票数统计的JPQL查询，同时保留分页和搜索功能：

public interface RestaurantRepository extends PagingAndSortingRepository<Restaurant, Long> {
    @Query("SELECT r, COUNT(v) as voteCount FROM Restaurant r " +
           "LEFT JOIN r.votes v " +
           "WHERE r.name LIKE %:name% OR r.address LIKE %:address% " +
           "GROUP BY r " +
           "ORDER BY voteCount :sortDirection")
    Page<Object[]> findAllWithVoteCount(@Param("name") String name, 
                                        @Param("address") String address, 
                                        @Param("sortDirection") Sort.Direction sortDirection,
                                        Pageable pageable);
}

调用时需处理返回的Object[]数组，第一个元素是Restaurant实例，第二个元素为对应投票数：

// 示例：按投票数降序分页查询
Page<Object[]> resultPage = restaurantRepository.findAllWithVoteCount("汉堡", "北京", Sort.Direction.DESC, PageRequest.of(0, 10));
List<Restaurant> restaurants = resultPage.getContent().stream()
    .map(obj -> (Restaurant) obj[0])
    .collect(Collectors.toList());

注意：此处Pageable无需再指定排序字段，避免与JPQL中的排序逻辑冲突。

方案二：在实体中添加派生字段 #

给Restaurant实体添加一个通过@Formula注解映射的派生字段，直接从数据库层面统计投票数，这样就能像普通字段一样使用Sort.by()排序：

@Entity
@Table(name = "restaurants")
public class Restaurant extends BaseEntity {

    @OneToMany(mappedBy = "restaurant", fetch = FetchType.LAZY)
    @JsonManagedReference
    private List<Vote> votes;

    // 数据库层面统计当前餐厅的投票数
    @Formula("(SELECT COUNT(v.id) FROM votes v WHERE v.restaurant_id = id)")
    private Integer voteCount;

    // 仅需生成getter，无需setter
    public Integer getVoteCount() {
        return voteCount;
    }
}

之后可直接用该字段排序，原有Repository方法无需修改：

Pageable pageable = PageRequest.of(0, 10, Sort.by("voteCount").descending());
Page<Restaurant> restaurants = restaurantRepository.findAllByNameContainsOrAddressContains("汉堡", "北京", pageable);

⚠️ 注意：@Formula是Hibernate专属注解，若使用其他JPA实现可能不兼容。

方案三：使用Criteria API动态构建查询 #

如果需要更灵活的动态查询逻辑（比如支持切换排序字段），可结合JpaSpecificationExecutor和Criteria API实现：

首先让Repository继承JpaSpecificationExecutor：

public interface RestaurantRepository extends PagingAndSortingRepository<Restaurant, Long>, JpaSpecificationExecutor<Restaurant> {
}

然后构建Specification处理搜索与排序逻辑：

public Specification<Restaurant> getRestaurantSpecification(String name, String address, String sortField, Sort.Direction direction) {
    return (root, query, cb) -> {
        List<Predicate> predicates = new ArrayList<>();
        // 处理名称搜索条件
        if (name != null && !name.isEmpty()) {
            predicates.add(cb.like(root.get("name"), "%" + name + "%"));
        }
        // 处理地址搜索条件
        if (address != null && !address.isEmpty()) {
            predicates.add(cb.like(root.get("address"), "%" + address + "%"));
        }

        // 处理投票数排序
        if ("voteCount".equals(sortField)) {
            Join<Restaurant, Vote> voteJoin = root.join("votes", JoinType.LEFT);
            Expression<Long> voteCount = cb.count(voteJoin);
            query.groupBy(root.get("id"));
            query.orderBy(direction == Sort.Direction.ASC ? cb.asc(voteCount) : cb.desc(voteCount));
        } else {
            // 按普通字段排序
            query.orderBy(direction == Sort.Direction.ASC ? cb.asc(root.get(sortField)) : cb.desc(root.get(sortField)));
        }

        return cb.or(predicates.toArray(new Predicate[0]));
    };
}

调用示例：

Specification<Restaurant> spec = getRestaurantSpecification("汉堡", "北京", "voteCount", Sort.Direction.DESC);
Page<Restaurant> restaurants = restaurantRepository.findAll(spec, PageRequest.of(0, 10));

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