From 1129c91762fdb8c9940d87ca300851bc3065eef2 Mon Sep 17 00:00:00 2001 From: wangpengfei <274878379@qq.com> Date: 星期五, 11 八月 2023 16:58:35 +0800 Subject: [PATCH] add --- model/saleChance.go | 64 +++++++++++++++++++++++++------ 1 files changed, 51 insertions(+), 13 deletions(-) diff --git a/model/saleChance.go b/model/saleChance.go index 3667873..d5c444d 100644 --- a/model/saleChance.go +++ b/model/saleChance.go @@ -48,11 +48,11 @@ SaleChanceSearch struct { SaleChance - Orm *gorm.DB - Keyword string - OrderBy string - PageNum int - PageSize int + Orm *gorm.DB + SearchMap map[string]interface{} + OrderBy string + PageNum int + PageSize int } ) @@ -66,14 +66,52 @@ func (slf *SaleChanceSearch) build() *gorm.DB { var db = slf.Orm.Model(&SaleChance{}) - if slf.Keyword != "" { - db = db.Where("name LIKE ?", "%"+slf.Keyword+"%") - } if slf.Id > 0 { db = db.Where("id = ?", slf.Id) } if slf.Name != "" { db = db.Where("name = ?", slf.Name) + } + + if len(slf.SearchMap) > 0 { + for key, value := range slf.SearchMap { + switch v := value.(type) { + case string: + if key == "name" || key == "number" { + db = db.Where(key+" LIKE ?", "%"+v+"%") + } + + if key == "client_name" { + db = db.Joins("Client").Where("Client.name LIKE ?", "%"+v+"%") + } + + if key == "contact_name" { + db = db.Joins("Contact").Where("Contact.name LIKE ?", "%"+v+"%") + } + + if key == "member_name" { + db = db.Joins("User").Where("User.name LIKE ?", "%"+v+"%") + } + + if key == "sale_stage" { + db = db.Joins("SaleStage").Where("SaleStage.name LIKE ?", "%"+v+"%") + } + + if key == "possibilities" { + db = db.Joins("Possibilities").Where("Possibilities.name LIKE ?", "%"+v+"%") + } + + if key == "expected_time" { + db = db.Where("expected_time LIKE ?", "%"+v+"%") + } + + case int: + case float64: + if key == "member_id" || key == "budget" || key == "projected_amount" { + db = db.Where(key+" = ?", v) + } + } + } } return db @@ -128,11 +166,6 @@ return slf } -func (slf *SaleChanceSearch) SetKeyword(keyword string) *SaleChanceSearch { - slf.Keyword = keyword - return slf -} - func (slf *SaleChanceSearch) SetPage(page, size int) *SaleChanceSearch { slf.PageNum, slf.PageSize = page, size return slf @@ -142,3 +175,8 @@ slf.OrderBy = order return slf } + +func (slf *SaleChanceSearch) SetSearchMap(searchMap map[string]interface{}) *SaleChanceSearch { + slf.SearchMap = searchMap + return slf +} -- Gitblit v1.8.0