任务权限代码优化

modules/cloudbrain/cloudbrain.go

@@ -46,16 +46,7 @@ func isAdminOrOwnerOrJobCreater(ctx *context.Context, job *models.Cloudbrain, er
 
 }
 
-func CanDeleteDebugJob(ctx *context.Context, job *models.Cloudbrain) bool {
-
-	if job.Status != string(models.JobStopped) && job.Status != string(models.JobFailed) && job.Status != string(models.ModelArtsStartFailed) && job.Status != string(models.ModelArtsCreateFailed) {
-		return false
-	}
-
-	return isAdminOrOwnerOrJobCreater(ctx, job, nil)
-}
-
-func CanDeleteTrainJob(ctx *context.Context, job *models.Cloudbrain) bool {
+func CanDeleteJob(ctx *context.Context, job *models.Cloudbrain) bool {
 
 	return isAdminOrOwnerOrJobCreater(ctx, job, nil)
 }

routers/repo/cloudbrain.go

@@ -79,7 +79,7 @@ func CloudBrainIndex(ctx *context.Context) {
 			ciTasks[i].CanDebug = false
 		}
 
-		ciTasks[i].CanDel = cloudbrain.CanDeleteDebugJob(ctx, &task.Cloudbrain)
+		ciTasks[i].CanDel = cloudbrain.CanDeleteJob(ctx, &task.Cloudbrain)
 
 	}
 

routers/repo/modelarts.go

@@ -74,7 +74,7 @@ func NotebookIndex(ctx *context.Context) {
 		} else {
 			ciTasks[i].CanDebug = false
 		}
-		ciTasks[i].CanDel = cloudbrain.CanDeleteDebugJob(ctx, &task.Cloudbrain)
+		ciTasks[i].CanDel = cloudbrain.CanDeleteJob(ctx, &task.Cloudbrain)
 	}
 
 	pager := context.NewPagination(int(count), setting.UI.IssuePagingNum, page, 5)
@@ -306,7 +306,7 @@ func TrainJobIndex(ctx *context.Context) {
 	}
 
 	for i, task := range tasks {
-		tasks[i].CanDel = cloudbrain.CanDeleteTrainJob(ctx, &task.Cloudbrain)
+		tasks[i].CanDel = cloudbrain.CanDeleteJob(ctx, &task.Cloudbrain)
 		tasks[i].CanModify = cloudbrain.CanModifyJob(ctx, &task.Cloudbrain)
 	}