fix

wangpengfei

2023-08-09 c229f8922a4142a8ae2ac161bb6c121b9c9de672

 model/contact.go

@@ -1,175 +1,191 @@
package model



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:备注"`

      Address

      gorm.Model `json:"-"`

   }



   ContactSearch struct {

      Contact



      Orm      *gorm.DB

      Keyword  string

      OrderBy  string

      PageNum  int

      PageSize int

   }



   ContactDetail struct {

      Contact

      Client       Client

      FollowRecord []FollowRecord `gorm:"foreignKey:ContactId"`

   }

)



func (Contact) TableName() string {

   return "contacts"

}



func NewContactSearch(db *gorm.DB) *ContactSearch {

   if db == nil {

      db = mysqlx.GetDB()

   }

   return &ContactSearch{

      Orm: db,

   }

}



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)

   }

   if slf.Name != "" {

      db = db.Where("name = ?", slf.Name)

   }



   if slf.ClientId != 0 {

      db = db.Where("client_id = ?", slf.ClientId)

   }



   return db

}



func (slf *ContactSearch) Create(record *Contact) error {

   var db = slf.build()

   return db.Create(record).Error

}



func (slf *ContactSearch) Update(record *Contact) error {

   var db = slf.build()

   m := map[string]interface{}{

      "name":        record.Name,

      "number":      record.Number,

      "client_id":   record.ClientId,

      "position":    record.Position,

      "phone":       record.Phone,

      "member_id":   record.MemberId,

      "is_first":    record.IsFirst,

      "wechat":      record.Wechat,

      "birthday":    record.Birthday,

      "email":       record.Email,

      "desc":        record.Desc,

      "country_id":  record.CountryId,

      "province_id": record.ProvinceId,

      "city_id":     record.CityId,

      "region_id":   record.RegionId,

   }

   return db.Updates(m).Error

}



func (slf *ContactSearch) Delete() error {

   var db = slf.build()

   return db.Delete(&Contact{}).Error

}



func (slf *ContactSearch) Find() (*Contact, error) {

   var db = slf.build()

   var record = new(Contact)

   err := db.First(record).Error

   return record, err

}



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, total, err

}



func (slf *ContactSearch) SetId(id int) *ContactSearch {

   slf.Id = id

   return slf

}



func (slf *ContactSearch) SetName(name string) *ContactSearch {

   slf.Name = name

   return slf

}



func (slf *ContactSearch) First() (*Contact, error) {

   var db = slf.build()

   var record = new(Contact)

   err := db.First(record).Error

   return record, err

}



func (slf *ContactSearch) SetClientId(clientId int) *ContactSearch {

   slf.ClientId = clientId

   return slf

}



func (slf *ContactSearch) UpdateByMap(data map[string]interface{}) error {

   var db = slf.build()

   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

}



func (slf *ContactSearch) SetOrder(order string) *ContactSearch {

   slf.OrderBy = order

   return slf

}

func (slf *ContactSearch) SetIds(ids []int) *ContactSearch {

   slf.Orm = slf.Orm.Where("id in (?)", ids)

   return slf

}

package model

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:备注"`
      Address
      gorm.Model `json:"-"`
   }

   ContactSearch struct {
      Contact

      Orm       *gorm.DB
      SearchMap map[string]interface{}
      OrderBy   string
      PageNum   int
      PageSize  int
   }

   ContactDetail struct {
      Contact
      Client       Client
      FollowRecord []FollowRecord `gorm:"foreignKey:ContactId"`
   }
)

func (Contact) TableName() string {
   return "contacts"
}

func NewContactSearch(db *gorm.DB) *ContactSearch {
   if db == nil {
      db = mysqlx.GetDB()
   }
   return &ContactSearch{
      Orm: db,
   }
}

func (slf *ContactSearch) build() *gorm.DB {
   var db = slf.Orm.Model(&Contact{})
   if slf.Id != 0 {
      db = db.Where("id = ?", slf.Id)
   }
   if slf.Name != "" {
      db = db.Where("name = ?", slf.Name)
   }

   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+"%")
            }
         case int:
         }
      }
   }

   return db
}

func (slf *ContactSearch) Create(record *Contact) error {
   var db = slf.build()
   return db.Create(record).Error
}

func (slf *ContactSearch) Update(record *Contact) error {
   var db = slf.build()
   m := map[string]interface{}{
      "name":        record.Name,
      "number":      record.Number,
      "client_id":   record.ClientId,
      "position":    record.Position,
      "phone":       record.Phone,
      "member_id":   record.MemberId,
      "is_first":    record.IsFirst,
      "wechat":      record.Wechat,
      "birthday":    record.Birthday,
      "email":       record.Email,
      "desc":        record.Desc,
      "country_id":  record.CountryId,
      "province_id": record.ProvinceId,
      "city_id":     record.CityId,
      "region_id":   record.RegionId,
   }
   return db.Updates(m).Error
}

func (slf *ContactSearch) Delete() error {
   var db = slf.build()
   return db.Delete(&Contact{}).Error
}

func (slf *ContactSearch) Find() (*Contact, error) {
   var db = slf.build()
   var record = new(Contact)
   err := db.First(record).Error
   return record, err
}

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, total, err
}

func (slf *ContactSearch) SetId(id int) *ContactSearch {
   slf.Id = id
   return slf
}

func (slf *ContactSearch) SetName(name string) *ContactSearch {
   slf.Name = name
   return slf
}

func (slf *ContactSearch) First() (*Contact, error) {
   var db = slf.build()
   var record = new(Contact)
   err := db.First(record).Error
   return record, err
}

func (slf *ContactSearch) SetClientId(clientId int) *ContactSearch {
   slf.ClientId = clientId
   return slf
}

func (slf *ContactSearch) UpdateByMap(data map[string]interface{}) error {
   var db = slf.build()
   return db.Updates(data).Error
}

func (slf *ContactSearch) SetPage(page, size int) *ContactSearch {
   slf.PageNum, slf.PageSize = page, size
   return slf
}

func (slf *ContactSearch) SetOrder(order string) *ContactSearch {
   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
}