Elastic Search结合Wagtail API实现地理距离范围检索是否可行？

完全可以为Wagtail API配置Elasticsearch的地理距离（geo distance）特性，实现指定经纬度半径范围内的模型检索，完整实现流程如下：

1. 模型字段与搜索映射配置 #

首先给需要检索的模型添加经纬度存储字段，推荐用django.contrib.gis.db.models.PointField存储坐标，同时在Wagtail的搜索字段映射中指定Elasticsearch的字段类型为geo_point，适配地理查询：

from django.contrib.gis.db import models
from wagtail.search import index
from wagtail.models import Page

class LocationPage(Page):
    # 存储点位坐标，srid=4326对应WGS84通用经纬度坐标系
    coordinate = models.PointField(srid=4326, help_text="存储地点的经纬度坐标")
    address = models.CharField(max_length=255)
    city = models.CharField(max_length=100)

    # 配置搜索字段映射，指定coordinate为geo_point类型
    search_fields = Page.search_fields + [
        index.SearchField('city'),
        index.FilterField('coordinate', es_extra={'type': 'geo_point'}),
    ]

2. 自定义地理距离搜索查询 #

Wagtail默认的Elasticsearch后端没有内置地理距离过滤，你可以扩展官方的搜索查询类，注入geo_distance过滤逻辑：

# 若使用Elasticsearch8，替换为wagtail.search.backends.elasticsearch8下的对应类即可
from wagtail.search.backends.elasticsearch7 import Elasticsearch7SearchQuery
from django.contrib.gis.geos import Point

class GeoDistanceSearchQuery(Elasticsearch7SearchQuery):
    def __init__(self, *args, geo_point=None, radius_km=5, **kwargs):
        super().__init__(*args, **kwargs)
        self.geo_point = geo_point
        self.radius_km = radius_km

    def get_inner_query(self):
        # 继承原有查询逻辑，新增地理距离过滤条件
        query = super().get_inner_query()
        if self.geo_point and self.radius_km:
            query['bool']['filter'].append({
                'geo_distance': {
                    'distance': f'{self.radius_km}km',
                    'coordinate': {
                        'lat': self.geo_point.y,
                        'lon': self.geo_point.x
                    }
                }
            })
        return query

3. 对接Wagtail API实现业务逻辑 #

按照你预设的业务流程对接API即可：

• 接收用户传入的城市参数，调用地理编码服务将城市名转换为Point类型的经纬度坐标
• 调用自定义的GeoDistanceSearchQuery执行搜索，按需指定搜索半径
• 将搜索结果序列化后通过Wagtail API返回

示例API视图代码：

from wagtail.api.v2.views import PagesAPIViewSet
from .models import LocationPage
from .search import GeoDistanceSearchQuery
# geocode_city为你自行实现的地理编码方法，输入城市名返回对应Point坐标
from .utils import geocode_city

class LocationSearchAPIViewSet(PagesAPIViewSet):
    model = LocationPage

    def get_queryset(self):
        queryset = super().get_queryset()
        city = self.request.query_params.get('city')
        radius = int(self.request.query_params.get('radius', 5))
        if city:
            # 转换城市名为经纬度坐标
            target_point = geocode_city(city)
            if target_point:
                # 执行带地理过滤的搜索，无全文检索需求时关键词可留空
                queryset = LocationPage.search(
                    '',
                    query_class=GeoDistanceSearchQuery,
                    geo_point=target_point,
                    radius_km=radius
                )
        return queryset

注意：使用PointField需要你的数据库安装对应空间扩展，PostgreSQL需开启PostGIS扩展；配置完成后需要重新构建Wagtail搜索索引，地理过滤能力才能生效。

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