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
|
}
|
