From 9e864a183115ba2420797a8bda661187b1069e00 Mon Sep 17 00:00:00 2001 From: liujiandao <274878379@qq.com> Date: 星期二, 23 四月 2024 11:46:41 +0800 Subject: [PATCH] 考勤统计与车间修改 --- models/fineness.go | 29 ++++++++++++++++++++++++++++- 1 files changed, 28 insertions(+), 1 deletions(-) diff --git a/models/fineness.go b/models/fineness.go index c82e5a5..4ea1e11 100644 --- a/models/fineness.go +++ b/models/fineness.go @@ -37,7 +37,7 @@ ) func (slf *FinenessRegister) TableName() string { - return "fineness_register" + return "silk_fineness_register" } func NewFinenessRegisterSearch() *FinenessRegisterSearch { @@ -79,6 +79,21 @@ return slf } +func (slf *FinenessRegisterSearch) SetWorkshop(workshop string) *FinenessRegisterSearch { + slf.Workshop = workshop + return slf +} + +func (slf *FinenessRegisterSearch) SetMarket(market string) *FinenessRegisterSearch { + slf.Market = market + return slf +} + +func (slf *FinenessRegisterSearch) SetWorkshopGroup(group int) *FinenessRegisterSearch { + slf.WorkshopGroup = group + return slf +} + func (slf *FinenessRegisterSearch) build() *gorm.DB { var db = slf.Orm.Table(slf.TableName()) @@ -102,6 +117,18 @@ db = db.Where("number like ?", fmt.Sprintf("%%%s%%", slf.Keyword)) } + if slf.Workshop != "" { + db = db.Where("workshop = ?", slf.Workshop) + } + + if slf.Market != "" { + db = db.Where("market = ?", slf.Market) + } + + if slf.WorkshopGroup > 0 { + db = db.Where("workshop_group = ?", slf.WorkshopGroup) + } + return db } -- Gitblit v1.8.0