style: Optimize task timeout code (#267)

Author: tbphp

internal/services/key_delete_service.go

@@ -3,14 +3,12 @@ package services
 import (
 	"fmt"
 	"gpt-load/internal/models"
-	"time"
 
 	"github.com/sirupsen/logrus"
 )
 
 const (
 	deleteChunkSize = 1000
-	deleteTimeout   = 30 * time.Minute
 )
 
 // KeyDeleteResult holds the result of a delete task.
@@ -40,7 +38,7 @@ func (s *KeyDeleteService) StartDeleteTask(group *models.Group, keysText string)
 		return nil, fmt.Errorf("no valid keys found in the input text")
 	}
 
-	initialStatus, err := s.TaskService.StartTask(TaskTypeKeyDelete, group.Name, len(keys), deleteTimeout)
+	initialStatus, err := s.TaskService.StartTask(TaskTypeKeyDelete, group.Name, len(keys))
 	if err != nil {
 		return nil, err
 	}

internal/services/key_import_service.go

@@ -3,15 +3,10 @@ package services
 import (
 	"fmt"
 	"gpt-load/internal/models"
-	"time"
 
 	"github.com/sirupsen/logrus"
 )
 
-const (
-	importTimeout = 30 * time.Minute
-)
-
 // KeyImportResult holds the result of an import task.
 type KeyImportResult struct {
 	AddedCount   int `json:"added_count"`
@@ -39,7 +34,7 @@ func (s *KeyImportService) StartImportTask(group *models.Group, keysText string)
 		return nil, fmt.Errorf("no valid keys found in the input text")
 	}
 
-	initialStatus, err := s.TaskService.StartTask(TaskTypeKeyImport, group.Name, len(keys), importTimeout)
+	initialStatus, err := s.TaskService.StartTask(TaskTypeKeyImport, group.Name, len(keys))
 	if err != nil {
 		return nil, err
 	}

internal/services/key_manual_validation_service.go

@@ -58,9 +58,7 @@ func (s *KeyManualValidationService) StartValidationTask(group *models.Group, st
 		return nil, fmt.Errorf("no keys to validate in group %s", group.Name)
 	}
 
-	timeout := 30 * time.Minute
-
-	taskStatus, err := s.TaskService.StartTask(TaskTypeKeyValidation, group.Name, len(keys), timeout)
+	taskStatus, err := s.TaskService.StartTask(TaskTypeKeyValidation, group.Name, len(keys))
 	if err != nil {
 		return nil, err
 	}

internal/services/task_service.go

@@ -46,7 +46,7 @@ func NewTaskService(store store.Store) *TaskService {
 }
 
 // StartTask attempts to start a new task. It returns an error if a task is already running.
-func (s *TaskService) StartTask(taskType, groupName string, total int, timeout time.Duration) (*TaskStatus, error) {
+func (s *TaskService) StartTask(taskType, groupName string, total int) (*TaskStatus, error) {
 	currentStatus, err := s.GetTaskStatus()
 	if err != nil {
 		return nil, fmt.Errorf("failed to check current task status before starting a new one: %w", err)
@@ -69,7 +69,7 @@ func (s *TaskService) StartTask(taskType, groupName string, total int, timeout t
 		return nil, fmt.Errorf("failed to serialize new task status: %w", err)
 	}
 
-	if err := s.store.Set(globalTaskKey, statusBytes, timeout); err != nil {
+	if err := s.store.Set(globalTaskKey, statusBytes, ResultTTL); err != nil {
 		return nil, fmt.Errorf("failed to set initial task status: %w", err)
 	}