search_response.go

package types

import (

    "github.com/huichen/wukong/utils"

)

type SearchResponse struct {

    // 搜索用到的关键词

    Tokens []string

    // 搜索到的文档，已排序

    Docs []ScoredDocument

    // 搜索是否超时。超时的情况下也可能会返回部分结果

    Timeout bool

    // 搜索到的文档个数。注意这是全部文档中满足条件的个数，可能比返回的文档数要大

    NumDocs int

}

type ScoredDocument struct {

    DocId uint64

    // 文档的打分值

    // 搜索结果按照Scores的值排序，先按照第一个数排，如果相同则按照第二个数排序，依次类推。

    Scores []float32

    // 用于生成摘要的关键词在文本中的字节位置，该切片长度和SearchResponse.Tokens的长度一样

    // 只有当IndexType == LocationsIndex时不为空

    TokenSnippetLocations []int

    // 关键词出现的位置

    // 只有当IndexType == LocationsIndex时不为空

    TokenLocations [][]int

}

// 为了方便排序

type ScoredDocuments []ScoredDocument

func (docs ScoredDocuments) Len() int {

    return len(docs)

}

func (docs ScoredDocuments) Swap(i, j int) {

    docs[i], docs[j] = docs[j], docs[i]

}

func (docs ScoredDocuments) Less(i, j int) bool {

    // 为了从大到小排序，这实际上实现的是More的功能

    for iScore := 0; iScore < utils.MinInt(len(docs[i].Scores), len(docs[j].Scores)); iScore++ {

        if docs[i].Scores[iScore] > docs[j].Scores[iScore] {

            return true

        } else if docs[i].Scores[iScore] < docs[j].Scores[iScore] {

            return false

        }

    }

    return len(docs[i].Scores) > len(docs[j].Scores)

}