82 lines
2.3 KiB
Go
82 lines
2.3 KiB
Go
// Package model define model struct of model runtime service
|
||
package model
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"strings"
|
||
|
||
"modelRT/constants"
|
||
"modelRT/diagram"
|
||
|
||
"github.com/RediSearch/redisearch-go/redisearch"
|
||
"github.com/redis/go-redis/v9"
|
||
)
|
||
|
||
// RedisSearch define func of redis search by input string
|
||
func RedisSearch(ctx context.Context, input string) ([]string, error) {
|
||
var rdb redis.Client
|
||
if input == "" {
|
||
// 返回所有 grid 名
|
||
return getAllGridKeys(ctx, constants.RedisAllGridSetKey)
|
||
}
|
||
|
||
// 处理 input 不为空但是不含有.的情况
|
||
if strings.Contains(input, ".") == false {
|
||
setKey := fmt.Sprintf(constants.RedisSpecGridZoneSetKey, input)
|
||
return getSpecificZoneKeys(ctx, setKey)
|
||
}
|
||
|
||
// TODO 默认每次传递的带有.的前缀字符串一定正确
|
||
if strings.HasSuffix(input, ".") {
|
||
// TODO 首先判断.后是否为空字符串，为空则返回上次层级下的所有key
|
||
// TODO 其次进行前缀匹配
|
||
parentKey := strings.TrimSuffix(input, ".")
|
||
// 直接查询预存的下一级内容
|
||
results, err := rdb.ZRange(ctx, "autocomplete:"+parentKey, 0, -1).Result()
|
||
fmt.Println(err)
|
||
return results, nil
|
||
}
|
||
|
||
// TODO 处理前缀或模糊匹配
|
||
// TODO 使用 RediSearch 的 FT.SUGGET 与Fuzzy标志进行前缀或模糊查询
|
||
ac := redisearch.NewAutocompleter("localhost:6379", "my-autocomplete-dict")
|
||
|
||
results, err := ac.SuggestOpts(input, redisearch.SuggestOptions{
|
||
Num: 5,
|
||
Fuzzy: true,
|
||
WithScores: true,
|
||
WithPayloads: true,
|
||
})
|
||
if err != nil {
|
||
fmt.Println(err)
|
||
}
|
||
|
||
var terms []string
|
||
for _, result := range results {
|
||
terms = append(terms, result.Term)
|
||
}
|
||
// 返回结果
|
||
return terms, nil
|
||
}
|
||
|
||
func getAllGridKeys(ctx context.Context, setKey string) ([]string, error) {
|
||
// TODO 从redis set中获取所有的 grid key
|
||
gridSets := diagram.NewRedisSet(ctx, setKey, 10, true)
|
||
keys, err := gridSets.SMembers("grid_keys")
|
||
if err != nil {
|
||
return []string{}, fmt.Errorf("get all root keys failed, error: %v", err)
|
||
}
|
||
return keys, nil
|
||
}
|
||
|
||
func getSpecificZoneKeys(ctx context.Context, setKey string) ([]string, error) {
|
||
// TODO 从redis set中获取所有的 grid key
|
||
zoneSets := diagram.NewRedisSet(ctx, setKey, 10, true)
|
||
keys, err := zoneSets.SMembers("grid_keys")
|
||
if err != nil {
|
||
return []string{}, fmt.Errorf("get all root keys failed, error: %v", err)
|
||
}
|
||
return keys, nil
|
||
}
|