package models import ( "fmt" "gorm.io/gorm" "silkserver/pkg/mysqlx" ) type ( // WorkerPosition 机台管理 WorkerPosition struct { gorm.Model StartDate string `gorm:"type:varchar(255);not null;default:'';comment:开始日期"` //开始日期 EndDate string `gorm:"type:varchar(255);not null;default:'';comment:结束日期"` //结束日期 Workshop string `gorm:"type:varchar(255);not null;default:'';comment:车间编号" json:"workshop"` //车间编号 WorkshopName string `gorm:"type:varchar(255);not null;default:'';comment:车间名称" json:"workshopName"` //车间名称 WorkshopGroup int `gorm:"type:int(11);not null;default:0;comment:车组" json:"workshopGroup"` //车组 StartWorkerPosition int `json:"startWorkerPosition" gorm:"type:int(11);comment:开始车号"` //开始车号 EndWorkerPosition int `json:"endWorkerPosition" gorm:"type:int(11);comment:结束车号"` //结束车号 WorkerId string `gorm:"type:varchar(255);not null;default:'';comment:工人ID" json:"workerId"` //工人ID Worker Worker `gorm:"foreignkey:WorkerId" json:"worker"` //工人 Creator string `gorm:"type:varchar(255);not null;default:'';comment:添加人" json:"creator"` } WorkerPositionSearch struct { WorkerPosition Order string PageNum int PageSize int Orm *gorm.DB Preload bool Keyword string OverlappingDateStart string OverlappingDateEnd string WorkerDate string } ) func (slf *WorkerPosition) TableName() string { return "silk_worker_position" } func NewWorkerPositionSearch() *WorkerPositionSearch { return &WorkerPositionSearch{Orm: mysqlx.GetDB()} } func (slf *WorkerPositionSearch) SetOrm(tx *gorm.DB) *WorkerPositionSearch { slf.Orm = tx return slf } func (slf *WorkerPositionSearch) SetPage(page, size int) *WorkerPositionSearch { slf.PageNum, slf.PageSize = page, size return slf } func (slf *WorkerPositionSearch) SetOrder(order string) *WorkerPositionSearch { slf.Order = order return slf } func (slf *WorkerPositionSearch) SetPreload() *WorkerPositionSearch { slf.Preload = true return slf } func (slf *WorkerPositionSearch) SetID(id uint) *WorkerPositionSearch { slf.ID = id return slf } func (slf *WorkerPositionSearch) SetWorkerID(id string) *WorkerPositionSearch { slf.WorkerId = id return slf } func (slf *WorkerPositionSearch) SetKeyword(keyword string) *WorkerPositionSearch { slf.Keyword = keyword return slf } func (slf *WorkerPositionSearch) SetWorkshop(workshop string) *WorkerPositionSearch { slf.Workshop = workshop return slf } func (slf *WorkerPositionSearch) SetWorkshopGroup(group int) *WorkerPositionSearch { slf.WorkshopGroup = group return slf } func (slf *WorkerPositionSearch) SetOverlappingDate(startDate string, endDate string) *WorkerPositionSearch { slf.OverlappingDateStart = startDate slf.OverlappingDateEnd = endDate return slf } func (slf *WorkerPositionSearch) build() *gorm.DB { var db = slf.Orm.Table(slf.TableName()) if slf.ID != 0 { db = db.Where("id = ?", slf.ID) } if slf.Order != "" { db = db.Order(slf.Order) } if slf.Keyword != "" { kw := "%" + slf.Keyword + "%" db = db.Joins("Worker").Where("silk_worker_position.worker_id LIKE ? or Worker.name LIKE ? or silk_worker_position.workshop like ? or silk_worker_position.workshop_name like ?", kw, kw, kw, kw) } if slf.Workshop != "" { db = db.Where("workshop = ?", slf.Workshop) } if slf.WorkshopGroup > 0 { db = db.Where("workshop_group = ?", slf.WorkshopGroup) } if slf.WorkerId != "" { db = db.Where("worker_id = ?", slf.WorkerId) } if slf.Preload { db = db.Model(&WorkerPosition{}).Preload("Worker") } if slf.OverlappingDateStart != "" && slf.OverlappingDateEnd != "" { db = db.Where("start_date <= ? and end_date >= ?", slf.OverlappingDateEnd, slf.OverlappingDateStart) } return db } // Create 单条插入 func (slf *WorkerPositionSearch) Create(record *WorkerPosition) 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 *WorkerPositionSearch) CreateBatch(records []*WorkerPosition) 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 *WorkerPositionSearch) Save(record *WorkerPosition) 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 *WorkerPositionSearch) Update(record *WorkerPosition) error { var db = slf.build() if err := db.Updates(record).Error; err != nil { return fmt.Errorf("save err: %v, record: %+v", err, record) } return nil } func (slf *WorkerPositionSearch) 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 *WorkerPositionSearch) 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 *WorkerPositionSearch) Delete() error { var db = slf.build() if err := db.Unscoped().Delete(&WorkerPosition{}).Error; err != nil { return err } return nil } func (slf *WorkerPositionSearch) First() (*WorkerPosition, error) { var ( record = new(WorkerPosition) db = slf.build() ) if err := db.First(record).Error; err != nil { return record, err } return record, nil } func (slf *WorkerPositionSearch) Find() ([]*WorkerPosition, int64, error) { var ( records = make([]*WorkerPosition, 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 err := db.Find(&records).Error; err != nil { return records, total, fmt.Errorf("find records err: %v", err) } return records, total, nil } func (slf *WorkerPositionSearch) FindAll() ([]*WorkerPosition, error) { var ( records = make([]*WorkerPosition, 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 *WorkerPositionSearch) FindByQuery(query string, args []interface{}) ([]*WorkerPosition, int64, error) { var ( records = make([]*WorkerPosition, 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 *WorkerPositionSearch) FindByQueryNotTotal(query string, args []interface{}) ([]*WorkerPosition, error) { var ( records = make([]*WorkerPosition, 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 } // InitDefaultData 初始化数据 func (slf *WorkerPositionSearch) InitDefaultData() error { return nil }