Spring Data JPA：按多列与参数计算的距离值排序查询Point实体

现有如下Point实体类：

@Entity
@NoArgsConstructor
@Data
public class Point{
    @Id
    @GeneratedValue
    private Long id;

    private double x;
    private double y;
}

需要在PointRepository中实现方法：

List<Point> findAllOrderByDistanceFromPointAsc(double a, double b);

距离计算逻辑为：

double distance = Math.sqrt( Math.pow((a-x), 2) + Math.pow((b-y), 2) )

要求尽量不使用原生查询，可通过CriteriaQuery、Specification实现，需要具体操作步骤。

方法一：通过Specification实现 #

这种方式适合需要灵活组合查询条件的场景，步骤如下：

1. 让PointRepository继承JpaSpecificationExecutor<Point>，开启Specification支持：

public interface PointRepository extends JpaRepository<Point, Long>, JpaSpecificationExecutor<Point> {
}

2. 编写Specification类，构建距离排序逻辑：

public class PointSpecifications {
    public static Specification<Point> orderByDistanceFromPointAsc(double a, double b) {
        return (root, query, criteriaBuilder) -> {
            // 计算距离的平方（省略sqrt不影响排序结果，提升性能）
            Expression<Double> distanceSquared = criteriaBuilder.sum(
                criteriaBuilder.pow(criteriaBuilder.diff(a, root.get("x")), 2),
                criteriaBuilder.pow(criteriaBuilder.diff(b, root.get("y")), 2)
            );
            // 设置升序排序
            query.orderBy(criteriaBuilder.asc(distanceSquared));
            return criteriaBuilder.conjunction(); // 返回所有数据，无过滤条件
        };
    }
}

3. 调用查询：

List<Point> points = pointRepository.findAll(PointSpecifications.orderByDistanceFromPointAsc(a, b));

方法二：自定义Repository方法+CriteriaQuery实现 #

如果希望直接在Repository中调用目标方法，无需额外传入Specification，可采用自定义Repository实现：

1. 定义Repository接口和自定义方法接口：

// 主Repository接口
public interface PointRepository extends JpaRepository<Point, Long>, PointRepositoryCustom {
}

// 自定义方法接口
public interface PointRepositoryCustom {
    List<Point> findAllOrderByDistanceFromPointAsc(double a, double b);
}

2. 编写自定义实现类（命名需遵循「接口名+Impl」规则）：

@Repository
public class PointRepositoryImpl implements PointRepositoryCustom {
    @PersistenceContext
    private EntityManager entityManager;

    @Override
    public List<Point> findAllOrderByDistanceFromPointAsc(double a, double b) {
        CriteriaBuilder cb = entityManager.getCriteriaBuilder();
        CriteriaQuery<Point> query = cb.createQuery(Point.class);
        Root<Point> root = query.from(Point.class);

        // 构建距离平方表达式
        Expression<Double> distanceSquared = cb.sum(
            cb.pow(cb.diff(a, root.get("x")), 2),
            cb.pow(cb.diff(b, root.get("y")), 2)
        );

        // 按距离平方升序排序
        query.orderBy(cb.asc(distanceSquared));

        // 执行查询并返回结果
        return entityManager.createQuery(query).getResultList();
    }
}

3. 直接调用目标方法：

List<Point> points = pointRepository.findAllOrderByDistanceFromPointAsc(a, b);

关键说明 #

• 省略平方根计算：因为平方根是单调递增函数，sqrt(d1²) < sqrt(d2²)等价于d1² < d2²，省略sqrt能减少运算开销，提升查询性能，且排序结果完全一致。
• 两种方案均基于JPA标准API实现，未使用原生SQL，符合需求。

内容的提问来源于stack exchange，提问作者soumik.saha.lion5