From dad1ae79fe78e770d88f41713b4c68fad562d9ef Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期一, 09 十月 2023 21:36:34 +0800 Subject: [PATCH] 当前使用的工艺编号置顶 --- service/process_model.go | 4 ++-- model/process_model.go | 24 ++++++++++++------------ api/v1/process_model.go | 25 ++++++++++++++++--------- 3 files changed, 30 insertions(+), 23 deletions(-) diff --git a/api/v1/process_model.go b/api/v1/process_model.go index 61dad3b..8a03909 100644 --- a/api/v1/process_model.go +++ b/api/v1/process_model.go @@ -24,20 +24,15 @@ if !ok { return } - oldProcessModel, _ := service.GetProcessModelByNumber(params.Number) - - if params.Page == 1 && oldProcessModel.IsNew { //灏戞煡涓�涓� - params.PageSize-- - } - - list, total, err := service.GetProcessModelList(params.Page, params.PageSize, params.Number) + offset, limit, push := GetOffsetAndLimit(oldProcessModel, params.Page, params.PageSize) + list, total, err := service.GetProcessModelList(offset, limit, params.Number) if err != nil { ctx.Fail(ecode.DBErr) return } - if params.Page == 1 && oldProcessModel.IsNew { //绮樺湪鍓嶉潰 + if push { //绮樺湪鍓嶉潰 list = append([]*model.ProcessModel{oldProcessModel}, list...) } for _, processModel := range list { @@ -49,6 +44,18 @@ processModel.Number, processModel.ParamsMap = oldProcessModel.Number, oldProcessModel.ParamsMap } } - ctx.ResultList(list, total) } + +func GetOffsetAndLimit(current *model.ProcessModel, page, pageSize int) (offset, limit int, push bool) { + offset = (page - 1) * pageSize + if current.IsNew { + offset-- + } + push = current.IsNew && page == 1 + limit = pageSize + if push { + limit-- + } + return offset, limit, push +} diff --git a/model/process_model.go b/model/process_model.go index 56ab7be..a08e694 100644 --- a/model/process_model.go +++ b/model/process_model.go @@ -27,8 +27,8 @@ ProcessModelSearch struct { ProcessModel Order string - PageNum int - PageSize int + Limit int + Offset int Orm *gorm.DB Procedures []string CurrentNumber string @@ -57,8 +57,8 @@ return slf } -func (slf *ProcessModelSearch) SetPage(page, size int) *ProcessModelSearch { - slf.PageNum, slf.PageSize = page, size +func (slf *ProcessModelSearch) SetPage(offset, limit int) *ProcessModelSearch { + slf.Offset, slf.Limit = offset, limit return slf } @@ -218,8 +218,8 @@ 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.Offset > 0 || slf.Limit > 0 { + db = db.Offset(slf.Offset).Limit(slf.Limit) } if err := db.Find(&records).Error; err != nil { return records, total, fmt.Errorf("find records err: %v", err) @@ -234,8 +234,8 @@ db = slf.build() ) - if slf.PageNum*slf.PageSize > 0 { - db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize) + if slf.Offset > 0 || slf.Limit > 0 { + db = db.Offset(slf.Offset).Limit(slf.Limit) } if err := db.Find(&records).Error; err != nil { return records, fmt.Errorf("find records err: %v", err) @@ -255,8 +255,8 @@ 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 slf.Offset > 0 || slf.Limit > 0 { + db = db.Offset(slf.Offset).Limit(slf.Limit) } 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) @@ -272,8 +272,8 @@ 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 slf.Offset > 0 || slf.Limit > 0 { + db = db.Offset(slf.Offset).Limit(slf.Limit) } 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) diff --git a/service/process_model.go b/service/process_model.go index adf53ba..bb42646 100644 --- a/service/process_model.go +++ b/service/process_model.go @@ -5,14 +5,14 @@ "apsClient/model" ) -func GetProcessModelList(page, pageSize int, currentNumber string) (list []*model.ProcessModel, total int64, err error) { +func GetProcessModelList(offset, limit int, currentNumber string) (list []*model.ProcessModel, total int64, err error) { device, err := model.NewDeviceSearch().SetDeviceId(conf.Conf.System.DeviceId).First() if err != nil { return } list, total, err = model.NewProcessModelSearch(). SetIsNew(true).SetOrder("id desc").SetCurrentNumber(currentNumber). - SetPage(page, pageSize).SetProcedures(device.ProceduresArr).Find() + SetPage(offset, limit).SetProcedures(device.ProceduresArr).Find() return } -- Gitblit v1.8.0