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