package models import ( "fmt" "gorm.io/gorm" "silkserver/pkg/mysqlx" ) type ( // ShiftTime 班次时间 ShiftTime struct { BaseModelInt ShiftID string `gorm:"index;type:varchar(191);not null;comment:班次ID" json:"shiftId"` StartTime string `gorm:"comment:班次开始时间" json:"startTime"` EndTime string `gorm:"comment:班次结束时间" json:"endTime"` } ShiftTimeSearch struct { ShiftTime Order string PageNum int PageSize int Orm *gorm.DB } ) func (slf ShiftTime) TableName() string { return "shift_time" } func NewShiftTimeSearch() *ShiftTimeSearch { return &ShiftTimeSearch{Orm: mysqlx.GetDB()} } func (slf *ShiftTimeSearch) SetOrm(tx *gorm.DB) *ShiftTimeSearch { slf.Orm = tx return slf } func (slf *ShiftTimeSearch) SetPage(page, size int) *ShiftTimeSearch { slf.PageNum, slf.PageSize = page, size return slf } func (slf *ShiftTimeSearch) SetOrder(order string) *ShiftTimeSearch { slf.Order = order return slf } func (slf *ShiftTimeSearch) SetShiftID(shiftID string) *ShiftTimeSearch { slf.ShiftID = shiftID return slf } func (slf *ShiftTimeSearch) SetID(id uint) *ShiftTimeSearch { slf.ID = id return slf } func (slf *ShiftTimeSearch) build() *gorm.DB { var db = slf.Orm.Table(slf.TableName()) if slf.ID > 0 { db = db.Where("id = ?", slf.ID) } if slf.ShiftID != "" { db = db.Where("shift_id = ?", slf.ShiftID) } if slf.Order != "" { db = db.Order(slf.Order) } return db } // Create 单条插入 func (slf *ShiftTimeSearch) Create(record *ShiftTime) 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 *ShiftTimeSearch) CreateBatch(records []*ShiftTime) 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 *ShiftTimeSearch) Save(record *ShiftTime) 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 *ShiftTimeSearch) 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 *ShiftTimeSearch) 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 *ShiftTimeSearch) Delete() error { var db = slf.build() if err := db.Unscoped().Delete(&ShiftTime{}).Error; err != nil { return err } return nil } func (slf *ShiftTimeSearch) First() (*ShiftTime, error) { var ( record = new(ShiftTime) db = slf.build() ) if err := db.First(record).Error; err != nil { return record, err } return record, nil } func (slf *ShiftTimeSearch) Find() ([]*ShiftTime, int64, error) { var ( records = make([]*ShiftTime, 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 *ShiftTimeSearch) FindNotTotal() ([]*ShiftTime, error) { var ( records = make([]*ShiftTime, 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 *ShiftTimeSearch) FindByQuery(query string, args []interface{}) ([]*ShiftTime, int64, error) { var ( records = make([]*ShiftTime, 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 *ShiftTimeSearch) FindByQueryNotTotal(query string, args []interface{}) ([]*ShiftTime, error) { var ( records = make([]*ShiftTime, 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 }