package model import ( "aps_crm/pkg/mysqlx" "gorm.io/gorm" "sync" ) type ( // QuotationStatus 商机阶段 QuotationStatus struct { Id int `json:"id" gorm:"column:id;primary_key;AUTO_INCREMENT"` Name string `json:"name" gorm:"column:name;type:varchar(255);comment:商机阶段名称"` } QuotationStatusSearch struct { QuotationStatus Orm *gorm.DB } ) func (QuotationStatus) TableName() string { return "quotation_status" } func NewQuotationStatusSearch() *QuotationStatusSearch { return &QuotationStatusSearch{ Orm: mysqlx.GetDB(), } } func (slf *QuotationStatusSearch) build() *gorm.DB { var db = slf.Orm.Model(&QuotationStatus{}) if slf.Id != 0 { db = db.Where("id = ?", slf.Id) } if slf.Name != "" { db = db.Where("name = ?", slf.Name) } return db } func (slf *QuotationStatusSearch) Create(record *QuotationStatus) error { var db = slf.build() return db.Create(record).Error } func (slf *QuotationStatusSearch) Delete() error { var db = slf.build() return db.Delete(&QuotationStatus{}).Error } func (slf *QuotationStatusSearch) Update(record *QuotationStatus) error { var db = slf.build() return db.Updates(record).Error } func (slf *QuotationStatusSearch) Find() (*QuotationStatus, error) { var db = slf.build() var record = new(QuotationStatus) err := db.First(record).Error return record, err } func (slf *QuotationStatusSearch) FindAll() ([]*QuotationStatus, error) { var db = slf.build() var records = make([]*QuotationStatus, 0) err := db.Find(&records).Error return records, err } func (slf *QuotationStatusSearch) SetId(id int) *QuotationStatusSearch { slf.Id = id return slf } func (slf *QuotationStatusSearch) SetName(name string) *QuotationStatusSearch { slf.Name = name return slf } func (slf *QuotationStatusSearch) Updates(data map[string]interface{}) error { var db = slf.build() return db.Updates(data).Error } func (slf *QuotationStatusSearch) CreateBatch(records []*QuotationStatus) error { var db = slf.build() return db.Create(records).Error } // InitDefaultData 初始化数据 func (slf *QuotationStatusSearch) InitDefaultData(errCh chan<- error, wg *sync.WaitGroup) { var ( db = slf.Orm.Table(slf.TableName()) total int64 = 0 ) defer wg.Done() if err := db.Count(&total).Error; err != nil { errCh <- err return } if total != 0 { return } records := []*QuotationStatus{ {1, "已创建"}, {2, "已发送"}, {3, "已审批"}, {4, "已拒绝"}, {5, "已接受"}, } err := slf.CreateBatch(records) if err != nil { errCh <- err return } }