From 4497935a55e286ee922b99e6e478b7714a35d0b2 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期五, 03 十一月 2023 16:28:33 +0800
Subject: [PATCH] fix
---
model/process_model.go | 101 +++++++++++++++++++++++++++++++++++++++-----------
1 files changed, 78 insertions(+), 23 deletions(-)
diff --git a/model/process_model.go b/model/process_model.go
index 1828814..1614b26 100644
--- a/model/process_model.go
+++ b/model/process_model.go
@@ -1,34 +1,52 @@
package model
import (
+ "apsClient/pkg/logx"
"apsClient/pkg/sqlitex"
+ "encoding/json"
"fmt"
- "gorm.io/gorm"
+ "github.com/jinzhu/gorm"
)
type (
- // ProcessModel 宸ヨ壓娴佺▼鍙傛暟
+ // 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娑堟伅
+ 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
+ Order string
+ Limit int
+ Offset int
+ Orm *gorm.DB
+ Procedures []string
+ CurrentNumber string
+ Numbers []string
}
)
func (slf *ProcessModel) TableName() string {
return "process_model"
+}
+
+func (slf *ProcessModel) AfterFind(db *gorm.DB) error {
+ err := json.Unmarshal([]byte(slf.Params), &slf.ParamsMap)
+ if err != nil {
+ logx.Errorf("process model json.Unmarshal:%v", err)
+ return err
+ }
+ return nil
}
func NewProcessModelSearch() *ProcessModelSearch {
@@ -40,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
}
@@ -60,12 +78,33 @@
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
+}
+
func (slf *ProcessModelSearch) SetProduct(product string) *ProcessModelSearch {
slf.Product = product
return slf
}
+
func (slf *ProcessModelSearch) SetProcedure(procedure string) *ProcessModelSearch {
slf.Procedure = procedure
+ return slf
+}
+
+func (slf *ProcessModelSearch) SetProcedures(procedures []string) *ProcessModelSearch {
+ slf.Procedures = procedures
+ return slf
+}
+
+func (slf *ProcessModelSearch) SetIsNew(isNew bool) *ProcessModelSearch {
+ slf.IsNew = isNew
return slf
}
@@ -84,12 +123,28 @@
db = db.Where("`procedure` = ?", slf.Procedure)
}
+ if len(slf.Procedures) != 0 {
+ db = db.Where("`procedure` IN (?)", slf.Procedures)
+ }
+
if len(slf.Number) != 0 {
db = db.Where("number = ?", slf.Number)
}
if slf.Order != "" {
db = db.Order(slf.Order)
+ }
+
+ 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
@@ -173,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)
@@ -189,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)
@@ -210,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)
@@ -227,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