Function CacheGetRandomSatisfiedChannel

model/channel_cache.go:83–125

(c *gin.Context, group string, model string, retry int)

}

func CacheGetRandomSatisfiedChannel(c *gin.Context, group string, model string, retry int) (*Channel, string, error) {
	var channel *Channel
	var err error
	selectGroup := group
	// 获取令牌渠道标签
	tokenChannelTag := common.GetContextKeyString(c, constant.ContextKeyTokenChannelTag)
	var channelTag *string = nil
	if tokenChannelTag != "" {
		channelTag = &tokenChannelTag
	}

	if group == "auto" {
		if len(setting.AutoGroups) == 0 {
			return nil, selectGroup, errors.New("auto groups is not enabled")
		}
		for _, autoGroup := range setting.AutoGroups {
			if common.DebugEnabled {
				println("autoGroup:", autoGroup)
			}
			channel, _ = getRandomSatisfiedChannel(autoGroup, model, retry)
			if channel == nil {
				continue
			} else {
				c.Set("auto_group", autoGroup)
				selectGroup = autoGroup
				if common.DebugEnabled {
					println("selectGroup:", selectGroup)
				}
				break
			}
		}
	} else {
		// 传递channelTag参数给getRandomSatisfiedChannel
		channel, err = getRandomSatisfiedChannelWithTag(group, model, retry, channelTag)
		if err != nil {
			return nil, group, err
		}
	}
	if channel == nil {
		return nil, group, errors.New("channel not found")
	}
	return channel, selectGroup, nil
}

// 新增带标签过滤的渠道选择函数
func getRandomSatisfiedChannelWithTag(group string, model string, retry int, channelTag *string) (*Channel, error) {