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

feat: 分组密钥检查增加状态选项 (#86)

Browse Source
This commit is contained in:
tbphp
2025-08-01 10:01:49 +08:00
committed by GitHub
parent 1c658fad56
commit d4dd5eab67
5 changed files with 60 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
internal/services/key_manual_validation_service.go
View File

@@ -41,10 +41,14 @@ func NewKeyManualValidationService(db *gorm.DB, validator *keypool.KeyValidator,
}
// StartValidationTask starts a new manual validation task for a given group.
func (s *KeyManualValidationService) StartValidationTask(group *models.Group) (*TaskStatus, error) {
func (s *KeyManualValidationService) StartValidationTask(group *models.Group, status string) (*TaskStatus, error) {
var keys []models.APIKey
if err := s.DB.Where("group_id = ?", group.ID).Find(&keys).Error; err != nil {
return nil, fmt.Errorf("failed to get keys for group %s: %w", group.Name, err)
query := s.DB.Where("group_id = ?", group.ID)
if status != "" {
query = query.Where("status = ?", status)
}
if err := query.Find(&keys).Error; err != nil {
return nil, fmt.Errorf("failed to get keys for group %s with status '%s': %w", group.Name, status, err)
}
if len(keys) == 0 {
@@ -59,13 +63,20 @@ func (s *KeyManualValidationService) StartValidationTask(group *models.Group) (*
}
// Run the validation in a separate goroutine
go s.runValidation(group, keys)
go s.runValidation(group, keys, status)
return taskStatus, nil
}
func (s *KeyManualValidationService) runValidation(group *models.Group, keys []models.APIKey) {
logrus.Infof("Starting manual validation for group %s", group.Name)
func (s *KeyManualValidationService) runValidation(group *models.Group, keys []models.APIKey, status string) {
logFields := logrus.Fields{
"group": group.Name,
"status": status,
}
if status == "" {
logFields["status"] = "all"
}
logrus.WithFields(logFields).Info("Starting manual validation")
jobs := make(chan models.APIKey, len(keys))
results := make(chan bool, len(keys))