Function getRandomSatisfiedChannelWithTag

model/channel_cache.go:128–245

新增带标签过滤的渠道选择函数

(group string, model string, retry int, channelTag *string)

// 新增带标签过滤的渠道选择函数
func getRandomSatisfiedChannelWithTag(group string, model string, retry int, channelTag *string) (*Channel, error) {
	if strings.HasPrefix(model, "gpt-4-gizmo") {
		model = "gpt-4-gizmo-*"
	}
	if strings.HasPrefix(model, "gpt-4o-gizmo") {
		model = "gpt-4o-gizmo-*"
	}

	// if memory cache is disabled, get channel directly from database
	if !common.MemoryCacheEnabled {
		return GetRandomSatisfiedChannel(group, model, retry, channelTag)
	}

	channelSyncLock.RLock()
	defer channelSyncLock.RUnlock()
	channels := group2model2channels[group][model]

	if len(channels) == 0 {
		return nil, errors.New("channel not found")
	}

	// 过滤符合标签要求的渠道
	var filteredChannels []int
	for _, channelId := range channels {
		if channel, ok := channelsIDM[channelId]; ok {
			// 如果没有指定标签要求，则所有渠道都符合
			if channelTag == nil || *channelTag == "" {
				filteredChannels = append(filteredChannels, channelId)
			} else {
				// 如果指定了标签要求，则只选择匹配标签的渠道
				channelTagStr := channel.GetTag()
				if channelTagStr == *channelTag {
					filteredChannels = append(filteredChannels, channelId)
				}
			}
		}
	}

	// 如果没有符合标签要求的渠道，返回错误
	if len(filteredChannels) == 0 {
		if channelTag != nil && *channelTag != "" {
			return nil, fmt.Errorf("没有找到标签为 '%s' 的可用渠道", *channelTag)
		}
		return nil, errors.New("channel not found")
	}

	if len(filteredChannels) == 1 {
		if channel, ok := channelsIDM[filteredChannels[0]]; ok {
			return channel, nil
		}
		return nil, fmt.Errorf("数据库一致性错误，渠道# %d 不存在，请联系管理员修复", filteredChannels[0])
	}

	uniquePriorities := make(map[int]bool)
	for _, channelId := range filteredChannels {
		if channel, ok := channelsIDM[channelId]; ok {
			uniquePriorities[int(channel.GetPriority())] = true
		} else {