package model import ( "aps_crm/pkg/mysqlx" "gorm.io/gorm" ) type ( Department struct { Id int `json:"id" gorm:"column:id;primaryKey;autoIncrement;not null"` Name string `json:"name" gorm:"column:name;type:varchar(255);comment:部门名称"` ParentId int `json:"-" gorm:"column:parent_id;type:int(11);comment:上级部门ID"` } DepartmentSearch struct { Department Orm *gorm.DB } ) func (Department) TableName() string { return "department" } func NewDepartmentSearch() *DepartmentSearch { return &DepartmentSearch{ Orm: mysqlx.GetDB(), } } func (slf *DepartmentSearch) build() *gorm.DB { var db = slf.Orm.Model(&Department{}) if slf.Id != 0 { db = db.Where("id = ?", slf.Id) } if slf.Name != "" { db = db.Where("name = ?", slf.Name) } if slf.ParentId != 0 { db = db.Where("parent_id = ?", slf.ParentId) } return db } func (slf *DepartmentSearch) Create(record *Department) error { var db = slf.build() return db.Create(record).Error } func (slf *DepartmentSearch) Delete() error { var db = slf.build() return db.Delete(&Department{}).Error } func (slf *DepartmentSearch) Update(record *Department) error { var db = slf.build() return db.Updates(record).Error } func (slf *DepartmentSearch) Find() ([]*Department, error) { var db = slf.build() var result []*Department err := db.Find(&result).Error return result, err } func (slf *DepartmentSearch) SetId(id int) *DepartmentSearch { slf.Id = id return slf } func (slf *DepartmentSearch) Updates(data map[string]interface{}) error { var db = slf.build() return db.Updates(data).Error }