From 00b3f9dc46d100fb190a2d323490f848c98e014e Mon Sep 17 00:00:00 2001 From: zouap Date: Wed, 18 Jan 2023 15:15:56 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=AD=A3SQL=E6=B3=A8=E5=85=A5?= =?UTF-8?q?=E7=9A=84=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: zouap --- models/ai_model_manage.go | 20 ++++++++++++-------- 1 file changed, 12 insertions(+), 8 deletions(-) diff --git a/models/ai_model_manage.go b/models/ai_model_manage.go index 98c67274a..c1ab3afab 100644 --- a/models/ai_model_manage.go +++ b/models/ai_model_manage.go @@ -237,15 +237,19 @@ func QueryModelConvertByName(name string, repoId int64) ([]*AiModelConvert, erro func QueryModelConvertById(id string) (*AiModelConvert, error) { sess := x.NewSession() defer sess.Close() - sess.Table(new(AiModelConvert)).ID(id) - //sess.Select("*").Table(new(AiModelConvert)).Where("id = ?", id) - aiModelManageConvertList := make([]*AiModelConvert, 0) - err := sess.Find(&aiModelManageConvertList) - if err == nil { - if len(aiModelManageConvertList) == 1 { - return aiModelManageConvertList[0], nil - } + re := new(AiModelConvert) + isExist, err := sess.Table(new(AiModelConvert)).ID(id).Get(re) + if isExist { + return re, nil } + //sess.Select("*").Table(new(AiModelConvert)).Where("id = ?", id) + //aiModelManageConvertList := make([]*AiModelConvert, 0) + //err := sess.Find(&aiModelManageConvertList) + //if err == nil { + // if len(aiModelManageConvertList) == 1 { + // return aiModelManageConvertList[0], nil + // } + //} return nil, err }