From 76a84d89d63041232c646ca28c59239dd00f7fc5 Mon Sep 17 00:00:00 2001 From: liujiandao <274878379@qq.com> Date: 星期二, 30 四月 2024 09:58:46 +0800 Subject: [PATCH] 薪资计算 --- models/mini_dict.go | 20 ++++++++++++++++++++ 1 files changed, 20 insertions(+), 0 deletions(-) diff --git a/models/mini_dict.go b/models/mini_dict.go index 729307e..8941a78 100644 --- a/models/mini_dict.go +++ b/models/mini_dict.go @@ -20,6 +20,7 @@ MiniDictSearch struct { MiniDict Order string + Ids []uint PageNum int PageSize int Orm *gorm.DB @@ -54,6 +55,11 @@ return slf } +func (slf *MiniDictSearch) SetIds(ids []uint) *MiniDictSearch { + slf.Ids = ids + return slf +} + func (slf *MiniDictSearch) SetType(tp constvar.MiniDictType) *MiniDictSearch { slf.Type = tp return slf @@ -74,6 +80,10 @@ if slf.ID > 0 { db = db.Where("id = ?", slf.ID) + } + + if len(slf.Ids) > 0 { + db = db.Where("id in (?)", slf.Ids) } if slf.Type > 0 { @@ -128,6 +138,16 @@ return nil } +func (slf *MiniDictSearch) SaveBatch(record []*MiniDict) error { + var db = slf.build() + + if err := db.Omit("CreatedAt").Save(record).Error; err != nil { + return fmt.Errorf("save err: %v, record: %+v", err, record) + } + + return nil +} + func (slf *MiniDictSearch) UpdateByMap(upMap map[string]interface{}) error { var ( db = slf.build() -- Gitblit v1.8.0