🎨 新增设置用户vip接口,优化机器人查询接口&修改跨域配置
This commit is contained in:
@@ -5,6 +5,8 @@ import (
|
||||
"git.echol.cn/loser/lckt/global"
|
||||
"git.echol.cn/loser/lckt/model/bot"
|
||||
botReq "git.echol.cn/loser/lckt/model/bot/request"
|
||||
"gorm.io/gorm"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type BotService struct{}
|
||||
@@ -68,7 +70,98 @@ func (btService *BotService) GetBotInfoList(ctx context.Context, info botReq.Bot
|
||||
err = db.Find(&bts).Error
|
||||
return bts, total, err
|
||||
}
|
||||
|
||||
// GetBotPublic 模糊搜索公开机器人
|
||||
func (btService *BotService) GetBotPublic(keyWord botReq.FindKey) (bt bot.Bot, err error) {
|
||||
err = global.GVA_DB.Where("keyword Like ?", "%"+keyWord.KeyWord+"%").First(&bt).Error
|
||||
userInput := strings.TrimSpace(keyWord.KeyWord)
|
||||
|
||||
// 生成用户输入的所有变体
|
||||
variants := btService.generateKeywordVariants(userInput)
|
||||
|
||||
var conditions []string
|
||||
var args []interface{}
|
||||
|
||||
// 对每个变体进行多种匹配策略
|
||||
for _, variant := range variants {
|
||||
// 1. 精确匹配(逗号分隔的关键词)
|
||||
conditions = append(conditions, "FIND_IN_SET(?, keyword)")
|
||||
args = append(args, variant)
|
||||
|
||||
// 2. 包含匹配
|
||||
conditions = append(conditions, "keyword LIKE ?")
|
||||
args = append(args, "%"+variant+"%")
|
||||
|
||||
// 3. 反向包含匹配(关键词包含用户输入)
|
||||
conditions = append(conditions, "? LIKE CONCAT('%', keyword, '%')")
|
||||
args = append(args, variant)
|
||||
}
|
||||
|
||||
// 4. 特殊处理:如果用户输入较短,尝试作为关键词的子串匹配
|
||||
if len([]rune(userInput)) <= 3 {
|
||||
conditions = append(conditions, "keyword LIKE ?")
|
||||
args = append(args, userInput+"%")
|
||||
|
||||
conditions = append(conditions, "keyword LIKE ?")
|
||||
args = append(args, "%"+userInput)
|
||||
}
|
||||
|
||||
whereClause := strings.Join(conditions, " OR ")
|
||||
err = global.GVA_DB.Where(whereClause, args...).First(&bt).Error
|
||||
|
||||
go func() {
|
||||
// 更新查询次数
|
||||
if err == nil && bt.ID != 0 {
|
||||
global.GVA_DB.Model(&bot.Bot{}).Where("id = ?", bt.ID).UpdateColumn("search_count", gorm.Expr("search_count + ?", 1))
|
||||
}
|
||||
}()
|
||||
return
|
||||
}
|
||||
|
||||
// generateKeywordVariants 生成关键词的所有可能变体
|
||||
func (btService *BotService) generateKeywordVariants(input string) []string {
|
||||
variants := make(map[string]bool)
|
||||
variants[input] = true
|
||||
|
||||
// 中文数字到阿拉伯数字的映射
|
||||
chineseToNum := map[string]string{
|
||||
"一": "1", "二": "2", "三": "3", "四": "4", "五": "5",
|
||||
"六": "6", "七": "7", "八": "8", "九": "9", "十": "10",
|
||||
"零": "0",
|
||||
}
|
||||
|
||||
// 阿拉伯数字到中文数字的映射
|
||||
numToChinese := map[string]string{
|
||||
"1": "一", "2": "二", "3": "三", "4": "四", "5": "五",
|
||||
"6": "六", "7": "七", "8": "八", "9": "九", "10": "十",
|
||||
"0": "零",
|
||||
}
|
||||
|
||||
// 生成数字转换变体
|
||||
current := input
|
||||
|
||||
// 中文数字 -> 阿拉伯数字
|
||||
for cn, num := range chineseToNum {
|
||||
if strings.Contains(current, cn) {
|
||||
numVariant := strings.ReplaceAll(current, cn, num)
|
||||
variants[numVariant] = true
|
||||
}
|
||||
}
|
||||
|
||||
// 阿拉伯数字 -> 中文数字
|
||||
for num, cn := range numToChinese {
|
||||
if strings.Contains(current, num) {
|
||||
cnVariant := strings.ReplaceAll(current, num, cn)
|
||||
variants[cnVariant] = true
|
||||
}
|
||||
}
|
||||
|
||||
// 转换为切片返回
|
||||
result := make([]string, 0, len(variants))
|
||||
for v := range variants {
|
||||
if v != "" && len(strings.TrimSpace(v)) > 0 {
|
||||
result = append(result, v)
|
||||
}
|
||||
}
|
||||
|
||||
return result
|
||||
}
|
||||
|
@@ -228,3 +228,23 @@ func (u *UserService) UpdateTeacherApplyStatus(p app.TeacherApply) (err error) {
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (u *UserService) SetUserVip(p request.SetUserVipReq) error {
|
||||
var user user.User
|
||||
err := global.GVA_DB.Model(&user).Where("id = ?", p.Id).First(&user).Error
|
||||
if err != nil {
|
||||
global.GVA_LOG.Error("查询用户信息失败", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
|
||||
user.IsVip = 1
|
||||
user.UserLabel = p.UserLabel
|
||||
user.VipExpireTime = p.VipExpireTime
|
||||
|
||||
err = global.GVA_DB.Save(&user).Error
|
||||
if err != nil {
|
||||
global.GVA_LOG.Error("设置用户VIP失败", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
Reference in New Issue
Block a user