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