提交代码。

Signed-off-by: zouap <zouap@pcl.ac.cn>

models/ai_model_manage.go

@@ -11,29 +11,30 @@ import (
 )
 
 type AiModelManage struct {
-	ID            string             `xorm:"pk"`
-	Name          string             `xorm:"NOT NULL"`
-	Version       string             `xorm:"NOT NULL"`
-	VersionCount  int                `xorm:"NOT NULL DEFAULT 0"`
-	New           int                `xorm:"NOT NULL"`
-	Type          int                `xorm:"NOT NULL"`
-	Size          int64              `xorm:"NOT NULL"`
-	Description   string             `xorm:"varchar(2000)"`
-	Label         string             `xorm:"varchar(1000)"`
-	Path          string             `xorm:"varchar(400) NOT NULL"`
-	DownloadCount int                `xorm:"NOT NULL DEFAULT 0"`
-	Engine        int64              `xorm:"NOT NULL DEFAULT 0"`
-	Status        int                `xorm:"NOT NULL DEFAULT 0"`
-	Accuracy      string             `xorm:"varchar(1000)"`
-	AttachmentId  string             `xorm:"NULL"`
-	RepoId        int64              `xorm:"NULL"`
-	CodeBranch    string             `xorm:"varchar(400) NULL"`
-	CodeCommitID  string             `xorm:"NULL"`
-	UserId        int64              `xorm:"NOT NULL"`
-	UserName      string             `xorm:"NULL"`
-	TrainTaskInfo string             `xorm:"text NULL"`
-	CreatedUnix   timeutil.TimeStamp `xorm:"created"`
-	UpdatedUnix   timeutil.TimeStamp `xorm:"INDEX updated"`
+	ID                string             `xorm:"pk"`
+	Name              string             `xorm:"NOT NULL"`
+	Version           string             `xorm:"NOT NULL"`
+	VersionCount      int                `xorm:"NOT NULL DEFAULT 0"`
+	New               int                `xorm:"NOT NULL"`
+	Type              int                `xorm:"NOT NULL"`
+	Size              int64              `xorm:"NOT NULL"`
+	Description       string             `xorm:"varchar(2000)"`
+	Label             string             `xorm:"varchar(1000)"`
+	Path              string             `xorm:"varchar(400) NOT NULL"`
+	DownloadCount     int                `xorm:"NOT NULL DEFAULT 0"`
+	Engine            int64              `xorm:"NOT NULL DEFAULT 0"`
+	Status            int                `xorm:"NOT NULL DEFAULT 0"`
+	Accuracy          string             `xorm:"varchar(1000)"`
+	AttachmentId      string             `xorm:"NULL"`
+	RepoId            int64              `xorm:"NULL"`
+	CodeBranch        string             `xorm:"varchar(400) NULL"`
+	CodeCommitID      string             `xorm:"NULL"`
+	UserId            int64              `xorm:"NOT NULL"`
+	UserName          string             `xorm:"NULL"`
+	UserRelAvatarLink string             `xorm:"NULL"`
+	TrainTaskInfo     string             `xorm:"text NULL"`
+	CreatedUnix       timeutil.TimeStamp `xorm:"created"`
+	UpdatedUnix       timeutil.TimeStamp `xorm:"INDEX updated"`
 }
 
 type AiModelQueryOptions struct {

routers/repo/ai_model_manage.go

@@ -25,7 +25,7 @@ const (
 	MODEL_NOT_LATEST       = 0
 )
 
-func saveModelByParameters(jobId string, versionName string, name string, version string, label string, description string, userId int64, userName string) error {
+func saveModelByParameters(jobId string, versionName string, name string, version string, label string, description string, userId int64, userName string, userHeadUrl string) error {
 	aiTask, err := models.GetCloudbrainByJobIDAndVersionName(jobId, versionName)
 	//aiTask, err := models.GetCloudbrainByJobID(jobId)
 	if err != nil {
@@ -69,25 +69,26 @@ func saveModelByParameters(jobId string, versionName string, name string, versio
 
 	//taskConfigInfo,err := models.GetCloudbrainByJobIDAndVersionName(jobId,aiTask.VersionName)
 	model := &models.AiModelManage{
-		ID:            id,
-		Version:       version,
-		VersionCount:  len(aimodels) + 1,
-		Label:         label,
-		Name:          name,
-		Description:   description,
-		New:           MODEL_LATEST,
-		Type:          cloudType,
-		Path:          modelPath,
-		Size:          modelSize,
-		AttachmentId:  aiTask.Uuid,
-		RepoId:        aiTask.RepoID,
-		UserId:        userId,
-		UserName:      userName,
-		CodeBranch:    aiTask.BranchName,
-		CodeCommitID:  aiTask.CommitID,
-		Engine:        aiTask.EngineID,
-		TrainTaskInfo: string(aiTaskJson),
-		Accuracy:      string(accuracyJson),
+		ID:                id,
+		Version:           version,
+		VersionCount:      len(aimodels) + 1,
+		Label:             label,
+		Name:              name,
+		Description:       description,
+		New:               MODEL_LATEST,
+		Type:              cloudType,
+		Path:              modelPath,
+		Size:              modelSize,
+		AttachmentId:      aiTask.Uuid,
+		RepoId:            aiTask.RepoID,
+		UserId:            userId,
+		UserName:          userName,
+		UserRelAvatarLink: userHeadUrl,
+		CodeBranch:        aiTask.BranchName,
+		CodeCommitID:      aiTask.CommitID,
+		Engine:            aiTask.EngineID,
+		TrainTaskInfo:     string(aiTaskJson),
+		Accuracy:          string(accuracyJson),
 	}
 
 	err = models.SaveModelToDb(model)
@@ -123,7 +124,7 @@ func SaveModel(ctx *context.Context) {
 		return
 	}
 
-	err := saveModelByParameters(JobId, VersionName, name, version, label, description, ctx.User.ID, ctx.User.Name)
+	err := saveModelByParameters(JobId, VersionName, name, version, label, description, ctx.User.ID, ctx.User.Name, ctx.User.RelAvatarLink())
 
 	if err != nil {
 		log.Info("save model error." + err.Error())