From b0d1fe4e0acb0fa40c684069c0e9d77d7a65ace3 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期二, 17 十月 2023 15:53:53 +0800
Subject: [PATCH] fix

---
 model/salesReturn.go |   34 +++++++++++++++++++++++++++++++++-
 1 files changed, 33 insertions(+), 1 deletions(-)

diff --git a/model/salesReturn.go b/model/salesReturn.go
index e8babad..df99ba7 100644
--- a/model/salesReturn.go
+++ b/model/salesReturn.go
@@ -46,6 +46,7 @@
 		PageSize    int
 		Preload     bool
 		Ids         []int
+		MemberIds   []int
 	}
 )
 
@@ -71,7 +72,15 @@
 	if len(slf.Ids) != 0 {
 		db = db.Where("id in ?", slf.Ids)
 	}
-
+	if slf.SourceType != 0 {
+		db = db.Where("source_type = ?", slf.SourceType)
+	}
+	if slf.SourceId != 0 {
+		db = db.Where("source_id = ?", slf.SourceId)
+	}
+	if slf.ClientId != 0 {
+		db = db.Where("client_id = ?", slf.ClientId)
+	}
 	if slf.Preload {
 		db = db.Preload("Client").
 			Preload("Member").
@@ -100,6 +109,10 @@
 			db = db.Where("amount_has_refund = ?", slf.Keyword)
 
 		}
+	}
+
+	if len(slf.MemberIds) > 0 {
+		db = db.Where("sales_return.member_id in ?", slf.MemberIds)
 	}
 
 	return db
@@ -189,8 +202,27 @@
 	slf.Preload = preload
 	return slf
 }
+func (slf *SalesReturnSearch) SetSourceType(sourceType constvar.SalesReturnSourceType) *SalesReturnSearch {
+	slf.SourceType = sourceType
+	return slf
+}
+
+func (slf *SalesReturnSearch) SetSourceId(sourceId int) *SalesReturnSearch {
+	slf.SourceId = sourceId
+	return slf
+}
+
+func (slf *SalesReturnSearch) SetClientId(clientId int) *SalesReturnSearch {
+	slf.ClientId = clientId
+	return slf
+}
 
 func (slf *SalesReturnSearch) UpdateByMap(data map[string]interface{}) error {
 	var db = slf.build()
 	return db.Updates(data).Error
 }
+
+func (slf *SalesReturnSearch) SetMemberIds(memberIds []int) *SalesReturnSearch {
+	slf.MemberIds = memberIds
+	return slf
+}

--
Gitblit v1.8.0