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 |   96 ++++++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 92 insertions(+), 4 deletions(-)

diff --git a/model/client.go b/model/client.go
index c9b994f..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"
@@ -12,11 +13,16 @@
 		Name              string         `json:"name" gorm:"column:name;unique;type:varchar(255);comment:瀹㈡埛鍚嶇О"`
 		Number            string         `json:"number" gorm:"column:number;type:varchar(255);comment:瀹㈡埛缂栧彿"`
 		ClientStatusId    int            `json:"client_status_id" gorm:"column:client_status_id;type:int(11);comment:瀹㈡埛鐘舵�両D"`
+		ClientStatus      ClientStatus   `json:"client_status" gorm:"foreignKey:ClientStatusId"`
 		ClientTypeId      int            `json:"client_type_id" gorm:"column:client_type_id;type:int(11);comment:瀹㈡埛绫诲瀷ID"`
+		ClientType        ClientType     `json:"client_type" gorm:"foreignKey:ClientTypeId"`
 		ClientOriginId    int            `json:"client_origin_id" gorm:"column:client_origin_id;type:int(11);comment:瀹㈡埛鏉ユ簮ID"`
+		ClientOrigin      ClientOrigin   `json:"client_origin" gorm:"foreignKey:ClientOriginId"`
 		ClientLevelId     int            `json:"client_level_id" gorm:"column:client_level_id;type:int(11);comment:瀹㈡埛绛夌骇ID"`
+		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:澶囨敞"`
@@ -30,7 +36,12 @@
 
 	ClientSearch struct {
 		Client
-		Orm *gorm.DB
+
+		Orm       *gorm.DB
+		SearchMap map[string]interface{}
+		OrderBy   string
+		PageNum   int
+		PageSize  int
 	}
 )
 
@@ -55,6 +66,56 @@
 	}
 	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
@@ -85,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
@@ -97,14 +163,36 @@
 	return record, err
 }
 
-func (slf *ClientSearch) Find() ([]*Client, error) {
+func (slf *ClientSearch) Find() ([]*Client, int64, error) {
 	var db = slf.build()
 	var records = make([]*Client, 0)
-	err := db.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, err
+	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("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) SetPage(page, size int) *ClientSearch {
+	slf.PageNum, slf.PageSize = page, size
+	return slf
+}
+
+func (slf *ClientSearch) SetOrder(order string) *ClientSearch {
+	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