package model import ( "aps_crm/pkg/mysqlx" "gorm.io/gorm" ) type ( // Currency 商机阶段 Currency struct { Id int `json:"id" gorm:"column:id;primary_key;AUTO_INCREMENT"` Name string `json:"name" gorm:"column:name;type:varchar(255);comment:商机阶段名称"` } CurrencySearch struct { Currency Orm *gorm.DB } ) func (Currency) TableName() string { return "currency" } func NewCurrencySearch() *CurrencySearch { return &CurrencySearch{ Orm: mysqlx.GetDB(), } } func (slf *CurrencySearch) build() *gorm.DB { var db = slf.Orm.Model(&Currency{}) if slf.Id != 0 { db = db.Where("id = ?", slf.Id) } if slf.Name != "" { db = db.Where("name = ?", slf.Name) } return db } func (slf *CurrencySearch) Create(record *Currency) error { var db = slf.build() return db.Create(record).Error } func (slf *CurrencySearch) Delete() error { var db = slf.build() return db.Delete(&Currency{}).Error } func (slf *CurrencySearch) Update(record *Currency) error { var db = slf.build() return db.Updates(record).Error } func (slf *CurrencySearch) Find() (*Currency, error) { var db = slf.build() var record = new(Currency) err := db.First(record).Error return record, err } func (slf *CurrencySearch) FindAll() ([]*Currency, error) { var db = slf.build() var records = make([]*Currency, 0) err := db.Find(&records).Error return records, err } func (slf *CurrencySearch) SetId(id int) *CurrencySearch { slf.Id = id return slf } func (slf *CurrencySearch) SetName(name string) *CurrencySearch { slf.Name = name return slf } func (slf *CurrencySearch) Updates(data map[string]interface{}) error { var db = slf.build() return db.Updates(data).Error }