From 550bd8a218224e6c73201f444387d66a299f438b Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期二, 15 八月 2023 19:56:35 +0800 Subject: [PATCH] 消费排程任务存储任务数据,给前端展示任务列表 --- model/materials.go | 94 +++++++++++++++++++++------------------------- 1 files changed, 43 insertions(+), 51 deletions(-) diff --git a/model/user_menu.go b/model/materials.go similarity index 60% rename from model/user_menu.go rename to model/materials.go index 4e53a25..8fd66e5 100644 --- a/model/user_menu.go +++ b/model/materials.go @@ -1,19 +1,29 @@ package model import ( - "apsClient/pkg/mysqlx" + "apsClient/pkg/sqlitex" "fmt" "gorm.io/gorm" ) type ( - UserMenu struct { - MenuId uint `json:"menuId" gorm:"type:bigint(20);comment:鑿滃崟ID"` - UserId string `json:"userId" gorm:"index;type:varchar(255);comment:鐢ㄦ埛ID"` + + //"materialId": "PS", + //"materialName": "涓ョ墝PS", + //"amount": "10", + //"unit": "g/銕�" + + Materials struct { + gorm.Model `json:"-"` + Id int `json:"id"` + MaterialId string `json:"materialId"` + MaterialName string `json:"materialName"` + Amount string `json:"amount"` + Unit string `json:"unit"` } - UserMenuSearch struct { - UserMenu + MaterialsSearch struct { + Materials Order string PageNum int PageSize int @@ -21,52 +31,34 @@ } ) -func (slf *UserMenu) TableName() string { - return "user_menu" +func (slf *Materials) TableName() string { + return "materials" } -func NewUserMenuSearch(db *gorm.DB) *UserMenuSearch { +func NewMaterialsSearch(db *gorm.DB) *MaterialsSearch { if db == nil { - db = mysqlx.GetDB() + db = sqlitex.GetDB() } - return &UserMenuSearch{Orm: db} + return &MaterialsSearch{Orm: db} } -func (slf *UserMenuSearch) SetOrm(tx *gorm.DB) *UserMenuSearch { +func (slf *MaterialsSearch) SetOrm(tx *gorm.DB) *MaterialsSearch { slf.Orm = tx return slf } -func (slf *UserMenuSearch) SetPage(page, size int) *UserMenuSearch { +func (slf *MaterialsSearch) SetPage(page, size int) *MaterialsSearch { slf.PageNum, slf.PageSize = page, size return slf } -func (slf *UserMenuSearch) SetOrder(order string) *UserMenuSearch { +func (slf *MaterialsSearch) SetOrder(order string) *MaterialsSearch { slf.Order = order return slf } -func (slf *UserMenuSearch) SetUserId(userId string) *UserMenuSearch { - slf.UserId = userId - return slf -} - -func (slf *UserMenuSearch) SetMenuId(menuId uint) *UserMenuSearch { - slf.MenuId = menuId - return slf -} - -func (slf *UserMenuSearch) build() *gorm.DB { - var db = slf.Orm.Model(&UserMenu{}) - - if slf.MenuId > 0 { - db = db.Where("menu_id = ?", slf.MenuId) - } - - if slf.UserId != "" { - db = db.Where("user_id = ?", slf.UserId) - } +func (slf *MaterialsSearch) build() *gorm.DB { + var db = slf.Orm.Model(&Materials{}) if slf.Order != "" { db = db.Order(slf.Order) @@ -76,7 +68,7 @@ } // Create 鍗曟潯鎻掑叆 -func (slf *UserMenuSearch) Create(record *UserMenu) error { +func (slf *MaterialsSearch) Create(record *Materials) error { var db = slf.build() if err := db.Create(record).Error; err != nil { @@ -87,7 +79,7 @@ } // CreateBatch 鎵归噺鎻掑叆 -func (slf *UserMenuSearch) CreateBatch(records []*UserMenu) error { +func (slf *MaterialsSearch) CreateBatch(records []*Materials) error { var db = slf.build() if err := db.Create(&records).Error; err != nil { @@ -97,7 +89,7 @@ return nil } -func (slf *UserMenuSearch) Save(record *UserMenu) error { +func (slf *MaterialsSearch) Save(record *Materials) error { var db = slf.build() if err := db.Save(record).Error; err != nil { @@ -107,7 +99,7 @@ return nil } -func (slf *UserMenuSearch) UpdateByMap(upMap map[string]interface{}) error { +func (slf *MaterialsSearch) UpdateByMap(upMap map[string]interface{}) error { var ( db = slf.build() ) @@ -119,7 +111,7 @@ return nil } -func (slf *UserMenuSearch) UpdateByQuery(query string, args []interface{}, upMap map[string]interface{}) error { +func (slf *MaterialsSearch) UpdateByQuery(query string, args []interface{}, upMap map[string]interface{}) error { var ( db = slf.Orm.Table(slf.TableName()).Where(query, args...) ) @@ -131,19 +123,19 @@ return nil } -func (slf *UserMenuSearch) Delete() error { +func (slf *MaterialsSearch) Delete() error { var db = slf.build() - if err := db.Unscoped().Delete(&UserMenu{}).Error; err != nil { + if err := db.Unscoped().Delete(&Materials{}).Error; err != nil { return err } return nil } -func (slf *UserMenuSearch) First() (*UserMenu, error) { +func (slf *MaterialsSearch) First() (*Materials, error) { var ( - record = new(UserMenu) + record = new(Materials) db = slf.build() ) @@ -154,9 +146,9 @@ return record, nil } -func (slf *UserMenuSearch) Find() ([]*UserMenu, int64, error) { +func (slf *MaterialsSearch) Find() ([]*Materials, int64, error) { var ( - records = make([]*UserMenu, 0) + records = make([]*Materials, 0) total int64 db = slf.build() ) @@ -174,9 +166,9 @@ return records, total, nil } -func (slf *UserMenuSearch) FindNotTotal() ([]*UserMenu, error) { +func (slf *MaterialsSearch) FindNotTotal() ([]*Materials, error) { var ( - records = make([]*UserMenu, 0) + records = make([]*Materials, 0) db = slf.build() ) @@ -191,9 +183,9 @@ } // FindByQuery 鎸囧畾鏉′欢鏌ヨ. -func (slf *UserMenuSearch) FindByQuery(query string, args []interface{}) ([]*UserMenu, int64, error) { +func (slf *MaterialsSearch) FindByQuery(query string, args []interface{}) ([]*Materials, int64, error) { var ( - records = make([]*UserMenu, 0) + records = make([]*Materials, 0) total int64 db = slf.Orm.Table(slf.TableName()).Where(query, args...) ) @@ -212,9 +204,9 @@ } // FindByQueryNotTotal 鎸囧畾鏉′欢鏌ヨ&涓嶆煡璇㈡�绘潯鏁�. -func (slf *UserMenuSearch) FindByQueryNotTotal(query string, args []interface{}) ([]*UserMenu, error) { +func (slf *MaterialsSearch) FindByQueryNotTotal(query string, args []interface{}) ([]*Materials, error) { var ( - records = make([]*UserMenu, 0) + records = make([]*Materials, 0) db = slf.Orm.Table(slf.TableName()).Where(query, args...) ) -- Gitblit v1.8.0