package model import ( "aps_crm/pkg/mysqlx" "gorm.io/gorm" "sync" ) type ( // Status 状态 Status struct { Id int `json:"id" gorm:"column:id;primary_key;AUTO_INCREMENT"` Name string `json:"name" gorm:"column:name;type:varchar(255);comment:状态名称"` } // StatusSearch 状态搜索条件 StatusSearch struct { Status Orm *gorm.DB } ) func (Status) TableName() string { return "status" } func NewStatusSearch() *StatusSearch { return &StatusSearch{ Orm: mysqlx.GetDB(), } } func (slf *StatusSearch) build() *gorm.DB { var db = slf.Orm.Model(&Status{}) if slf.Id != 0 { db = db.Where("id = ?", slf.Id) } if slf.Name != "" { db = db.Where("name = ?", slf.Name) } return db } func (slf *StatusSearch) Create(record *Status) error { var db = slf.build() return db.Create(record).Error } func (slf *StatusSearch) Delete() error { var db = slf.build() return db.Delete(&Status{}).Error } func (slf *StatusSearch) Update(record *Status) error { var db = slf.build() return db.Updates(record).Error } func (slf *StatusSearch) Find() (*Status, error) { var db = slf.build() var record = &Status{} err := db.First(record).Error return record, err } func (slf *StatusSearch) FindAll() ([]*Status, error) { var db = slf.build() var records = make([]*Status, 0) err := db.Find(&records).Error return records, err } func (slf *StatusSearch) SetId(id int) *StatusSearch { slf.Id = id return slf } func (slf *StatusSearch) SetName(name string) *StatusSearch { slf.Name = name return slf } func (slf *StatusSearch) Updates(data map[string]interface{}) error { var db = slf.build() return db.Updates(data).Error } func (slf *StatusSearch) CreateBatch(records []*Status) error { var db = slf.build() return db.Create(records).Error } // InitDefaultData 初始化数据 func (slf *StatusSearch) 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 := []*Status{ {1, "已成交"}, {2, "未成交"}, } err := slf.CreateBatch(records) if err != nil { errCh <- err return } }