package models import ( "fmt" "gorm.io/gorm" "silkserver/constvar" "silkserver/pkg/mysqlx" ) type ( // Worker 人力资源 Worker struct { BaseModelString DeletedAt gorm.DeletedAt `gorm:"index"` Name string `gorm:"index;type:varchar(191);not null;comment:人员姓名" json:"name"` PhoneNum string `gorm:"type:varchar(191);comment:手机号" json:"phoneNum"` ShopID string `gorm:"type:varchar(191);comment:所属车间ID" json:"shopId"` ShopNumber string `gorm:"type:varchar(191);comment:所属车间编号" json:"shopNumber"` ShopName string `gorm:"type:varchar(191);comment:车间名称,仅查询用" json:"shopName"` Status constvar.WorkerStatus `gorm:"index;type:int(11);comment:人员状态" json:"status"` WorkTypeId uint `gorm:"type:int(11);comment:工种id" json:"workTypeId"` WorkTypeCode constvar.JobType `gorm:"type:varchar(255);comment:工种代码" json:"workTypeCode"` WorkType string `gorm:"type:varchar(255);comment:工种名称" json:"workType"` GroupNumber int `gorm:"type:int(11);comment:组别" json:"groupNumber"` EmploymentTime string `gorm:"type:varchar(255);comment:入职时间" json:"employmentTime"` AddPeople string `gorm:"type:varchar(255);comment:添加人" json:"addPeople"` } WorkerSearch struct { Worker Keyword string Preload bool Order string PageNum int PageSize int Orm *gorm.DB Fields string Ids []string } ) func (slf Worker) TableName() string { return "worker" } func NewWorkerSearch() *WorkerSearch { return &WorkerSearch{Orm: mysqlx.GetDB()} } func (slf *WorkerSearch) SetOrm(tx *gorm.DB) *WorkerSearch { slf.Orm = tx return slf } func (slf *WorkerSearch) SetPage(page, size int) *WorkerSearch { slf.PageNum, slf.PageSize = page, size return slf } func (slf *WorkerSearch) SetOrder(order string) *WorkerSearch { slf.Order = order return slf } func (slf *WorkerSearch) SetID(id string) *WorkerSearch { slf.ID = id return slf } func (slf *WorkerSearch) SetName(name string) *WorkerSearch { slf.Name = name return slf } func (slf *WorkerSearch) SetKeyword(keyword string) *WorkerSearch { slf.Keyword = keyword return slf } func (slf *WorkerSearch) SetShopID(shopID string) *WorkerSearch { slf.ShopID = shopID return slf } func (slf *WorkerSearch) SetPreload(preload bool) *WorkerSearch { slf.Preload = preload return slf } func (slf *WorkerSearch) SetStatus(status constvar.WorkerStatus) *WorkerSearch { slf.Status = status return slf } func (slf *WorkerSearch) SetFields(fields string) *WorkerSearch { slf.Fields = fields return slf } func (slf *WorkerSearch) SetIds(ids []string) *WorkerSearch { slf.Ids = ids return slf } func (slf *WorkerSearch) build() *gorm.DB { var db = slf.Orm.Table(slf.TableName()) if slf.ID != "" { db = db.Where("id = ?", slf.ID) } if slf.Name != "" { db = db.Where("name = ?", slf.Name) } if slf.ShopID != "" { db = db.Where("shop_id = ?", slf.ShopID) } if slf.Keyword != "" { db = db.Where("worker.name LIKE ? or worker.phone_num LIKE ? ", "%"+slf.Keyword+"%", "%"+slf.Keyword+"%") } if slf.Status > 0 { db = db.Where("status = ?", slf.Status) } if slf.Order != "" { db = db.Order(slf.Order) } if slf.Preload { db = db.Model(&Worker{}).Preload("WorkerSkill") } if slf.Fields != "" { db = db.Select(slf.Fields) } if len(slf.Ids) > 0 { db = db.Where("id in (?)", slf.Ids) } return db } // Create 单条插入 func (slf *WorkerSearch) Create(record *Worker) error { var db = slf.build() if err := db.Create(record).Error; err != nil { return fmt.Errorf("create err: %v, record: %+v", err, record) } return nil } // CreateBatch 批量插入 func (slf *WorkerSearch) CreateBatch(records []*Worker) error { var db = slf.build() if err := db.Create(&records).Error; err != nil { return fmt.Errorf("create batch err: %v, records: %+v", err, records) } return nil } func (slf *WorkerSearch) Save(record *Worker) error { var db = slf.build() if err := db.Omit("CreatedAt").Save(record).Error; err != nil { return fmt.Errorf("save err: %v, record: %+v", err, record) } return nil } func (slf *WorkerSearch) UpdateByMap(upMap map[string]interface{}) error { var ( db = slf.build() ) if err := db.Updates(upMap).Error; err != nil { return fmt.Errorf("update by map err: %v, upMap: %+v", err, upMap) } return nil } func (slf *WorkerSearch) UpdateByQuery(query string, args []interface{}, upMap map[string]interface{}) error { var ( db = slf.Orm.Table(slf.TableName()).Where(query, args...) ) if err := db.Updates(upMap).Error; err != nil { return fmt.Errorf("update by query err: %v, query: %s, args: %+v, upMap: %+v", err, query, args, upMap) } return nil } func (slf *WorkerSearch) Delete() error { var db = slf.build() if err := db.Delete(&Worker{}).Error; err != nil { return err } return nil } func (slf *WorkerSearch) First() (*Worker, error) { var ( record = new(Worker) db = slf.build() ) if err := db.First(record).Error; err != nil { return record, err } return record, nil } func (slf *WorkerSearch) Find() ([]*Worker, int64, error) { var ( records = make([]*Worker, 0) total int64 db = slf.build() ) if err := db.Count(&total).Error; err != nil { return records, total, fmt.Errorf("find count err: %v", err) } if slf.PageNum*slf.PageSize > 0 { db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize) } if slf.Keyword != "" { db = db.Distinct("worker.*") } if err := db.Find(&records).Error; err != nil { return records, total, fmt.Errorf("find records err: %v", err) } return records, total, nil } func (slf *WorkerSearch) FindNotTotal() ([]*Worker, error) { var ( records = make([]*Worker, 0) db = slf.build() ) if slf.PageNum*slf.PageSize > 0 { db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize) } if err := db.Find(&records).Error; err != nil { return records, fmt.Errorf("find records err: %v", err) } return records, nil } // FindByQuery 指定条件查询. func (slf *WorkerSearch) FindByQuery(query string, args []interface{}) ([]*Worker, int64, error) { var ( records = make([]*Worker, 0) total int64 db = slf.Orm.Table(slf.TableName()).Where(query, args...) ) if err := db.Count(&total).Error; err != nil { return records, total, fmt.Errorf("find by query count err: %v", err) } if slf.PageNum*slf.PageSize > 0 { db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize) } if err := db.Find(&records).Error; err != nil { return records, total, fmt.Errorf("find by query records err: %v, query: %s, args: %+v", err, query, args) } return records, total, nil } // FindByQueryNotTotal 指定条件查询&不查询总条数. func (slf *WorkerSearch) FindByQueryNotTotal(query string, args []interface{}) ([]*Worker, error) { var ( records = make([]*Worker, 0) db = slf.Orm.Table(slf.TableName()).Where(query, args...) ) if slf.PageNum*slf.PageSize > 0 { db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize) } if err := db.Find(&records).Error; err != nil { return records, fmt.Errorf("find by query records err: %v, query: %s, args: %+v", err, query, args) } return records, nil } func (slf *WorkerSearch) Count() (int64, error) { var ( total int64 db = slf.build() ) if err := db.Count(&total).Error; err != nil { return total, fmt.Errorf("find count err: %v", err) } return total, nil } func (slf *WorkerSearch) MaxAutoIncr() (int64, error) { type Result struct { Max int64 } var ( result Result db = slf.build() ) err := db.Select("MAX(auto_incr) as max").Scan(&result).Error if err != nil { return result.Max, fmt.Errorf("max err: %v", err) } return result.Max, nil } func (slf *WorkerSearch) FirstOrCreate(record *Worker) error { var db = slf.build() if err := db.FirstOrCreate(record).Error; err != nil { return fmt.Errorf("first or create err: %v, record: %+v", err, record) } return nil }