From 11af32523705ea3e550abbb7477a8f176382e6f7 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期五, 25 八月 2023 11:05:57 +0800
Subject: [PATCH] Merge branch 'hotfix'
---
model/receipt.go | 21 +++++++++++++++++++++
1 files changed, 21 insertions(+), 0 deletions(-)
diff --git a/model/receipt.go b/model/receipt.go
index 9ec62f4..5eb54df 100644
--- a/model/receipt.go
+++ b/model/receipt.go
@@ -53,11 +53,32 @@
}
}
+func (slf *ReceiptSearch) SetSourceType(sourceType constvar.ReceiptSourceType) *ReceiptSearch {
+ slf.SourceType = sourceType
+ return slf
+}
+
+func (slf *ReceiptSearch) SetSourceId(sourceId int) *ReceiptSearch {
+ slf.SourceId = sourceId
+ return slf
+}
+
+func (slf *ReceiptSearch) SetPage(page, size int) *ReceiptSearch {
+ slf.PageNum, slf.PageSize = page, size
+ return slf
+}
+
func (slf *ReceiptSearch) build() *gorm.DB {
var db = slf.Orm.Model(&Receipt{})
if slf.Id != 0 {
db = db.Where("id = ?", slf.Id)
}
+ if slf.SourceType != 0 {
+ db = db.Where("source_type = ?", slf.SourceType)
+ }
+ if slf.SourceId != 0 {
+ db = db.Where("source_id = ?", slf.SourceId)
+ }
return db
}
--
Gitblit v1.8.0