Go rtreego 怎样查找距离给定点最近的对象

如果要在一批二维对象中找出离给定坐标最近的一个，rtreego 直接提供了最近邻查询 API。核心做法是：让对象实现 Spatial.Bounds()，用返回的矩形描述空间范围，再把对象插入二维 R-tree；查询单个结果用 NearestNeighbor，需要前几个候选时用 NearestNeighbors。

官方文档：https://pkg.go.dev/github.com/dhconnelly/rtreego

下面用“地图上的服务点”这个小场景串起来。文中的图片都是静态结构说明图，用来解释对象关系，不是运行结果截图。

先让对象具备空间边界

rtreego 并不要求业务对象继承某个基类，只要求它满足 Spatial 接口。对二维点对象来说，最简单的边界是以坐标为中心、带有很小容差的矩形；如果对象本身有宽高，就可以直接返回它的包围盒。

关键是保证所有坐标维度一致。树创建时的 dim 是 2，Point 和 Rect 也必须使用二维数据。把一维点或三维边界放进二维树，会在几何计算阶段产生维度错误。

用一个小项目建立最近邻索引

先定义服务点。这里用 Point.ToRect 把点转换成一个极小矩形，这样既能满足 Bounds，也能保留点查询的语义。

package main

import (
    "fmt"

    "github.com/dhconnelly/rtreego"
)

type Place struct {
    Name string
    Pos  rtreego.Point
}

// Bounds 返回对象的空间包围盒；容差让点对象拥有非零范围。
func (p *Place) Bounds() rtreego.Rect {
    return p.Pos.ToRect(0.01)
}

func main() {
    // dim=2 表示二维坐标；min/max 是节点的最小和最大分支因子。
    tree := rtreego.NewTree(2, 25, 50)
    places := []*Place{
        {Name: "东门服务点", Pos: rtreego.Point{2.0, 1.0}},
        {Name: "中央服务点", Pos: rtreego.Point{5.0, 4.0}},
        {Name: "西侧服务点", Pos: rtreego.Point{8.0, 2.0}},
    }
    for _, place := range places {
        // 插入的是实现 Spatial 的对象，而不是单独的 Point。
        tree.Insert(place)
    }

    query := rtreego.Point{4.6, 3.8}
    nearest := tree.NearestNeighbor(query)
    if nearest == nil {
        // 空树没有最近对象，业务层应先处理这个分支。
        fmt.Println("没有可用服务点")
        return
    }
    fmt.Println(nearest.(*Place).Name)
}

NewTree(2, 25, 50) 的第一个参数决定空间维度，后两个参数控制节点分支范围。它们不是“查询最近距离”的阈值，也不会改变距离公式。插入对象后，树会维护自己的空间索引。

单个最近对象和前 k 个对象怎么选

只需要一个候选时，NearestNeighbor(query) 返回一个 Spatial。它可能是 nil，所以不要直接断言为业务类型。需要展示备选点、做二次营业状态筛选，或者希望保留多个候选时，改用 NearestNeighbors(k, query, filters...)。

query := rtreego.Point{4.6, 3.8}

// 只取一个最近对象；结果仍以 Spatial 接口返回。
one := tree.NearestNeighbor(query)

// 取前两个候选，k 应根据业务兜底数量设置。
topTwo := tree.NearestNeighbors(2, query)
for _, item := range topTwo {
    place := item.(*Place)
    fmt.Println(place.Name)
}

如果要在索引查询阶段拒绝某些对象，可以给 NearestNeighbors 传入过滤器。过滤器适合表达“对象仍在营业”“对象属于当前租户”这类查询边界；不要把复杂业务排序偷偷混进空间索引，否则读者很难判断返回顺序代表距离还是业务优先级。

移动对象时不要只改坐标

rtreego 的索引依赖对象当前的边界。如果一个服务点移动了，直接修改 Pos 而不更新树，会让树中保存的空间边界与对象实际位置不一致。正确的处理顺序是删除旧对象、更新坐标、再插入对象。

// 先从索引中移除旧边界，避免树保留过期位置。
tree.Delete(place)
place.Pos = rtreego.Point{6.0, 3.0}
// 重新插入，让索引按新坐标建立边界。
tree.Insert(place)

如果手里没有原来的指针，默认删除比较可能无法定位对象，可以按业务 ID 使用 DeleteWithComparator 提供自定义比较器。这个边界很适合封装在“移动服务点”方法里，避免调用方漏掉重插入。

最后做一次结果验收

可以用下面这张清单确认最近邻接入是否完整：

常见问题是“为什么最近对象不是我以为的那个”：先确认查询点与对象边界使用的是同一坐标系，再确认没有把经纬度顺序、单位或维度混用。rtreego 负责空间索引，不会替你把经纬度转换成球面距离；如果业务需要地理距离，应先定义合适的坐标或距离模型。

相关问题

• 只查一个对象和查前 k 个对象有什么区别？ 前者用 NearestNeighbor，后者用 NearestNeighbors，多候选适合做业务过滤和兜底。
• 对象没有矩形范围怎么办？ 可以用点的 ToRect 生成带容差的矩形，让点对象实现 Bounds。
• rtreego 能直接处理经纬度距离吗？ 它提供的是 N 维空间索引；经纬度的投影、单位和球面距离语义需要由业务层先处理。