From ff3e203423f296965472d1f09347cda5cfe3a786 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期五, 11 八月 2023 11:37:19 +0800 Subject: [PATCH] update --- model/client.go | 87 +++++++++++++++++++++++++++++++++++-------- 1 files changed, 70 insertions(+), 17 deletions(-) diff --git a/model/client.go b/model/client.go index 2dc0889..7cec5b9 100644 --- a/model/client.go +++ b/model/client.go @@ -1,6 +1,7 @@ package model import ( + "aps_crm/constvar" "aps_crm/pkg/mysqlx" "gorm.io/gorm" "time" @@ -21,6 +22,7 @@ ClientLevel ClientLevel `json:"client_level" gorm:"foreignKey:ClientLevelId"` Contacts []Contact `json:"contacts" gorm:"foreignKey:ClientId"` MemberId int `json:"member_id" gorm:"column:member_id;type:int(11);comment:閿�鍞礋璐d汉ID"` + Member User `json:"member" gorm:"foreignKey:MemberId"` ServiceMemberId int `json:"service_member_id" gorm:"column:service_member_id;type:int(11);comment:鏈嶅姟璐熻矗浜篒D"` DetailAddress string `json:"detail_address" gorm:"column:detail_address;type:varchar(255);comment:璇︾粏鍦板潃"` Remark string `json:"remark" gorm:"column:remark;type:varchar(255);comment:澶囨敞"` @@ -35,11 +37,11 @@ ClientSearch struct { Client - Orm *gorm.DB - Keyword string - OrderBy string - PageNum int - PageSize int + Orm *gorm.DB + SearchMap map[string]interface{} + OrderBy string + PageNum int + PageSize int } ) @@ -59,14 +61,61 @@ func (slf *ClientSearch) build() *gorm.DB { var db = slf.Orm.Model(&Client{}) - if slf.Keyword != "" { - db = db.Where("name LIKE ?", "%"+slf.Keyword+"%") - } if slf.Id != 0 { db.Where("id = ?", slf.Id) } if slf.Name != "" { 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" || key == "detail_address" { + db = db.Where(key+" LIKE ?", "%"+v+"%") + } + + if key == "phone" { + db = db.Joins("Contacts").Where("Contacts.phone LIKE ?", "%"+v+"%") + } + + if key == "contact_name" { + db = db.Joins("Contacts").Where("Contacts.name LIKE ?", "%"+v+"%") + } + + if key == "next_visit_time" { + db = db.Where(key+" = ?", v) + } + + // 鏌ヨ瓒呭嚭15澶╂湭鑱旂郴鐨勫鎴� + if key == "not_contact" { + db = db.Joins("FollowRecord").Where("FollowRecord.follow_time < ?", time.Now().AddDate(0, 0, -15)).Group("clients.id") + } + + // 鏌ヨ鍏捣瀹㈡埛 + if key == "public_sea" { + db = db.Where("member_id = ?", constvar.UserTypeSuper) + } + + if key == "client_status" { + db = db.Joins("ClientStatus").Where("ClientStatus.name LIKE ?", "%"+v+"%") + } + + if key == "client_level" { + db = db.Joins("ClientLevel").Where("ClientLevel.name LIKE ?", "%"+v+"%") + } + + if key == "member_name" { + db = db.Joins("Member").Where("Member.username LIKE ?", "%"+v+"%") + } + + case int: + if key == "id" || key == "client_type_id" || key == "client_status_id" || key == "member_id" { + db = db.Where(key+" = ?", v) + } + } + } } return db @@ -97,6 +146,11 @@ return slf } +func (slf *ClientSearch) SetSearchMap(data map[string]interface{}) *ClientSearch { + slf.SearchMap = data + return slf +} + func (slf *ClientSearch) SetNumber(number string) *ClientSearch { slf.Number = number return slf @@ -112,26 +166,21 @@ func (slf *ClientSearch) Find() ([]*Client, int64, error) { var db = slf.build() var records = make([]*Client, 0) - if slf.PageNum > 0 && slf.PageSize > 0 { - db = db.Limit(slf.PageSize).Offset((slf.PageNum - 1) * slf.PageSize) - } var total int64 if err := db.Count(&total).Error; err != nil { return records, total, err } + if slf.PageNum > 0 && slf.PageSize > 0 { + db = db.Limit(slf.PageSize).Offset((slf.PageNum - 1) * slf.PageSize) + } - err := db.Preload("ClientStatus").Preload("ClientType").Preload("ClientOrigin").Preload("ClientLevel").Preload("FollowRecord").Preload("EnterpriseNature").Preload("RegisteredCapital").Preload("Industry").Preload("EnterpriseScale").Preload("Contacts").Preload("Country").Preload("Province").Preload("City").Preload("Region").Find(&records).Error + err := db.Preload("Member").Preload("ClientStatus").Preload("ClientType").Preload("ClientOrigin").Preload("ClientLevel").Preload("FollowRecord").Preload("EnterpriseNature").Preload("RegisteredCapital").Preload("Industry").Preload("EnterpriseScale").Preload("Contacts").Preload("Country").Preload("Province").Preload("City").Preload("Region").Find(&records).Error return records, total, err } func (slf *ClientSearch) UpdateByMap(data map[string]interface{}) error { var db = slf.build() return db.Updates(data).Error -} - -func (slf *ClientSearch) SetKeyword(keyword string) *ClientSearch { - slf.Keyword = keyword - return slf } func (slf *ClientSearch) SetPage(page, size int) *ClientSearch { @@ -143,3 +192,7 @@ slf.OrderBy = order return slf } +func (slf *ClientSearch) SetIds(ids []int) *ClientSearch { + slf.Orm = slf.Orm.Where("id in (?)", ids) + return slf +} -- Gitblit v1.8.0