debug

routers/repo/modelarts.go

@@ -2230,13 +2230,13 @@ func ModelDownload(ctx *context.Context) {
 		err error
 	)
 
-	var jobID = ctx.Params(":jobid")
+	jobID := ctx.Params(":jobid")
 	versionName := ctx.Query("version_name")
 	parentDir := ctx.Query("parent_dir")
 	fileName := ctx.Query("file_name")
 	task, err := models.GetCloudbrainByJobIDAndVersionName(jobID, versionName)
 	if err != nil {
-		log.Error("GetCloudbrainByJobID(%s) failed:%v", task.JobName, err.Error())
+		log.Error("GetCloudbrainByJobIDAndVersionName(%s) failed:%v", task.JobName, err.Error())
 		return
 	}
 
@@ -2250,11 +2250,11 @@ func ModelDownload(ctx *context.Context) {
 			return
 		}
 	} else if task.ComputeResource == models.GPUResource {
-		path := strings.TrimPrefix(path.Join(setting.TrainJobModelPath, task.JobName, setting.OutPutPath, versionName, parentDir, fileName), "/")
-		url, err = storage.GetObsCreateSignedUrlByBucketAndKey(setting.Bucket, path)
+		filePath := setting.CBCodePathPrefix + task.JobName + cloudbrain.ModelMountPath + "/" + parentDir
+		url, err = storage.Attachments.PresignedGetURL(filePath, fileName)
 		if err != nil {
-			log.Error("GetObsCreateSignedUrl failed: %v", err.Error(), ctx.Data["msgID"])
-			ctx.ServerError("GetObsCreateSignedUrl", err)
+			log.Error("PresignedGetURL failed: %v", err.Error(), ctx.Data["msgID"])
+			ctx.ServerError("PresignedGetURL", err)
 			return
 		}
 	}