如何在Java Spring+PostgreSQL中实现MongoDB式的半径内地理空间查询？

结合你之前MongoDB+Express的实现经验，下面给出两种在Spring Boot+PostgreSQL环境下实现半径查询的具体方案，涵盖代码实现和优化细节：

方案一：基于Haversine公式实现（无需额外扩展） #

这种方式不需要安装PostgreSQL扩展，直接通过SQL公式计算两点间距离，适合小型数据集场景。

1. 实体类定义 #

首先定义包含经纬度字段的实体：

import jakarta.persistence.Entity;
import jakarta.persistence.GeneratedValue;
import jakarta.persistence.GenerationType;
import jakarta.persistence.Id;
import lombok.Data;

@Entity(name = "location_entity") // 对应数据库表名
@Data
public class LocationEntity {
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;
    private Double latitude; // 纬度
    private Double longitude; // 经度
    // 其他业务字段（如名称、描述等）
}

2. Repository层实现 #

通过Spring Data JPA的原生SQL注解实现查询：

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 LocationRepository extends JpaRepository<LocationEntity, Long> {

    /**
     * 查询指定中心点半径内的所有对象
     * @param centerLat 中心点纬度
     * @param centerLng 中心点经度
     * @param radiusKm 半径（单位：公里，若前端传米需转成公里）
     * @return 符合条件的实体列表
     */
    @Query(value = "SELECT * FROM location_entity " +
            "WHERE (6371 * acos(cos(radians(:centerLat)) * cos(radians(latitude)) * " +
            "cos(radians(longitude) - radians(:centerLng)) + sin(radians(:centerLat)) * sin(radians(latitude)))) < :radiusKm",
            nativeQuery = true)
    List<LocationEntity> findWithinRadius(@Param("centerLat") Double centerLat,
                                          @Param("centerLng") Double centerLng,
                                          @Param("radiusKm") Double radiusKm);
}

3. 服务层与控制器调用 #

import org.springframework.stereotype.Service;
import java.util.List;

@Service
public class LocationService {
    private final LocationRepository locationRepository;

    public LocationService(LocationRepository locationRepository) {
        this.locationRepository = locationRepository;
    }

    // 接收前端传入的米级半径，转成公里后查询
    public List<LocationEntity> getLocationsInRadius(Double centerLat, Double centerLng, Double radiusMeter) {
        Double radiusKm = radiusMeter / 1000.0;
        return locationRepository.findWithinRadius(centerLat, centerLng, radiusKm);
    }
}

控制器示例：

import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RestController;
import java.util.List;

@RestController
public class LocationController {
    private final LocationService locationService;

    public LocationController(LocationService locationService) {
        this.locationService = locationService;
    }

    @PostMapping("/locations/radius")
    public ResponseEntity<List<LocationEntity>> queryByRadius(@RequestBody RadiusQueryRequest request) {
        List<LocationEntity> result = locationService.getLocationsInRadius(request.getLat(), request.getLng(), request.getRadius());
        return ResponseEntity.ok(result);
    }

    // 请求参数DTO
    static class RadiusQueryRequest {
        private Double lat;
        private Double lng;
        private Double radius; // 单位：米

        // getter & setter
        public Double getLat() { return lat; }
        public void setLat(Double lat) { this.lat = lat; }
        public Double getLng() { return lng; }
        public void setLng(Double lng) { this.lng = lng; }
        public Double getRadius() { return radius; }
        public void setRadius(Double radius) { this.radius = radius; }
    }
}

方案二：基于PostgreSQL earthdistance扩展（高效查询） #

如果你的数据集较大，推荐使用PostgreSQL的earthdistance扩展，它结合空间索引能大幅提升查询效率，类似MongoDB的2dsphere索引效果。

1. 安装PostgreSQL扩展 #

首先在数据库中执行以下SQL安装扩展：

CREATE EXTENSION earthdistance;
CREATE EXTENSION postgis; -- 可选，支持更复杂的地理空间操作

2. 优化后的Repository查询 #

利用earth_box先做范围过滤，再用earth_distance精确计算距离，效率远高于纯Haversine公式：

@Query(value = "SELECT * FROM location_entity " +
        "WHERE earth_box(ll_to_earth(:centerLat, :centerLng), :radiusMeter) @> ll_to_earth(latitude, longitude) " +
        "AND earth_distance(ll_to_earth(:centerLat, :centerLng), ll_to_earth(latitude, longitude)) < :radiusMeter",
        nativeQuery = true)
List<LocationEntity> findWithinRadiusByEarthDistance(@Param("centerLat") Double centerLat,
                                                     @Param("centerLng") Double centerLng,
                                                     @Param("radiusMeter") Double radiusMeter);

3. 创建空间索引提升性能 #

在数据库中为经纬度字段创建基于ll_to_earth的GIST索引：

CREATE INDEX idx_location_earth_point ON location_entity USING gist(ll_to_earth(latitude, longitude));

两种方案对比 #

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