zhangqian
2023-11-01 530fed8ec225453572d57b15c200ab062c335457
model/salesLeads.go
@@ -3,29 +3,41 @@
import (
   "aps_crm/constvar"
   "aps_crm/pkg/mysqlx"
   "fmt"
   "gorm.io/gorm"
)
type (
   SalesLeads struct {
      Id              int                  `json:"id" gorm:"column:id;primary_key;AUTO_INCREMENT"`
      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:销售线索编号"`
      ContactName     string               `json:"contact_name" gorm:"column:contact_name;type:varchar(255);comment:联系人姓名"`
      ContactPhone    string               `json:"contact_phone" gorm:"column:contact_phone;type:varchar(255);comment:联系人电话"`
      ContactPosition string               `json:"contact_position" gorm:"column:contact_position;type:varchar(255);comment:联系人职位"`
      SalesSourcesId  int                  `json:"sales_sources_id" gorm:"column:sales_sources_id;type:int(11);comment:商机来源ID"`
      SalesSources    SalesSources         `json:"sales_sources" gorm:"foreignKey:SalesSourcesId"`
      MemberId        int                  `json:"member_id" gorm:"column:member_id;type:int(11);comment:销售负责人ID"`
      Member          User                 `json:"member" gorm:"foreignKey:MemberId"`
      SalesStatus     constvar.SalesStatus `json:"sales_status" gorm:"column:sales_status;type:int(11);comment:销售状态"`
      Desc            string               `json:"desc" gorm:"column:desc;type:varchar(255);comment:备注"`
      Reason          string               `json:"reason" gorm:"column:reason;type:text;comment:失败原因"`
      FollowRecord    []FollowRecord       `gorm:"foreignKey:SalesLeadsId"`
      DetailAddress   string               `json:"detail_address" gorm:"column:detail_address;type:varchar(255);comment:详细地址"`
      CodeStandID     string               `json:"codeStandID" gorm:"column:code_stand_id;type:varchar(255);comment:编码id"`
      Address
      gorm.Model `json:"-"`
   }
   SalesLeadsSearch struct {
      SalesLeads
      Orm *gorm.DB
      Orm       *gorm.DB
      SearchMap map[string]interface{}
      OrderBy   string
      PageNum   int
      PageSize  int
   }
)
@@ -33,9 +45,13 @@
   return "sales_leads"
}
func NewSalesLeadsSearch() *SalesLeadsSearch {
func NewSalesLeadsSearch(db *gorm.DB) *SalesLeadsSearch {
   if db == nil {
      db = mysqlx.GetDB()
   }
   return &SalesLeadsSearch{
      Orm: mysqlx.GetDB(),
      Orm: db,
   }
}
@@ -47,29 +63,41 @@
   if slf.Name != "" {
      db = db.Where("name = ?", slf.Name)
   }
   if slf.Number != "" {
      db = db.Where("number = ?", slf.Number)
   }
   if slf.ContactName != "" {
      db = db.Where("contact_name = ?", slf.ContactName)
   }
   if len(slf.SearchMap) > 0 {
      for key, value := range slf.SearchMap {
         switch v := value.(type) {
         case string:
            if key == "number" || key == "contact_name" || key == "contact_phone" || key == "name" {
               db = db.Where(key+" LIKE ?", "%"+v+"%")
            }
   if slf.ContactPhone != "" {
      db = db.Where("contact_phone = ?", slf.ContactPhone)
   }
            if key == "city" {
               db = db.Joins("City").Where("City.name LIKE ?", "%"+v+"%")
            }
   if slf.ContactPosition != "" {
      db = db.Where("contact_position = ?", slf.ContactPosition)
   }
            if key == "province" {
               db = db.Joins("Province").Where("Province.name LIKE ?", "%"+v+"%")
            }
   if slf.SalesSourcesId != 0 {
      db = db.Where("sales_sources_id = ?", slf.SalesSourcesId)
   }
            if key == "sales_sources" {
               db = db.Joins("SalesSources").Where("SalesSources.name LIKE ?", "%"+v+"%")
            }
   if slf.MemberId != 0 {
      db = db.Where("member_id = ?", slf.MemberId)
            if key == "member_name" {
               db = db.Joins("Member").Where("Member.username LIKE ?", "%"+v+"%")
            }
         case int:
         case []int:
            if key == "member_ids" {
               db = db.Where("sales_leads.member_id in ?", v)
            }
         }
      }
   }
   return db
@@ -87,6 +115,7 @@
func (slf *SalesLeadsSearch) Delete() error {
   var db = slf.build()
   db.Update("is_deleted", 1)
   return db.Delete(&SalesLeads{}).Error
}
@@ -102,10 +131,43 @@
   return
}
func (slf *SalesLeadsSearch) FindAll() (result []*SalesLeads, err error) {
func (slf *SalesLeadsSearch) FindAll() ([]*SalesLeads, int64, error) {
   var db = slf.build()
   err = db.Preload("FollowRecord").Find(&result).Error
   return
   var records = make([]*SalesLeads, 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("SalesSources").Preload("Member").Preload("Province").Preload("City").Preload("FollowRecord").Order("id desc").Find(&records).Error
   return records, total, err
}
func (slf *SalesLeadsSearch) Count() (int64, error) {
   var db = slf.build()
   var total int64
   err := db.Count(&total).Error
   return total, err
}
func (slf *SalesLeadsSearch) 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 *SalesLeadsSearch) SetId(id int) *SalesLeadsSearch {
@@ -123,3 +185,32 @@
   err = db.First(&result).Error
   return
}
func (slf *SalesLeadsSearch) SetPage(page, size int) *SalesLeadsSearch {
   slf.PageNum, slf.PageSize = page, size
   return slf
}
func (slf *SalesLeadsSearch) SetOrder(order string) *SalesLeadsSearch {
   slf.OrderBy = order
   return slf
}
func (slf *SalesLeadsSearch) UpdateMap(data map[string]interface{}) error {
   var db = slf.build()
   return db.Updates(data).Error
}
func (slf *SalesLeadsSearch) SetIds(ids []int) *SalesLeadsSearch {
   slf.Orm = slf.Orm.Where("id in (?)", ids)
   return slf
}
func (slf *SalesLeadsSearch) SetNumber(number string) *SalesLeadsSearch {
   slf.Number = number
   return slf
}
func (slf *SalesLeadsSearch) SetSearchMap(data map[string]interface{}) *SalesLeadsSearch {
   slf.SearchMap = data
   return slf
}