Include "distance" to output when user specifically requests

fixes #599
This commit is contained in:
tidwall 2021-03-30 11:49:01 -07:00
parent dd26839351
commit 901919c4e3
3 changed files with 10 additions and 7 deletions

View File

@ -172,11 +172,12 @@ func fenceMatch(
sw.fullFields = true sw.fullFields = true
sw.msg.OutputType = JSON sw.msg.OutputType = JSON
sw.writeObject(ScanWriterParams{ sw.writeObject(ScanWriterParams{
id: details.id, id: details.id,
o: details.obj, o: details.obj,
fields: details.fields, fields: details.fields,
noLock: true, noLock: true,
distance: distance, distance: distance,
distOutput: fence.distance,
}) })
if sw.wr.Len() == 0 { if sw.wr.Len() == 0 {

View File

@ -66,6 +66,7 @@ type ScanWriterParams struct {
o geojson.Object o geojson.Object
fields []float64 fields []float64
distance float64 distance float64
distOutput bool // query or fence requested distance output
noLock bool noLock bool
ignoreGlobMatch bool ignoreGlobMatch bool
clip geojson.Object clip geojson.Object
@ -433,7 +434,7 @@ func (sw *scanWriter) writeObject(opts ScanWriterParams) bool {
wr.WriteString(jsfields) wr.WriteString(jsfields)
if opts.distance > 0 { if opts.distOutput || opts.distance > 0 {
wr.WriteString(`,"distance":` + strconv.FormatFloat(opts.distance, 'f', -1, 64)) wr.WriteString(`,"distance":` + strconv.FormatFloat(opts.distance, 'f', -1, 64))
} }
@ -496,7 +497,7 @@ func (sw *scanWriter) writeObject(opts ScanWriterParams) bool {
vals = append(vals, resp.ArrayValue(fvals)) vals = append(vals, resp.ArrayValue(fvals))
} }
} }
if opts.distance > 0 { if opts.distOutput || opts.distance > 0 {
vals = append(vals, resp.FloatValue(opts.distance)) vals = append(vals, resp.FloatValue(opts.distance))
} }

View File

@ -380,6 +380,7 @@ func (server *Server) cmdNearby(msg *Message) (res resp.Value, err error) {
o: o, o: o,
fields: fields, fields: fields,
distance: meters, distance: meters,
distOutput: s.distance,
noLock: true, noLock: true,
ignoreGlobMatch: true, ignoreGlobMatch: true,
skipTesting: true, skipTesting: true,