From 72150f8f03abde8c63253a6a3ffdfffaae8def3f Mon Sep 17 00:00:00 2001 From: chenyifan01 Date: Wed, 14 Dec 2022 10:24:04 +0800 Subject: [PATCH 1/2] #3364 repo-square:fix bug --- routers/repo/ai_model_manage.go | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go index 605284430..b55c92885 100644 --- a/routers/repo/ai_model_manage.go +++ b/routers/repo/ai_model_manage.go @@ -122,9 +122,6 @@ func saveModelByParameters(jobId string, versionName string, name string, versio if err != nil { return "", err } - if modelSize > 0 { - go repository.IncreaseRepoModelNum(aiTask.RepoID) - } if len(lastNewModelId) > 0 { //udpate status and version count models.ModifyModelNewProperty(lastNewModelId, MODEL_NOT_LATEST, 0) @@ -178,6 +175,13 @@ func updateStatus(id string, modelSize int64, status int, modelPath string, stat if err != nil { log.Info("update status error." + err.Error()) } + m, err := models.QueryModelById(id) + if err == nil { + if modelSize > 0 && m.Size == 0 { + go repository.IncreaseRepoModelNum(m.RepoId) + } + } + } func SaveNewNameModel(ctx *context.Context) { From c0888e04830c483a8ad07be2241e0cadee4d5e3d Mon Sep 17 00:00:00 2001 From: chenyifan01 Date: Wed, 14 Dec 2022 11:38:34 +0800 Subject: [PATCH 2/2] #3364 repo-square:fix bug --- routers/repo/ai_model_manage.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go index b55c92885..a45f21cca 100644 --- a/routers/repo/ai_model_manage.go +++ b/routers/repo/ai_model_manage.go @@ -171,12 +171,12 @@ func updateStatus(id string, modelSize int64, status int, modelPath string, stat if len(statusDesc) > 400 { statusDesc = statusDesc[0:400] } + m, _ := models.QueryModelById(id) err := models.ModifyModelStatus(id, modelSize, status, modelPath, statusDesc) if err != nil { log.Info("update status error." + err.Error()) } - m, err := models.QueryModelById(id) - if err == nil { + if m != nil { if modelSize > 0 && m.Size == 0 { go repository.IncreaseRepoModelNum(m.RepoId) }