提交代码

routers/repo/cloudbrain.go

@@ -636,6 +636,7 @@ func CloudBrainTrainJobShow(ctx *context.Context) {
 func cloudBrainShow(ctx *context.Context, tpName base.TplName, jobType models.JobType) {
 	ctx.Data["PageIsCloudBrain"] = true
 	debugListType := ctx.Query("debugListType")
+	cloudbrain.InitSpecialPool()
 
 	var task *models.Cloudbrain
 	var err error
@@ -647,22 +648,22 @@ func cloudBrainShow(ctx *context.Context, tpName base.TplName, jobType models.Jo
 
 	if err != nil {
 		log.Info("error:" + err.Error())
-		ctx.Data["error"] = err.Error()
+		ctx.NotFound(ctx.Req.URL.RequestURI(), nil)
 		return
 	}
 
 	result, err := cloudbrain.GetJob(task.JobID)
 	if err != nil {
 		log.Info("error:" + err.Error())
-		ctx.Data["error"] = err.Error()
+		ctx.NotFound(ctx.Req.URL.RequestURI(), nil)
 		return
 	}
-
+	hasSpec := false
 	if task.JobType == string(models.JobTypeTrain) {
 		if cloudbrain.TrainResourceSpecs == nil {
 			json.Unmarshal([]byte(setting.TrainResourceSpecs), &cloudbrain.TrainResourceSpecs)
 		}
-		hasSpec := false
+
 		for _, tmp := range cloudbrain.TrainResourceSpecs.ResourceSpec {
 			if tmp.Id == task.ResourceSpecId {
 				hasSpec = true
@@ -670,24 +671,7 @@ func cloudBrainShow(ctx *context.Context, tpName base.TplName, jobType models.Jo
 				ctx.Data["CpuNum"] = tmp.CpuNum
 				ctx.Data["MemMiB"] = tmp.MemMiB
 				ctx.Data["ShareMemMiB"] = tmp.ShareMemMiB
-			}
-		}
-
-		if !hasSpec && cloudbrain.SpecialPools != nil {
-			for _, specialPool := range cloudbrain.SpecialPools.Pools {
-
-				if specialPool.ResourceSpec != nil {
-
-					for _, spec := range specialPool.ResourceSpec {
-						if task.ResourceSpecId == spec.Id {
-							ctx.Data["GpuNum"] = spec.GpuNum
-							ctx.Data["CpuNum"] = spec.CpuNum
-							ctx.Data["MemMiB"] = spec.MemMiB
-							ctx.Data["ShareMemMiB"] = spec.ShareMemMiB
-							break
-						}
-					}
-				}
+				break
 			}
 		}
 
@@ -697,10 +681,12 @@ func cloudBrainShow(ctx *context.Context, tpName base.TplName, jobType models.Jo
 		}
 		for _, tmp := range cloudbrain.InferenceResourceSpecs.ResourceSpec {
 			if tmp.Id == task.ResourceSpecId {
+				hasSpec = true
 				ctx.Data["GpuNum"] = tmp.GpuNum
 				ctx.Data["CpuNum"] = tmp.CpuNum
 				ctx.Data["MemMiB"] = tmp.MemMiB
 				ctx.Data["ShareMemMiB"] = tmp.ShareMemMiB
+				break
 			}
 		}
 	} else {
@@ -709,10 +695,32 @@ func cloudBrainShow(ctx *context.Context, tpName base.TplName, jobType models.Jo
 		}
 		for _, tmp := range cloudbrain.ResourceSpecs.ResourceSpec {
 			if tmp.Id == task.ResourceSpecId {
+				hasSpec = true
 				ctx.Data["GpuNum"] = tmp.GpuNum
 				ctx.Data["CpuNum"] = tmp.CpuNum
 				ctx.Data["MemMiB"] = tmp.MemMiB
 				ctx.Data["ShareMemMiB"] = tmp.ShareMemMiB
+				break
+
+			}
+		}
+	}
+
+	if !hasSpec && cloudbrain.SpecialPools != nil {
+
+		for _, specialPool := range cloudbrain.SpecialPools.Pools {
+
+			if specialPool.ResourceSpec != nil {
+
+				for _, spec := range specialPool.ResourceSpec {
+					if task.ResourceSpecId == spec.Id {
+						ctx.Data["GpuNum"] = spec.GpuNum
+						ctx.Data["CpuNum"] = spec.CpuNum
+						ctx.Data["MemMiB"] = spec.MemMiB
+						ctx.Data["ShareMemMiB"] = spec.ShareMemMiB
+						break
+					}
+				}
 			}
 		}
 	}
@@ -731,14 +739,6 @@ func cloudBrainShow(ctx *context.Context, tpName base.TplName, jobType models.Jo
 					ctx.Data["resource_type"] = resourceType.Value
 				}
 			}
-			for _, specialPool := range cloudbrain.SpecialPools.Pools {
-
-				for _, resourceType := range specialPool.Pool {
-					if resourceType.Queue == jobRes.Config.GpuType {
-						ctx.Data["resource_type"] = resourceType.Value
-					}
-				}
-			}
 
 		} else if task.JobType == string(models.JobTypeInference) {
 			if inferenceGpuInfos == nil {
@@ -770,6 +770,16 @@ func cloudBrainShow(ctx *context.Context, tpName base.TplName, jobType models.Jo
 				}
 			}
 		}
+
+		if cloudbrain.SpecialPools != nil {
+			for _, specialPool := range cloudbrain.SpecialPools.Pools {
+				for _, resourceType := range specialPool.Pool {
+					if resourceType.Queue == jobRes.Config.GpuType {
+						ctx.Data["resource_type"] = resourceType.Value
+					}
+				}
+			}
+		}
 		taskRoles := jobRes.TaskRoles
 		taskRes, _ := models.ConvertToTaskPod(taskRoles[cloudbrain.SubTaskName].(map[string]interface{}))
 		ctx.Data["taskRes"] = taskRes

routers/repo/grampus.go

@@ -626,7 +626,7 @@ func GrampusTrainJobShow(ctx *context.Context) {
 	task, err := models.GetCloudbrainByJobIDWithDeleted(ctx.Params(":jobid"))
 	if err != nil {
 		log.Error("GetCloudbrainByJobID failed:" + err.Error())
-		ctx.ServerError("system error", err)
+		ctx.NotFound(ctx.Req.URL.RequestURI(), nil)
 		return
 	}
 
@@ -634,8 +634,8 @@ func GrampusTrainJobShow(ctx *context.Context) {
 		result, err := grampus.GetJob(task.JobID)
 		if err != nil {
 			log.Error("GetJob failed:" + err.Error())
-			//ctx.ServerError("GetJob failed", err)
-			//return
+			ctx.NotFound(ctx.Req.URL.RequestURI(), nil)
+			return
 		}
 
 		if result != nil {

routers/repo/modelarts.go

@@ -257,15 +257,15 @@ func NotebookShow(ctx *context.Context) {
 	var ID = ctx.Params(":id")
 	task, err := models.GetCloudbrainByIDWithDeleted(ID)
 	if err != nil {
-		ctx.Data["error"] = err.Error()
-		ctx.RenderWithErr(err.Error(), tplModelArtsNotebookShow, nil)
+		log.Error("GET job error", err.Error())
+		ctx.NotFound(ctx.Req.URL.RequestURI(), nil)
 		return
 	}
 
 	result, err := modelarts.GetNotebook2(task.JobID)
 	if err != nil {
-		ctx.Data["error"] = err.Error()
-		ctx.RenderWithErr(err.Error(), tplModelArtsNotebookShow, nil)
+		log.Error("GET job error", err.Error())
+		ctx.NotFound(ctx.Req.URL.RequestURI(), nil)
 		return
 	}
 
@@ -276,8 +276,8 @@ func NotebookShow(ctx *context.Context) {
 				models.ParseAndSetDurationFromModelArtsNotebook(result, task)
 				err = models.UpdateJob(task)
 				if err != nil {
-					ctx.Data["error"] = err.Error()
-					ctx.RenderWithErr(err.Error(), tplModelArtsNotebookShow, nil)
+					log.Error("GET job error", err.Error())
+					ctx.NotFound(ctx.Req.URL.RequestURI(), nil)
 					return
 				}
 			}
@@ -1643,7 +1643,11 @@ func TrainJobShow(ctx *context.Context) {
 
 	if err != nil {
 		log.Error("GetVersionListTasks(%s) failed:%v", jobID, err.Error())
-		ctx.RenderWithErr(err.Error(), tplModelArtsTrainJobShow, nil)
+		ctx.NotFound(ctx.Req.URL.RequestURI(), nil)
+		return
+	}
+	if len(VersionListTasks) == 0 {
+		ctx.NotFound(ctx.Req.URL.RequestURI(), nil)
 		return
 	}
 	//设置权限
@@ -2297,7 +2301,7 @@ func InferenceJobShow(ctx *context.Context) {
 
 	if err != nil {
 		log.Error("GetInferenceTask(%s) failed:%v", jobID, err.Error())
-		ctx.RenderWithErr(err.Error(), tplModelArtsInferenceJobShow, nil)
+		ctx.NotFound(ctx.Req.URL.RequestURI(), nil)
 		return
 	}
 	//设置权限