mirror
/
tile38
mirror of https://github.com/tidwall/tile38.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Attempt to fix NEARBY to filter prior to limiting.

This commit is contained in:
Alex Roitman 2017-10-30 16:01:37 -07:00
parent 8b06aedf73
commit 4eca0cdee7
1 changed files with 5 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
controller/search.go
View File

@ -352,20 +352,14 @@ type iterItem struct {
} }
func nearestNeighbors(sw *scanWriter, lat, lon float64, iter func(id string, o geojson.Object, fields []float64, dist *float64) bool) { func nearestNeighbors(sw *scanWriter, lat, lon float64, iter func(id string, o geojson.Object, fields []float64, dist *float64) bool) {
limit := sw.limit * 10 limit := int(sw.limit)
if limit > limitItems*10 {
limit = limitItems * 10
}
k := sw.cursor + limit
var items []iterItem var items []iterItem
sw.col.NearestNeighbors(lat, lon, func(id string, o geojson.Object, fields []float64) bool { sw.col.NearestNeighbors(lat, lon, func(id string, o geojson.Object, fields []float64) bool {
if k == 0 { if _, ok := sw.fieldMatch(fields, o); ok {
return false dist := o.CalculatedPoint().DistanceTo(geojson.Position{X: lon, Y: lat, Z: 0})
items = append(items, iterItem{id: id, o: o, fields: fields, dist: dist})
} }
dist := o.CalculatedPoint().DistanceTo(geojson.Position{X: lon, Y: lat, Z: 0}) return len(items) < limit
items = append(items, iterItem{id: id, o: o, fields: fields, dist: dist})
k--
return true
}) })
sort.Slice(items, func(i, j int) bool { sort.Slice(items, func(i, j int) bool {
return items[i].dist < items[j].dist return items[i].dist < items[j].dist