From cab6eea89a0d82710369604ecb51f1cdb122e433 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期四, 19 十月 2023 17:26:34 +0800 Subject: [PATCH] fix --- model/process_model.go | 74 ++++++++++++++++++++++++------------- 1 files changed, 48 insertions(+), 26 deletions(-) diff --git a/model/process_model.go b/model/process_model.go index 9ff8e4f..253860f 100644 --- a/model/process_model.go +++ b/model/process_model.go @@ -5,30 +5,34 @@ "apsClient/pkg/sqlitex" "encoding/json" "fmt" - "gorm.io/gorm" + "github.com/jinzhu/gorm" ) type ( // ProcessModel 宸ヨ壓鍙傛暟 ProcessModel struct { - gorm.Model `json:"-"` - Number string `gorm:"index;column:number;type:varchar(255);not null;default '';comment:宸ヨ壓妯″瀷缂栧彿" json:"number"` //宸ヨ壓妯″瀷缂栧彿 - Product string `gorm:"column:product;type:varchar(255);not null;default '';comment:浜у搧鍚嶇О" json:"product"` //浜у搧鍚嶇О - Procedure string `gorm:"column:procedure;type:varchar(255);not null;default '';comment:宸ュ簭" json:"procedure"` //宸ュ簭 - Params string `gorm:"type:text;comment:宸ヨ壓鍙傛暟閿�煎json涓�"` - ParamsMap map[string]interface{} `json:"paramsMap" gorm:"-"` - DeviceId string `json:"deviceId" gorm:"-"` //鐢ㄤ簬杩囨护鑾峰彇nsq娑堟伅 - IsNew bool `json:"isNew" gorm:"column:is_new;comment:鏄惁鏈�鏂扮殑"` //鏄惁鏈�鏂扮殑 - IsUpdate bool `json:"isUpdate" gorm:"-"` //鍓嶇鐢� + gorm.Model + Number string `gorm:"index;column:number;type:varchar(255);not null;default ''" json:"number"` //宸ヨ壓妯″瀷缂栧彿 + Product string `gorm:"column:product;type:varchar(255);not null;default ''" json:"product"` //浜у搧鍚嶇О + Procedure string `gorm:"column:procedure;type:varchar(255);not null;default ''" json:"procedure"` //宸ュ簭 + Params string + ParamsMap map[string]interface{} `json:"paramsMap" gorm:"-"` + DeviceId string `json:"deviceId" gorm:"-"` //鐢ㄤ簬杩囨护鑾峰彇nsq娑堟伅 + IsNew bool `json:"-" gorm:"column:is_new"` //鏄惁鏈�鏂扮殑 + IsUpdate bool `json:"isUpdate" gorm:"-"` //鍓嶇鐢� + NewParamsMap map[string]interface{} `json:"newParamsMap" gorm:"-"` + NewNumber string `json:"newNumber" gorm:"-"` } ProcessModelSearch struct { ProcessModel - Order string - PageNum int - PageSize int - Orm *gorm.DB - Procedures []string + Order string + Limit int + Offset int + Orm *gorm.DB + Procedures []string + CurrentNumber string + Numbers []string } ) @@ -54,8 +58,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 } @@ -71,6 +75,16 @@ func (slf *ProcessModelSearch) SetNumber(number string) *ProcessModelSearch { slf.Number = number + return slf +} + +func (slf *ProcessModelSearch) SetNumbers(numbers []string) *ProcessModelSearch { + slf.Numbers = numbers + return slf +} + +func (slf *ProcessModelSearch) SetCurrentNumber(number string) *ProcessModelSearch { + slf.CurrentNumber = number return slf } @@ -110,7 +124,7 @@ } if len(slf.Procedures) != 0 { - db = db.Where("`procedure` in ?", slf.Procedures) + db = db.Where("`procedure` IN (?)", slf.Procedures) } if len(slf.Number) != 0 { @@ -123,6 +137,14 @@ if slf.IsNew { db = db.Where("is_new = ?", 1) + } + + if len(slf.CurrentNumber) != 0 { + db = db.Where("number != ?", slf.CurrentNumber) + } + + if len(slf.Numbers) != 0 { + db = db.Where("number in ?", slf.Numbers) } return db @@ -206,8 +228,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) @@ -222,8 +244,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) @@ -243,8 +265,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) @@ -260,8 +282,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) -- Gitblit v1.8.0