zhangqian
2023-11-01 530fed8ec225453572d57b15c200ab062c335457
model/contact.go
@@ -3,23 +3,25 @@
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"`
      Member      User        `json:"member" gorm:"foreignKey:MemberId"`
      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:备注"`
      CodeStandID string      `json:"codeStandID" gorm:"column:code_stand_id;type:varchar(255);comment:编码id"`
      Address
      gorm.Model `json:"-"`
   }
@@ -27,11 +29,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 +58,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 +67,45 @@
   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, uint, int64, float64:
            if key == "id" || key == "client_type_id" || key == "client_status_id" || key == "member_id" {
               db = db.Where(key+" = ?", v)
            }
            if key == "client_id" {
               db = db.Where("client_id = ? and is_first = true", v)
            }
         case []int:
            if key == "member_ids" {
               db = db.Where("contacts.member_id in ?", v)
            }
         }
      }
   }
   return db
@@ -112,14 +150,19 @@
   return record, err
}
func (slf *ContactSearch) FindAll() ([]*ContactDetail, error) {
func (slf *ContactSearch) FindAll() ([]*ContactDetail, int64, error) {
   var db = slf.build()
   var records = make([]*ContactDetail, 0)
   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("FollowRecord").Preload("Client").Preload("Country").Preload("Province").Preload("City").Preload("Region").Find(&records).Error
   return records, err
   err := db.Preload("Member").Preload("FollowRecord").Preload("Client").Preload("Country").Preload("Province").Preload("City").Preload("Region").Order("id desc").Find(&records).Error
   return records, total, err
}
func (slf *ContactSearch) SetId(id int) *ContactSearch {
@@ -149,11 +192,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
@@ -163,3 +201,12 @@
   slf.OrderBy = order
   return slf
}
func (slf *ContactSearch) SetIds(ids []int) *ContactSearch {
   slf.Orm = slf.Orm.Where("id in (?)", ids)
   return slf
}
func (slf *ContactSearch) SetSearchMap(data map[string]interface{}) *ContactSearch {
   slf.SearchMap = data
   return slf
}