chuan/CLIProxyAPI
1
0
Fork 0
mirror of https://github.com/router-for-me/CLIProxyAPI.git synced 2026-02-03 13:00:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(auth): Make round-robin auth selection deterministic

Browse Source
This commit is contained in:
hkfires
2025-09-26 09:49:53 +08:00
parent f228a4dcca
commit e836b4ac10
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
sdk/cliproxy/auth/manager.go
View File

@@ -15,6 +15,7 @@ import (
"github.com/router-for-me/CLIProxyAPI/v6/internal/util" "github.com/router-for-me/CLIProxyAPI/v6/internal/util"
cliproxyexecutor "github.com/router-for-me/CLIProxyAPI/v6/sdk/cliproxy/executor" cliproxyexecutor "github.com/router-for-me/CLIProxyAPI/v6/sdk/cliproxy/executor"
log "github.com/sirupsen/logrus" log "github.com/sirupsen/logrus"
"sort"
) )
// ProviderExecutor defines the contract required by Manager to execute provider calls. // ProviderExecutor defines the contract required by Manager to execute provider calls.
@@ -796,6 +797,12 @@ func (m *Manager) pickNext(ctx context.Context, provider, model string, opts cli
} }
candidates = append(candidates, auth.Clone()) candidates = append(candidates, auth.Clone())
} }
// Ensure stable iteration order across requests to make round-robin deterministic.
// Go map iteration order is randomized; without sorting, the selector's index-based
// rotation may appear non-round-robin.
if len(candidates) > 1 {
sort.Slice(candidates, func(i, j int) bool { return candidates[i].ID < candidates[j].ID })
}
m.mu.RUnlock() m.mu.RUnlock()
if len(candidates) == 0 { if len(candidates) == 0 {
return nil, nil, &Error{Code: "auth_not_found", Message: "no auth available"} return nil, nil, &Error{Code: "auth_not_found", Message: "no auth available"}