62 lines
1.4 KiB
Go
62 lines
1.4 KiB
Go
|
package lib
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"strings"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 定义一个结构体表示每条数据
|
|||
|
|
type MatchData struct {
|
|||
|
|
ID uint
|
|||
|
|
Text string
|
|||
|
|
Score int
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 定义一个结构体表示匹配器
|
|||
|
|
type Matcher struct {
|
|||
|
|
data []MatchData
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 定义一个构造函数，用于创建 Matcher 实例
|
|||
|
|
func NewMatcher(data []MatchData) *Matcher {
|
|||
|
|
return &Matcher{data: data}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 定义一个 Match 方法，用于计算最匹配的数据切片
|
|||
|
|
func (m *Matcher) Match(input string, exactMatch bool) (MatchData, bool) {
|
|||
|
|
// 遍历每条数据
|
|||
|
|
var maxScore int
|
|||
|
|
var result MatchData
|
|||
|
|
for _, d := range m.data {
|
|||
|
|
// 将数据切片中的中文逗号替换为英文逗号
|
|||
|
|
text := strings.ReplaceAll(d.Text, "，", ",")
|
|||
|
|
// 切分单词
|
|||
|
|
words := strings.Split(text, ",")
|
|||
|
|
// 初始化分值
|
|||
|
|
score := 0
|
|||
|
|
// 遍历每个单词
|
|||
|
|
for _, w := range words {
|
|||
|
|
// 如果是精准命中，判断输入字符串是否等于单词
|
|||
|
|
if exactMatch && input == w {
|
|||
|
|
score++
|
|||
|
|
}
|
|||
|
|
// 如果是包含，判断输入字符串是否包含单词
|
|||
|
|
if !exactMatch && strings.Contains(input, w) {
|
|||
|
|
score++
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
// 如果当前分值比最大分值大，则更新最大分值和结果
|
|||
|
|
if score > maxScore {
|
|||
|
|
maxScore = score
|
|||
|
|
result = d
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 如果最终分值为0，则返回 false
|
|||
|
|
if maxScore == 0 {
|
|||
|
|
return MatchData{}, false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 返回最匹配的结果
|
|||
|
|
return result, true
|
|||
|
|
}
|