github/new-api
1
0
Fork 0
mirror of https://github.com/QuantumNous/new-api.git synced 2026-03-30 04:40:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(relay): update channel retrieval to use RelayInfo structure

Browse Source
This commit is contained in:
CaIon
2025-12-12 22:04:38 +08:00
parent 2ac6a5b02f
commit ce6fb95f96
4 changed files with 29 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
service/channel_select.go
View File

@@ -12,12 +12,12 @@ import (
)
// CacheGetRandomSatisfiedChannel tries to get a random channel that satisfies the requirements.
func CacheGetRandomSatisfiedChannel(c *gin.Context, group string, modelName string, retry int) (*model.Channel, string, error) {
func CacheGetRandomSatisfiedChannel(c *gin.Context, tokenGroup string, modelName string, retry int) (*model.Channel, string, error) {
var channel *model.Channel
var err error
selectGroup := group
selectGroup := tokenGroup
userGroup := common.GetContextKeyString(c, constant.ContextKeyUserGroup)
if group == "auto" {
if tokenGroup == "auto" {
if len(setting.GetAutoGroups()) == 0 {
return nil, selectGroup, errors.New("auto groups is not enabled")
}
@@ -49,9 +49,9 @@ func CacheGetRandomSatisfiedChannel(c *gin.Context, group string, modelName stri
}
}
} else {
channel, err = model.GetRandomSatisfiedChannel(group, modelName, retry)
channel, err = model.GetRandomSatisfiedChannel(tokenGroup, modelName, retry)
if err != nil {
return nil, group, err
return nil, tokenGroup, err
}
}
return channel, selectGroup, nil