From 92c2d8f30f9615a53f3b3e3dd64e4fcedca0f9da Mon Sep 17 00:00:00 2001 From: wubijie Date: Tue, 10 Jan 2023 10:00:58 +0800 Subject: [PATCH] Add database error handling that IsExistName --- pkg/app/server/dao/batchmanagerdao.go | 9 ++++++--- pkg/app/server/service/batchservice.go | 8 +++++--- 2 files changed, 11 insertions(+), 6 deletions(-) diff --git a/pkg/app/server/dao/batchmanagerdao.go b/pkg/app/server/dao/batchmanagerdao.go index 8f6fd081..2aa66d54 100644 --- a/pkg/app/server/dao/batchmanagerdao.go +++ b/pkg/app/server/dao/batchmanagerdao.go @@ -8,10 +8,13 @@ import ( "openeuler.org/PilotGo/PilotGo/pkg/utils" ) -func IsExistName(name string) bool { +func IsExistName(name string) (bool, error) { var batch model.Batch - global.PILOTGO_DB.Where("name=?", name).Find(&batch) - return batch.ID != 0 + err := global.PILOTGO_DB.Where("name=?", name).Find(&batch).Error + if err != nil { + return batch.ID != 0, err + } + return batch.ID != 0, nil } func IsExistID(id int) bool { var batch model.Batch diff --git a/pkg/app/server/service/batchservice.go b/pkg/app/server/service/batchservice.go index 4bcaed70..93008ab9 100644 --- a/pkg/app/server/service/batchservice.go +++ b/pkg/app/server/service/batchservice.go @@ -28,11 +28,13 @@ func CreateBatch(batchinfo *model.CreateBatch) error { if len(batchinfo.Name) == 0 { return errors.New("请输入批次名称") } - - if dao.IsExistName(batchinfo.Name) { + ExistNameBool, err := dao.IsExistName(batchinfo.Name) + if err != nil { + return err + } + if ExistNameBool { return errors.New("已存在该名称批次") } - if len(batchinfo.Manager) == 0 { return errors.New("创建人未输入") } -- Gitee