基于elasticsearch_dsl按地理位置距离搜索门店并获取距离值

当然可以直接让Elasticsearch返回它计算好的距离值，完全不需要你手动处理经纬度来计算，而且用elasticsearch_dsl一次查询就能全部搞定！下面我给你详细说怎么做：

前提条件 #

首先要确保你的索引里存储门店位置的字段是geo_point类型（比如字段名为location），Elasticsearch只有针对这种类型才能进行地理距离计算。

具体实现步骤 #

1. 定义文档类（如果还没定义的话） #

先把你的门店文档结构和索引关联起来：

from elasticsearch_dsl import Document, GeoPoint, Index

# 替换成你的索引名称
store_index = Index('store_locations')

@store_index.document
class Store(Document):
    location = GeoPoint()  # 存储门店经纬度，格式可以是 {"lat": 纬度值, "lon": 经度值}
    name = Text()  # 门店名称
    address = Text()  # 门店地址
    # 其他你需要的字段...

2. 构造单次查询：过滤、排序、返回距离 #

下面的代码会同时完成三个需求：

• 筛选出指定位置附近的门店（可选，你可以去掉过滤只做排序）
• 按距离从近到远排序
• 将Elasticsearch计算好的距离纳入返回结果

from elasticsearch_dsl import Search, Q

# 替换成你要查询的目标位置
target_coords = {"lat": 40.7128, "lon": -74.0060}  # 示例：纽约市中心经纬度

# 初始化搜索对象
s = Search(index='store_locations')

# 1. 可选：过滤出目标位置100公里范围内的门店（可根据需求调整距离和单位）
s = s.query(
    Q('geo_distance', distance='100km', location=target_coords)
)

# 2. 按距离从近到远排序
s = s.sort({
    '_geo_distance': {
        'location': target_coords,
        'order': 'asc',
        'unit': 'km'  # 支持m（米）、mi（英里）等单位
    }
})

# 3. 添加script_fields，让Elasticsearch返回计算好的距离值（这里命名为distance_km）
s = s.script_fields(
    distance_km={
        'script': {
            'source': "doc['location'].arcDistance(params.lat, params.lon) / 1000",
            'params': target_coords
        }
    }
)

# 执行查询
response = s.execute()

# 遍历结果
for hit in response.hits:
    print(f"门店名称：{hit.name}")
    print(f"距离（公里）：{hit.distance_km.value:.2f}")
    print(f"门店地址：{hit.address}")
    print("---")

关键说明 #

• 无需手动计算距离：Elasticsearch在执行地理查询和排序时已经完成了距离计算，我们只是通过script_fields把这个结果提取出来，避免了自己编写Haversine公式之类的麻烦。
• 单次查询完成所有操作：上面的代码一次请求就完成了范围过滤、距离排序、返回距离值三个操作，不需要额外的二次处理。
• 灵活调整单位：如果需要米为单位，把script里的/1000去掉，同时把sort的unit改成m即可。

简化版本（不需要范围过滤） #

如果你想返回所有门店并按距离排序，直接去掉geo_distance查询部分就行：

s = Search(index='store_locations').sort(
    {'_geo_distance': {'location': target_coords, 'order': 'asc', 'unit': 'km'}}
).script_fields(
    distance_km={
        'script': {
            'source': "doc['location'].arcDistance(params.lat, params.lon) / 1000",
            'params': target_coords
        }
    }
)

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