zhangqian
2023-11-01 530fed8ec225453572d57b15c200ab062c335457
model/client.go
@@ -1,7 +1,9 @@
package model
import (
   "aps_crm/constvar"
   "aps_crm/pkg/mysqlx"
   "fmt"
   "gorm.io/gorm"
   "time"
)
@@ -9,37 +11,41 @@
type (
   Client struct {
      Id                int            `json:"id" gorm:"column:id;primaryKey;autoIncrement;not null"`
      Name              string         `json:"name" gorm:"column:name;unique;type:varchar(255);comment:客户名称"`
      Name              string         `json:"name" gorm:"column:name;uniqueIndex:name_isDeleted_idx;type:varchar(255);comment:客户名称"`
      IsDeleted         bool           `json:"-" gorm:"column:is_deleted;uniqueIndex:name_isDeleted_idx;type:tinyint(1);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:客户状态ID"`
      ClientStatusId    int            `json:"client_status_id" gorm:"column:client_status_id;type:int(11);default:1;comment:客户状态ID"`
      ClientStatus      ClientStatus   `json:"client_status" gorm:"foreignKey:ClientStatusId"`
      ClientTypeId      int            `json:"client_type_id" gorm:"column:client_type_id;type:int(11);comment:客户类型ID"`
      ClientTypeId      int            `json:"client_type_id" gorm:"column:client_type_id;type:int(11);default:1;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"`
      ClientOriginId    int            `json:"client_origin_id" gorm:"column:client_origin_id;type:int(11);default:1;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"`
      ClientLevelId     int            `json:"client_level_id" gorm:"column:client_level_id;type:int(11);default:1;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:销售负责人ID"`
      Member            User           `json:"member" gorm:"foreignKey:MemberId"`
      ServiceMemberId   int            `json:"service_member_id" gorm:"column:service_member_id;type:int(11);comment:服务负责人ID"`
      DetailAddress     string         `json:"detail_address" gorm:"column:detail_address;type:varchar(255);comment:详细地址"`
      Remark            string         `json:"remark" gorm:"column:remark;type:varchar(255);comment:备注"`
      NextVisitTime     time.Time      `json:"next_visit_time" gorm:"column:next_visit_time;type:datetime;comment:下次回访时间"`
      LatestServiceTime time.Time      `json:"latest_service_time" gorm:"column:latest_service_time;type:datetime;comment:最晚服务时间"`
      CreatorId         int            `json:"creator_id" gorm:"column:creator_id;type:int;comment:创建人id"`
      NextVisitTime     *CustomTime    `json:"next_visit_time" gorm:"column:next_visit_time;type:datetime;comment:下次回访时间"`
      LatestServiceTime *CustomTime    `json:"latest_service_time" gorm:"column:latest_service_time;type:datetime;comment:最晚服务时间"`
      FollowRecord      []FollowRecord `json:"follow_record" gorm:"foreignKey:ClientId"`
      CodeStandID       string         `json:"codeStandID" gorm:"column:code_stand_id;type:varchar(255);comment:编码id"`
      Address
      Business
      gorm.Model `json:"-"`
      CrmModel
   }
   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 +65,77 @@
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 slf.Number != "" {
      db.Where("number = ?", slf.Number)
   }
   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" || key == "remark" || key == "business_scope" || key == "registration_time" || key == "representative" {
               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("clients.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, uint, int64, float64, bool:
            if key == "id" || key == "client_type_id" || key == "client_status_id" || key == "member_id" {
               db = db.Where(key+" = ?", v)
            }
            if key == "is_public" {
               if val, ok := v.(bool); ok {
                  if val {
                     db = db.Where("clients.member_id = ?", 0)
                  }
               }
            }
         case []int:
            if key == "member_ids" {
               db = db.Where("clients.member_id in ?", v)
            }
         }
      }
   }
   return db
@@ -84,6 +153,7 @@
func (slf *ClientSearch) Delete() error {
   var db = slf.build()
   db.Update("is_deleted", 1)
   return db.Delete(&Client{}).Error
}
@@ -94,6 +164,11 @@
func (slf *ClientSearch) SetName(name string) *ClientSearch {
   slf.Name = name
   return slf
}
func (slf *ClientSearch) SetSearchMap(data map[string]interface{}) *ClientSearch {
   slf.SearchMap = data
   return slf
}
@@ -120,7 +195,7 @@
      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").Order("id desc").Find(&records).Error
   return records, total, err
}
@@ -129,9 +204,28 @@
   return db.Updates(data).Error
}
func (slf *ClientSearch) SetKeyword(keyword string) *ClientSearch {
   slf.Keyword = keyword
   return slf
func (slf *ClientSearch) Count() (int64, error) {
   var db = slf.build()
   var total int64
   err := db.Count(&total).Error
   return total, err
}
func (slf *ClientSearch) MaxAutoIncr() (int, error) {
   type Result struct {
      Max int
   }
   var (
      result Result
      db     = slf.build()
   )
   err := db.Select("MAX(id) as max").Scan(&result).Error
   if err != nil {
      return result.Max, fmt.Errorf("max err: %v", err)
   }
   return result.Max, nil
}
func (slf *ClientSearch) SetPage(page, size int) *ClientSearch {