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

feat: 优化key验证流程

Browse Source
This commit is contained in:
tbphp
2025-07-13 23:21:50 +08:00
parent 411f8f2e78
commit 1a800dd452
6 changed files with 112 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
internal/handler/key_handler.go
View File

@@ -201,11 +201,17 @@ func (s *Server) TestMultipleKeys(c *gin.Context) {
return
}
group, ok := s.findGroupByID(c, req.GroupID)
groupDB, ok := s.findGroupByID(c, req.GroupID)
if !ok {
return
}
group, err := s.GroupManager.GetGroupByName(groupDB.Name)
if err != nil {
response.Error(c, app_errors.NewAPIError(app_errors.ErrResourceNotFound, fmt.Sprintf("Group '%s' not found", groupDB.Name)))
return
}
if err := validateKeysText(req.KeysText); err != nil {
response.Error(c, app_errors.NewAPIError(app_errors.ErrValidation, err.Error()))
return
@@ -234,11 +240,17 @@ func (s *Server) ValidateGroupKeys(c *gin.Context) {
return
}
group, ok := s.findGroupByID(c, req.GroupID)
groupDB, ok := s.findGroupByID(c, req.GroupID)
if !ok {
return
}
group, err := s.GroupManager.GetGroupByName(groupDB.Name)
if err != nil {
response.Error(c, app_errors.NewAPIError(app_errors.ErrResourceNotFound, fmt.Sprintf("Group '%s' not found", groupDB.Name)))
return
}
taskStatus, err := s.KeyManualValidationService.StartValidationTask(group)
if err != nil {
response.Error(c, app_errors.NewAPIError(app_errors.ErrTaskInProgress, err.Error()))