lckt-server/service/bot/bot.go

package bot

import (
	"context"
	"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{}

// CreateBot 创建机器人记录
// Author [yourname](https://github.com/yourname)
func (btService *BotService) CreateBot(ctx context.Context, bt *bot.Bot) (err error) {
	err = global.GVA_DB.Create(bt).Error
	return err
}

// DeleteBot 删除机器人记录
// Author [yourname](https://github.com/yourname)
func (btService *BotService) DeleteBot(ctx context.Context, ID string) (err error) {
	err = global.GVA_DB.Delete(&bot.Bot{}, "id = ?", ID).Error
	return err
}

// DeleteBotByIds 批量删除机器人记录
// Author [yourname](https://github.com/yourname)
func (btService *BotService) DeleteBotByIds(ctx context.Context, IDs []string) (err error) {
	err = global.GVA_DB.Delete(&[]bot.Bot{}, "id in ?", IDs).Error
	return err
}

// UpdateBot 更新机器人记录
// Author [yourname](https://github.com/yourname)
func (btService *BotService) UpdateBot(ctx context.Context, bt bot.Bot) (err error) {
	err = global.GVA_DB.Model(&bot.Bot{}).Where("id = ?", bt.ID).Updates(&bt).Error
	return err
}

// GetBot 根据ID获取机器人记录
// Author [yourname](https://github.com/yourname)
func (btService *BotService) GetBot(ctx context.Context, ID string) (bt bot.Bot, err error) {
	err = global.GVA_DB.Where("id = ?", ID).First(&bt).Error
	return
}

// GetBotInfoList 分页获取机器人记录
// Author [yourname](https://github.com/yourname)
func (btService *BotService) GetBotInfoList(ctx context.Context, info botReq.BotSearch) (list []bot.Bot, total int64, err error) {
	limit := info.PageSize
	offset := info.PageSize * (info.Page - 1)
	// 创建db
	db := global.GVA_DB.Model(&bot.Bot{})
	var bts []bot.Bot
	// 如果有条件搜索 下方会自动创建搜索语句
	if info.StartCreatedAt != nil && info.EndCreatedAt != nil {
		db = db.Where("created_at BETWEEN ? AND ?", info.StartCreatedAt, info.EndCreatedAt)
	}
	err = db.Count(&total).Error
	if err != nil {
		return
	}

	if limit != 0 {
		db = db.Limit(limit).Offset(offset)
	}

	err = db.Find(&bts).Error
	return bts, total, err
}

// GetBotPublic 模糊搜索公开机器人
func (btService *BotService) GetBotPublic(keyWord botReq.FindKey) (bt bot.Bot, err 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
}