提交代码

modules/cloudbrain/cloudbrain.go

@@ -5,6 +5,7 @@ import (
 	"errors"
 	"os"
 	"strconv"
+	"strings"
 
 	"code.gitea.io/gitea/modules/timeutil"
 
@@ -145,7 +146,7 @@ func isAdminOrImageCreater(ctx *context.Context, image *models.Image, err error)
 func AdminOrOwnerOrJobCreaterRight(ctx *context.Context) {
 
 	var id = ctx.Params(":id")
-	job, err := GetCloudBrainByIdOrJobId(id)
+	job, err := GetCloudBrainByIdOrJobId(id, "id")
 	if err != nil {
 		log.Error("GetCloudbrainByID failed:%v", err.Error())
 		ctx.NotFound(ctx.Req.URL.RequestURI(), nil)
@@ -161,7 +162,7 @@ func AdminOrOwnerOrJobCreaterRight(ctx *context.Context) {
 func AdminOrJobCreaterRight(ctx *context.Context) {
 
 	var id = ctx.Params(":id")
-	job, err := GetCloudBrainByIdOrJobId(id)
+	job, err := GetCloudBrainByIdOrJobId(id, "id")
 	if err != nil {
 		log.Error("GetCloudbrainByID failed:%v", err.Error())
 		ctx.NotFound(ctx.Req.URL.RequestURI(), nil)
@@ -177,7 +178,7 @@ func AdminOrJobCreaterRight(ctx *context.Context) {
 func AdminOrOwnerOrJobCreaterRightForTrain(ctx *context.Context) {
 
 	var jobID = ctx.Params(":jobid")
-	job, err := GetCloudBrainByIdOrJobId(jobID)
+	job, err := GetCloudBrainByIdOrJobId(jobID, "jobid")
 	if err != nil {
 		log.Error("GetCloudbrainByJobID failed:%v", err.Error())
 		ctx.NotFound(ctx.Req.URL.RequestURI(), nil)
@@ -193,7 +194,7 @@ func AdminOrOwnerOrJobCreaterRightForTrain(ctx *context.Context) {
 func AdminOrJobCreaterRightForTrain(ctx *context.Context) {
 
 	var jobID = ctx.Params(":jobid")
-	job, err := GetCloudBrainByIdOrJobId(jobID)
+	job, err := GetCloudBrainByIdOrJobId(jobID, "jobid")
 	if err != nil {
 		log.Error("GetCloudbrainByJobID failed:%v", err.Error())
 		ctx.NotFound(ctx.Req.URL.RequestURI(), nil)
@@ -499,7 +500,7 @@ func RestartTask(ctx *context.Context, task *models.Cloudbrain, newID *string) e
 			volumes = append(volumes, models.Volume{
 				HostPath: models.StHostPath{
 					Path:      realPath,
-					MountPath: ModelMountPath,
+					MountPath: PretrainModelMountPath,
 					ReadOnly:  true,
 				},
 			})
@@ -669,18 +670,26 @@ func IsElementExist(s []string, str string) bool {
 	return false
 }
 
-func GetCloudBrainByIdOrJobId(id string) (*models.Cloudbrain,error) {
+func GetCloudBrainByIdOrJobId(id string, initialQuery string) (*models.Cloudbrain, error) {
 	_, err := strconv.ParseInt(id, 10, 64)
 	var job *models.Cloudbrain
 	if err != nil {
 
 		job, err = models.GetCloudbrainByJobID(id)
 	} else {
-		job, err = models.GetCloudbrainByID(id)
-		if err!=nil{
+
+		if strings.EqualFold(initialQuery, "id") {
+			job, err = models.GetCloudbrainByID(id)
+			if err != nil {
+				job, err = models.GetCloudbrainByJobID(id)
+			}
+		} else {
 			job, err = models.GetCloudbrainByJobID(id)
+			if err != nil {
+				job, err = models.GetCloudbrainByID(id)
+			}
 		}
 
 	}
-	return job,err
+	return job, err
 }

routers/api/v1/repo/cloudbrain.go

@@ -211,7 +211,7 @@ func GetCloudbrainTask(ctx *context.APIContext) {
 
 	ID := ctx.Params(":id")
 
-	job, err := cloudbrain.GetCloudBrainByIdOrJobId(ID)
+	job, err := cloudbrain.GetCloudBrainByIdOrJobId(ID, "id")
 
 	if err != nil {
 		ctx.NotFound(err)

routers/api/v1/repo/modelarts.go

@@ -39,7 +39,7 @@ func GetModelArtsNotebook2(ctx *context.APIContext) {
 
 	ID := ctx.Params(":id")
 
-	job,err := cloudbrain.GetCloudBrainByIdOrJobId(ID)
+	job, err := cloudbrain.GetCloudBrainByIdOrJobId(ID, "id")
 
 	if err != nil {
 		ctx.NotFound(err)