zhangqian
2023-08-14 e913ce7ca504a026ae418b5b73e3ff21001101a8
model/contact.go
@@ -3,23 +3,23 @@
import (
   "aps_crm/pkg/mysqlx"
   "gorm.io/gorm"
   "time"
)
type (
   Contact struct {
      Id       int       `json:"id" gorm:"column:id;primary_key;AUTO_INCREMENT"`
      Name     string    `json:"name" gorm:"column:name;type:varchar(255);comment:联系人姓名"`
      Number   string    `json:"number" gorm:"column:number;type:varchar(255);comment:联系人编号"`
      ClientId int       `json:"client_id" gorm:"column:client_id;type:int(11);comment:客户ID"`
      Position string    `json:"position" gorm:"column:position;type:varchar(255);comment:职位"`
      Phone    string    `json:"phone" gorm:"column:phone;type:varchar(255);comment:电话"`
      MemberId int       `json:"member_id" gorm:"column:member_id;type:int(11);comment:负责人ID"`
      IsFirst  bool      `json:"is_first" gorm:"column:is_first;type:tinyint(1);comment:是否首要联系人"`
      Wechat   string    `json:"wechat" gorm:"column:wechat;type:varchar(255);comment:微信"`
      Birthday time.Time `json:"birthday" gorm:"column:birthday;type:datetime;comment:生日"`
      Email    string    `json:"email" gorm:"column:email;type:varchar(255);comment:邮箱"`
      Desc     string    `json:"desc" gorm:"column:desc;type:varchar(255);comment:备注"`
      Id       int         `json:"id" gorm:"column:id;primary_key;AUTO_INCREMENT"`
      Name     string      `json:"name" gorm:"column:name;type:varchar(255);comment:联系人姓名"`
      Number   string      `json:"number" gorm:"column:number;type:varchar(255);comment:联系人编号"`
      ClientId int         `json:"client_id" gorm:"column:client_id;type:int(11);comment:客户ID"`
      Client   Client      `json:"-" gorm:"foreignKey:ClientId"`
      Position string      `json:"position" gorm:"column:position;type:varchar(255);comment:职位"`
      Phone    string      `json:"phone" gorm:"column:phone;type:varchar(255);comment:电话"`
      MemberId int         `json:"member_id" gorm:"column:member_id;type:int(11);comment:负责人ID"`
      IsFirst  bool        `json:"is_first" gorm:"column:is_first;type:tinyint(1);comment:是否首要联系人"`
      Wechat   string      `json:"wechat" gorm:"column:wechat;type:varchar(255);comment:微信"`
      Birthday *CustomTime `json:"birthday" gorm:"column:birthday;type:datetime;comment:生日"`
      Email    string      `json:"email" gorm:"column:email;type:varchar(255);comment:邮箱"`
      Desc     string      `json:"desc" gorm:"column:desc;type:varchar(255);comment:备注"`
      Address
      gorm.Model `json:"-"`
   }
@@ -27,11 +27,11 @@
   ContactSearch struct {
      Contact
      Orm      *gorm.DB
      Keyword  string
      OrderBy  string
      PageNum  int
      PageSize int
      Orm       *gorm.DB
      SearchMap map[string]interface{}
      OrderBy   string
      PageNum   int
      PageSize  int
   }
   ContactDetail struct {
@@ -56,9 +56,6 @@
func (slf *ContactSearch) build() *gorm.DB {
   var db = slf.Orm.Model(&Contact{})
   if slf.Keyword != "" {
      db = db.Where("name LIKE ?", "%"+slf.Keyword+"%")
   }
   if slf.Id != 0 {
      db = db.Where("id = ?", slf.Id)
   }
@@ -68,6 +65,38 @@
   if slf.ClientId != 0 {
      db = db.Where("client_id = ?", slf.ClientId)
   }
   if len(slf.SearchMap) > 0 {
      for key, value := range slf.SearchMap {
         switch v := value.(type) {
         case string:
            if key == "name" || key == "position" || key == "phone" || key == "Number" {
               db = db.Where(key+" LIKE ?", "%"+v+"%")
            }
            if key == "member_name" {
               db = db.Joins("Member").Where("Member.username LIKE ?", "%"+v+"%")
            }
            if key == "client_name" {
               //db = db.Joins("inner join clients on clients.id = contacts.client_id").Where("clients.name LIKE ?", "%"+v+"%")
               db = db.Joins("Client").Where("Client.name LIKE ?", "%"+v+"%")
            }
            if key == "is_first" {
               if v == "是" {
                  db = db.Where("is_first = ?", true)
               } else if v == "否" {
                  db = db.Where("is_first = ?", false)
               }
            }
         case int:
            if key == "client_id" {
               db = db.Where("client_id = ?", v)
            }
         }
      }
   }
   return db
@@ -123,7 +152,6 @@
      db = db.Limit(slf.PageSize).Offset((slf.PageNum - 1) * slf.PageSize)
   }
   err := db.Preload("FollowRecord").Preload("Client").Preload("Country").Preload("Province").Preload("City").Preload("Region").Find(&records).Error
   return records, total, err
}
@@ -155,11 +183,6 @@
   return db.Updates(data).Error
}
func (slf *ContactSearch) SetKeyword(keyword string) *ContactSearch {
   slf.Keyword = keyword
   return slf
}
func (slf *ContactSearch) SetPage(page, size int) *ContactSearch {
   slf.PageNum, slf.PageSize = page, size
   return slf
@@ -173,3 +196,8 @@
   slf.Orm = slf.Orm.Where("id in (?)", ids)
   return slf
}
func (slf *ContactSearch) SetSearchMap(data map[string]interface{}) *ContactSearch {
   slf.SearchMap = data
   return slf
}