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