fork-it/gpt-load
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 恢复key时重置失败次数

Browse Source
This commit is contained in:
tbphp
2025-07-07 22:34:35 +08:00
parent 76ab3fe784
commit 9896c975d3
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
internal/keypool/provider.go
View File

@@ -339,7 +339,11 @@ func (p *KeyProvider) RestoreKeys(groupID uint) (int64, error) {
return nil return nil
} }
result := tx.Model(&models.APIKey{}).Where("group_id = ? AND status = ?", groupID, models.KeyStatusInvalid).Update("status", models.KeyStatusActive) updates := map[string]any{
"status": models.KeyStatusActive,
"failure_count": 0,
}
result := tx.Model(&models.APIKey{}).Where("group_id = ? AND status = ?", groupID, models.KeyStatusInvalid).Updates(updates)
if result.Error != nil { if result.Error != nil {
return result.Error return result.Error
} }
@@ -347,6 +351,7 @@ func (p *KeyProvider) RestoreKeys(groupID uint) (int64, error) {
for _, key := range invalidKeys { for _, key := range invalidKeys {
key.Status = models.KeyStatusActive key.Status = models.KeyStatusActive
key.FailureCount = 0
if err := p.addKeyToStore(&key); err != nil { if err := p.addKeyToStore(&key); err != nil {
logrus.WithFields(logrus.Fields{"keyID": key.ID, "error": err}).Error("Failed to restore key in store after DB update") logrus.WithFields(logrus.Fields{"keyID": key.ID, "error": err}).Error("Failed to restore key in store after DB update")
} }