From 76a84d89d63041232c646ca28c59239dd00f7fc5 Mon Sep 17 00:00:00 2001
From: liujiandao <274878379@qq.com>
Date: 星期二, 30 四月 2024 09:58:46 +0800
Subject: [PATCH] 薪资计算
---
models/yield_register_item.go | 21 ++++++++++++++++++++-
1 files changed, 20 insertions(+), 1 deletions(-)
diff --git a/models/yield_register_item.go b/models/yield_register_item.go
index e9a643a..92720c5 100644
--- a/models/yield_register_item.go
+++ b/models/yield_register_item.go
@@ -20,7 +20,9 @@
}
YieldRegisterItemSearch struct {
YieldRegisterItem
- Orm *gorm.DB
+ YieldRegisterIds []uint
+ CarNumbers []int
+ Orm *gorm.DB
}
)
@@ -42,6 +44,15 @@
return slf
}
+func (slf *YieldRegisterItemSearch) SetYieldRegisterIds(ids []uint) *YieldRegisterItemSearch {
+ slf.YieldRegisterIds = ids
+ return slf
+}
+func (slf *YieldRegisterItemSearch) SetCarNumbers(carNumbers []int) *YieldRegisterItemSearch {
+ slf.CarNumbers = carNumbers
+ return slf
+}
+
func (slf *YieldRegisterItemSearch) build() *gorm.DB {
db := slf.Orm.Table(slf.TableName())
@@ -49,6 +60,14 @@
db = db.Where("yield_register_id = ?", slf.YieldRegisterId)
}
+ if len(slf.YieldRegisterIds) > 0 {
+ db = db.Where("yield_register_id in (?)", slf.YieldRegisterIds)
+ }
+
+ if len(slf.CarNumbers) > 0 {
+ db = db.Where("car_number in (?)", slf.CarNumbers)
+ }
+
return db
}
--
Gitblit v1.8.0