如何使用JpaRepository实现findObjectsNearLocation()方法

嘿，我来帮你搞定这个基于地理位置的查询需求！不用写原生SQL也能实现，咱们用Spring Data JPA结合PostgreSQL的空间支持就能搞定，一步步来：

第一步：先把依赖配好 #

首先你的项目需要引入支持地理空间查询的依赖，如果你用Maven，把这些加到pom.xml里：

<!-- Spring Data JPA核心依赖 -->
<dependency>
    <groupId>org.springframework.data</groupId>
    <artifactId>spring-data-jpa</artifactId>
    <version>你的Spring Data版本</version>
</dependency>
<!-- Hibernate空间扩展，支持PostgreSQL地理类型 -->
<dependency>
    <groupId>org.hibernate.orm</groupId>
    <artifactId>hibernate-spatial</artifactId>
    <version>你的Hibernate版本</version>
</dependency>
<!-- PostgreSQL驱动 -->
<dependency>
    <groupId>org.postgresql</groupId>
    <artifactId>postgresql</artifactId>
    <version>最新稳定版</version>
</dependency>
<!-- JTS地理工具库，处理Point/Circle等地理对象 -->
<dependency>
    <groupId>org.locationtech.jts</groupId>
    <artifactId>jts-core</artifactId>
    <version>1.19.0</version>
</dependency>

第二步：修正你的ObjectEntity的Location字段 #

你原来的Location属性需要改成能映射PostgreSQLPOINT类型的字段，用JTS的Point类型来表示，这样JPA才能识别地理坐标：

import jakarta.persistence.*;
import org.locationtech.jts.geom.Point;
import serverApp.classes.common.ObjectID;

@Entity
@Table(name = "objects")
public class ObjectEntity {
    @Id
    private ObjectID id;
    
    // 关键：用JTS的Point映射PostgreSQL的POINT字段
    @Column(columnDefinition = "POINT")
    private Point location;
    
    // 其他属性、getter、setter...
}

⚠️ 重要提醒：PostgreSQL的POINT存储顺序是经度（lng）在前，纬度（lat）在后，创建Point对象时别搞反：

// 用坐标工厂创建Point，参数顺序是(lng, lat)
GeometryFactory factory = new GeometryFactory();
Point location = factory.createPoint(new Coordinate(lng, lat));

第三步：编写JpaRepository的查询方法 #

这里有两种不用原生SQL的实现方式，按需选：

方式1：Spring Data JPA派生查询（最省心） #

Spring Data JPA支持空间查询的派生方法，直接定义方法签名就能自动生成逻辑，比如用within结合Circle类型：

import org.locationtech.jts.geom.Circle;
import org.springframework.data.jpa.repository.JpaRepository;
import java.util.List;

public interface ObjectRepository extends JpaRepository<ObjectEntity, ObjectID> {
    // 查找所有location在指定圆形范围内的对象
    List<ObjectEntity> findByLocationWithin(Circle circle);
}

业务层调用时，把传入的lat、lng、radius转换成Circle：

// 注意：如果radius单位是米，需要转成度（WGS84坐标系下1度≈111319米）
double radiusInDegrees = radius / 111319.0;
GeometryFactory factory = new GeometryFactory();
Point center = factory.createPoint(new Coordinate(lng, lat));
Circle searchCircle = new Circle(center, radiusInDegrees);

// 调用Repository方法
List<ObjectEntity> nearObjects = objectRepository.findByLocationWithin(searchCircle);

方式2：JPQL结合空间函数（更灵活） #

如果想明确控制查询逻辑，用@Query写JPQL，调用Hibernate支持的st_dwithin空间函数（需要PostGIS扩展支持）：

import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.jpa.repository.Query;
import org.springframework.data.repository.query.Param;
import java.util.List;

public interface ObjectRepository extends JpaRepository<ObjectEntity, ObjectID> {
    @Query("SELECT o FROM ObjectEntity o WHERE st_dwithin(o.location, :center, :radius) = true")
    List<ObjectEntity> findObjectsNearLocation(
            @Param("center") org.locationtech.jts.geom.Point center,
            @Param("radius") double radius);
}

这种方式下，如果用PostGIS，radius单位可以直接用米（更准确），不用转度，前提是数据库已经安装PostGIS扩展：

-- 在PostgreSQL中执行开启PostGIS
CREATE EXTENSION postgis;

关于你提到的GeoResult #

如果不仅需要返回附近对象，还想得到每个对象到中心点的距离，可以用Spring Data的GeoResult封装结果，不过需要自己处理下查询结果的映射，比如：

import org.springframework.data.domain.Page;
import org.springframework.data.domain.Pageable;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.jpa.repository.Query;
import org.springframework.data.repository.query.Param;

public interface ObjectRepository extends JpaRepository<ObjectEntity, ObjectID> {
    @Query(value = "SELECT o, st_distance(o.location, :center) AS distance FROM ObjectEntity o WHERE st_dwithin(o.location, :center, :radius) = true ORDER BY distance",
           countQuery = "SELECT COUNT(o) FROM ObjectEntity o WHERE st_dwithin(o.location, :center, :radius) = true")
    Page<ObjectEntity> findObjectsNearLocationWithDistance(
            @Param("center") org.locationtech.jts.geom.Point center,
            @Param("radius") double radius,
            Pageable pageable);
}

之后可以把返回的Page数据手动封装成GeoResult列表，满足带距离的查询需求。

备注：内容来源于stack exchange，提问作者SahiBalata