package model import ( "aps_crm/constvar" "aps_crm/pkg/mysqlx" "errors" "fmt" "gorm.io/gorm" ) type ( // WechatOrderStatus 微信订单状态 WechatOrderStatus struct { Id int `json:"id" gorm:"column:id;primary_key;AUTO_INCREMENT"` Name string `json:"name" gorm:"column:name"` } // WechatOrderStatusSearch 微信订单状态搜索条件 WechatOrderStatusSearch struct { WechatOrderStatus Orm *gorm.DB QueryClass constvar.WechatOrderStatusQueryClass KeywordType constvar.WechatOrderStatusKeywordType Keyword string PageNum int PageSize int } ) func (WechatOrderStatus) TableName() string { return "wechat_order_status" } func NewWechatOrderStatusSearch() *WechatOrderStatusSearch { return &WechatOrderStatusSearch{ Orm: mysqlx.GetDB(), } } func (slf *WechatOrderStatusSearch) build() *gorm.DB { var db = slf.Orm.Model(&WechatOrderStatus{}) if slf.Id != 0 { db = db.Where("id = ?", slf.Id) } return db } func (slf *WechatOrderStatusSearch) Create(record *WechatOrderStatus) error { var db = slf.build() return db.Create(record).Error } func (slf *WechatOrderStatusSearch) CreateBatch(records []*WechatOrderStatus) error { var db = slf.build() return db.Create(records).Error } func (slf *WechatOrderStatusSearch) Delete() error { var db = slf.build() return db.Delete(&WechatOrderStatus{}).Error } func (slf *WechatOrderStatusSearch) Update(record *WechatOrderStatus) error { var db = slf.build() return db.Updates(record).Error } func (slf *WechatOrderStatusSearch) FindAll() ([]*WechatOrderStatus, error) { var db = slf.build() var record = make([]*WechatOrderStatus, 0) err := db.Find(&record).Error return record, err } func (slf *WechatOrderStatusSearch) SetId(id int) *WechatOrderStatusSearch { slf.Id = id return slf } func (slf *WechatOrderStatusSearch) SetOrm(tx *gorm.DB) *WechatOrderStatusSearch { slf.Orm = tx return slf } func (slf *WechatOrderStatusSearch) First() (*WechatOrderStatus, error) { var db = slf.build() var record = new(WechatOrderStatus) err := db.First(record).Error return record, err } func (slf *WechatOrderStatusSearch) Updates(values interface{}) error { var db = slf.build() return db.Updates(values).Error } func (slf *WechatOrderStatusSearch) Save(record *WechatOrderStatus) error { if record.Id == 0 { return errors.New("id为空") } var db = slf.build() if err := db.Save(record).Error; err != nil { return fmt.Errorf("save err: %v, record: %+v", err, record) } return nil } func (slf *WechatOrderStatusSearch) Find() ([]*WechatOrderStatus, int64, error) { var db = slf.build() var records = make([]*WechatOrderStatus, 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.Find(&records).Error return records, total, err } // InitDefaultData 初始化数据 func (slf *WechatOrderStatusSearch) InitDefaultData() error { var ( db = slf.Orm.Table(slf.TableName()) total int64 = 0 ) if err := db.Count(&total).Error; err != nil { return err } if total != 0 { return nil } records := []*WechatOrderStatus{} return slf.CreateBatch(records) }