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