From be953da6ecba5036e63eac2bb5851e04dca1e8f7 Mon Sep 17 00:00:00 2001
From: wangpengfei <274878379@qq.com>
Date: 星期五, 11 八月 2023 14:06:56 +0800
Subject: [PATCH] fix

---
 model/collectionProjection.go         |   29 +++++++++++++++++++++++++----
 service/collectionProjection.go       |    4 ++--
 api/v1/collectionProjection.go        |    2 +-
 model/contact.go                      |    3 +++
 model/request/collectionProjection.go |    1 +
 5 files changed, 32 insertions(+), 7 deletions(-)

diff --git a/api/v1/collectionProjection.go b/api/v1/collectionProjection.go
index 49fb8a1..2caa97b 100644
--- a/api/v1/collectionProjection.go
+++ b/api/v1/collectionProjection.go
@@ -125,7 +125,7 @@
 		return
 	}
 
-	total, errCode, list := collectionProjectionService.GetCollectionProjectionList(params.Page, params.PageSize)
+	total, errCode, list := collectionProjectionService.GetCollectionProjectionList(params.Page, params.PageSize, params.SearchMap)
 	if errCode != ecode.OK {
 		ctx.Fail(errCode)
 		return
diff --git a/model/collectionProjection.go b/model/collectionProjection.go
index 2e160a2..de10709 100644
--- a/model/collectionProjection.go
+++ b/model/collectionProjection.go
@@ -19,10 +19,11 @@
 	CollectionProjectionSearch struct {
 		CollectionProjection
 
-		Orm      *gorm.DB
-		OrderBy  string
-		PageNum  int
-		PageSize int
+		Orm       *gorm.DB
+		SearchMap map[string]interface{}
+		OrderBy   string
+		PageNum   int
+		PageSize  int
 	}
 )
 
@@ -40,6 +41,21 @@
 	var db = slf.Orm.Model(&CollectionProjection{})
 	if slf.Id != 0 {
 		db = db.Where("id = ?", slf.Id)
+	}
+
+	if len(slf.SearchMap) > 0 {
+		for key, value := range slf.SearchMap {
+			switch v := value.(type) {
+			case string:
+				if key == "estimated_collection_date" {
+					db = db.Where(key+" = ?", v)
+				}
+			case int:
+				if key == "client_id" {
+					db = db.Where(key+" = ?", v)
+				}
+			}
+		}
 	}
 
 	return db
@@ -92,3 +108,8 @@
 	slf.PageNum, slf.PageSize = page, size
 	return slf
 }
+
+func (slf *CollectionProjectionSearch) SetSearchMap(data map[string]interface{}) *CollectionProjectionSearch {
+	slf.SearchMap = data
+	return slf
+}
diff --git a/model/contact.go b/model/contact.go
index 612fcba..560db43 100644
--- a/model/contact.go
+++ b/model/contact.go
@@ -93,6 +93,9 @@
 					}
 				}
 			case int:
+				if key == "client_id" {
+					db = db.Where("client_id = ?", v)
+				}
 			}
 		}
 	}
diff --git a/model/request/collectionProjection.go b/model/request/collectionProjection.go
index f60a1b1..579f593 100644
--- a/model/request/collectionProjection.go
+++ b/model/request/collectionProjection.go
@@ -17,4 +17,5 @@
 
 type GetCollectionProjectionList struct {
 	PageInfo
+	SearchMap map[string]interface{}
 }
diff --git a/service/collectionProjection.go b/service/collectionProjection.go
index 7da5f09..e482aa4 100644
--- a/service/collectionProjection.go
+++ b/service/collectionProjection.go
@@ -34,9 +34,9 @@
 	return ecode.OK
 }
 
-func (CollectionProjectionService) GetCollectionProjectionList(page, pageSize int) (int64, int, []*model.CollectionProjection) {
+func (CollectionProjectionService) GetCollectionProjectionList(page, pageSize int, data map[string]interface{}) (int64, int, []*model.CollectionProjection) {
 
-	total, err, list := model.NewCollectionProjectionSearch().SetPage(page, pageSize).Find()
+	total, err, list := model.NewCollectionProjectionSearch().SetPage(page, pageSize).SetSearchMap(data).Find()
 	if err != nil {
 		return total, ecode.CollectionProjectionListErr, list
 	}

--
Gitblit v1.8.0