From 747e890bdae1e5f894a4959d84853e74017ff91b Mon Sep 17 00:00:00 2001 From: wangpengfei <274878379@qq.com> Date: 星期二, 08 八月 2023 14:28:04 +0800 Subject: [PATCH] fix --- model/salesLeads.go | 347 +++++++++++++++++++++++++++++---------------------------- 1 files changed, 174 insertions(+), 173 deletions(-) diff --git a/model/salesLeads.go b/model/salesLeads.go index 80c14f7..6e9f65e 100644 --- a/model/salesLeads.go +++ b/model/salesLeads.go @@ -1,173 +1,174 @@ -package model - -import ( - "aps_crm/constvar" - "aps_crm/pkg/mysqlx" - "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:鍏徃鍚嶇О"` - 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"` - MemberId int `json:"member_id" gorm:"column:member_id;type:int(11);comment:閿�鍞礋璐d汉ID"` - 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:澶囨敞"` - FollowRecord []FollowRecord `gorm:"foreignKey:SalesLeadsId"` - Address - gorm.Model `json:"-"` - } - - SalesLeadsSearch struct { - SalesLeads - - Orm *gorm.DB - Keyword string - OrderBy string - PageNum int - PageSize int - } -) - -func (SalesLeads) TableName() string { - return "sales_leads" -} - -func NewSalesLeadsSearch(db *gorm.DB) *SalesLeadsSearch { - if db == nil { - db = mysqlx.GetDB() - } - - return &SalesLeadsSearch{ - Orm: db, - } -} - -func (slf *SalesLeadsSearch) build() *gorm.DB { - var db = slf.Orm.Model(&SalesLeads{}) - if slf.Keyword != "" { - db = db.Where("name LIKE ?", "%"+slf.Keyword+"%") - } - 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.Number != "" { - db = db.Where("number = ?", slf.Number) - } - - if slf.ContactName != "" { - db = db.Where("contact_name = ?", slf.ContactName) - } - - if slf.ContactPhone != "" { - db = db.Where("contact_phone = ?", slf.ContactPhone) - } - - if slf.ContactPosition != "" { - db = db.Where("contact_position = ?", slf.ContactPosition) - } - - if slf.SalesSourcesId != 0 { - db = db.Where("sales_sources_id = ?", slf.SalesSourcesId) - } - - if slf.MemberId != 0 { - db = db.Where("member_id = ?", slf.MemberId) - } - - return db -} - -func (slf *SalesLeadsSearch) Create(record *SalesLeads) error { - var db = slf.build() - return db.Create(record).Error -} - -func (slf *SalesLeadsSearch) Update(record *SalesLeads) error { - var db = slf.build() - return db.Updates(record).Error -} - -func (slf *SalesLeadsSearch) Delete() error { - var db = slf.build() - return db.Delete(&SalesLeads{}).Error -} - -func (slf *SalesLeadsSearch) Find() (result []SalesLeads, err error) { - var db = slf.build() - err = db.Find(&result).Error - return -} - -func (slf *SalesLeadsSearch) FindOne() (result SalesLeads, err error) { - var db = slf.build() - err = db.First(&result).Error - return -} - -func (slf *SalesLeadsSearch) FindAll() ([]*SalesLeads, int64, error) { - var db = slf.build() - 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("FollowRecord").Find(&records).Error - return records, total, err -} - -func (slf *SalesLeadsSearch) SetId(id int) *SalesLeadsSearch { - slf.Id = id - return slf -} - -func (slf *SalesLeadsSearch) SetName(name string) *SalesLeadsSearch { - slf.Name = name - return slf -} - -func (slf *SalesLeadsSearch) First() (result SalesLeads, err error) { - var db = slf.build() - err = db.First(&result).Error - return -} - -func (slf *SalesLeadsSearch) SetKeyword(keyword string) *SalesLeadsSearch { - slf.Keyword = keyword - return slf -} - -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 -} +package model + +import ( + "aps_crm/constvar" + "aps_crm/pkg/mysqlx" + "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:鍏徃鍚嶇О"` + 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"` + MemberId int `json:"member_id" gorm:"column:member_id;type:int(11);comment:閿�鍞礋璐d汉ID"` + 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"` + Address + gorm.Model `json:"-"` + } + + SalesLeadsSearch struct { + SalesLeads + + Orm *gorm.DB + Keyword string + OrderBy string + PageNum int + PageSize int + } +) + +func (SalesLeads) TableName() string { + return "sales_leads" +} + +func NewSalesLeadsSearch(db *gorm.DB) *SalesLeadsSearch { + if db == nil { + db = mysqlx.GetDB() + } + + return &SalesLeadsSearch{ + Orm: db, + } +} + +func (slf *SalesLeadsSearch) build() *gorm.DB { + var db = slf.Orm.Model(&SalesLeads{}) + if slf.Keyword != "" { + db = db.Where("name LIKE ?", "%"+slf.Keyword+"%") + } + 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.Number != "" { + db = db.Where("number = ?", slf.Number) + } + + if slf.ContactName != "" { + db = db.Where("contact_name = ?", slf.ContactName) + } + + if slf.ContactPhone != "" { + db = db.Where("contact_phone = ?", slf.ContactPhone) + } + + if slf.ContactPosition != "" { + db = db.Where("contact_position = ?", slf.ContactPosition) + } + + if slf.SalesSourcesId != 0 { + db = db.Where("sales_sources_id = ?", slf.SalesSourcesId) + } + + if slf.MemberId != 0 { + db = db.Where("member_id = ?", slf.MemberId) + } + + return db +} + +func (slf *SalesLeadsSearch) Create(record *SalesLeads) error { + var db = slf.build() + return db.Create(record).Error +} + +func (slf *SalesLeadsSearch) Update(record *SalesLeads) error { + var db = slf.build() + return db.Updates(record).Error +} + +func (slf *SalesLeadsSearch) Delete() error { + var db = slf.build() + return db.Delete(&SalesLeads{}).Error +} + +func (slf *SalesLeadsSearch) Find() (result []SalesLeads, err error) { + var db = slf.build() + err = db.Find(&result).Error + return +} + +func (slf *SalesLeadsSearch) FindOne() (result SalesLeads, err error) { + var db = slf.build() + err = db.First(&result).Error + return +} + +func (slf *SalesLeadsSearch) FindAll() ([]*SalesLeads, int64, error) { + var db = slf.build() + 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("FollowRecord").Find(&records).Error + return records, total, err +} + +func (slf *SalesLeadsSearch) SetId(id int) *SalesLeadsSearch { + slf.Id = id + return slf +} + +func (slf *SalesLeadsSearch) SetName(name string) *SalesLeadsSearch { + slf.Name = name + return slf +} + +func (slf *SalesLeadsSearch) First() (result SalesLeads, err error) { + var db = slf.build() + err = db.First(&result).Error + return +} + +func (slf *SalesLeadsSearch) SetKeyword(keyword string) *SalesLeadsSearch { + slf.Keyword = keyword + return slf +} + +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 +} -- Gitblit v1.8.0