fix task duplicateName issue

internal/logic/inference/createinferencetasklogic.go

@@ -53,7 +53,7 @@ func (l *CreateInferenceTaskLogic) CreateInferenceTask(req *types.CreateInferenc
 		return nil, err
 	}
 
-	taskName, err := l.svcCtx.Scheduler.AiService.HandleDuplicateTaskName(req.Name)
+	taskName, err := l.svcCtx.Scheduler.AiService.HandleDuplicateTaskName(req.Name, "inference")
 	if err != nil {
 		return nil, err
 	}

internal/logic/schedule/schedulecreatetasklogic.go

@@ -147,7 +147,7 @@ func (l *ScheduleCreateTaskLogic) ScheduleCreateTask(req *types.CreateTaskReq) (
 		return nil, err
 	}
 
-	taskName, err := l.svcCtx.Scheduler.AiService.HandleDuplicateTaskName(req.Name)
+	taskName, err := l.svcCtx.Scheduler.AiService.HandleDuplicateTaskName(req.Name, "training")
 	if err != nil {
 		return nil, err
 	}

internal/scheduler/database/aiStorage.go

@@ -103,13 +103,23 @@ func (s *AiStorage) GetAiTaskListById(id int64) ([]*models.TaskAi, error) {
 	return aiTaskList, nil
 }
 
-func (s *AiStorage) DoesTaskNameExist(name string) (bool, error) {
+func (s *AiStorage) DoesTaskNameExist(name string, taskType string) (bool, error) {
 	var total int32
-	tx := s.DbEngin.Raw("select count(*) from task where `name` = ?", name).Scan(&total)
-	if tx.Error != nil {
-		logx.Errorf(tx.Error.Error())
-		return false, tx.Error
+	switch taskType {
+	case "training":
+		tx := s.DbEngin.Raw("select count(*) from task where `name` = ?", name).Scan(&total)
+		if tx.Error != nil {
+			logx.Errorf(tx.Error.Error())
+			return false, tx.Error
+		}
+	case "inference":
+		tx := s.DbEngin.Raw("select count(*) from ai_deploy_instance_task where `name` = ?", name).Scan(&total)
+		if tx.Error != nil {
+			logx.Errorf(tx.Error.Error())
+			return false, tx.Error
+		}
 	}
+
 	return total > 0, nil
 }
 

internal/scheduler/service/aiService.go

@@ -143,8 +143,8 @@ func (as *AiService) UpdateClusterMaps(conf *config.Config, adapterId string, cl
 	}
 }
 
-func (as *AiService) HandleDuplicateTaskName(name string) (string, error) {
-	exist, err := as.Storage.DoesTaskNameExist(name)
+func (as *AiService) HandleDuplicateTaskName(name string, taskType string) (string, error) {
+	exist, err := as.Storage.DoesTaskNameExist(name, taskType)
 	if err != nil {
 		return "", err
 	}