Method documentMatchHandler

stream.go:139–213

(hit *search.DocumentMatch)

}

func (dmh *docMatchHandler) documentMatchHandler(hit *search.DocumentMatch) error {
	if hit != nil {
		/***** remove unnecessary fields *****/
		if dmh.s.req.Highlight == nil {
			if !dmh.s.req.IncludeLocations {
				hit.Locations = nil
			}

			hit.Fragments = nil
		}

		if len(dmh.s.req.Fields) == 0 {
			hit.Fields = nil
		}

		if !dmh.s.req.Explain {
			hit.Expl = nil
		}
		/***** removed unnecessary fields *****/

		if dmh.s.req.IncludeLocations || dmh.s.req.Highlight != nil {
			hit.Complete(nil)
		}

		if len(dmh.s.req.Fields) > 0 || dmh.highlighter != nil {
			bleve.LoadAndHighlightFields(hit, dmh.s.req, "", dmh.ctx.IndexReader,
				dmh.highlighter)
		}

		// If this is a multi collection index, then strip the colelction UID
		// from the hit ID.
		if dmh.collNameMap != nil {
			idBytes := []byte(hit.ID)
			cuid := binary.LittleEndian.Uint32(idBytes[:4])
			if collName, ok := dmh.collNameMap[cuid]; ok {
				hit.ID = string(idBytes[4:])
				if hit.Fields == nil {
					hit.Fields = make(map[string]interface{})
				}
				hit.Fields["_$c"] = collName
			}
		}

		// TODO: perf, perhaps encode directly into output buffer
		// (dmh.bits?)  instead of alloc'ing memory and copying?
		b, err := MarshalJSON(hit)
		if err != nil {
			log.Printf("streamHandler: json marshal err: %v", err)
			return err
		}

		if dmh.n > 0 {
			dmh.bhits = append(append(dmh.bhits, itemGlue...), b...)
		} else {
			dmh.bhits = append(dmh.bhits, b...)
		}
		// remember the ending offset position
		dmh.offsets = append(dmh.offsets, uint64(len(dmh.bhits)))