From 530fed8ec225453572d57b15c200ab062c335457 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期三, 01 十一月 2023 19:20:21 +0800 Subject: [PATCH] 公海member_id使用0 --- model/subOrder.go | 63 +++++++++++++++++++++++++++++-- 1 files changed, 58 insertions(+), 5 deletions(-) diff --git a/model/subOrder.go b/model/subOrder.go index 16d2b82..765b112 100644 --- a/model/subOrder.go +++ b/model/subOrder.go @@ -2,6 +2,7 @@ import ( "aps_crm/pkg/mysqlx" + "fmt" "gorm.io/gorm" ) @@ -16,6 +17,7 @@ MasterOrder MasterOrder `json:"masterOrder" gorm:"foreignKey:MasterOrderId"` Number string `json:"number" gorm:"column:number;type:varchar(255);comment:閿�鍞瓙鍗曞彿"` Products []Product `json:"products" gorm:"many2many:subOrder_product;"` + CodeStandID string `json:"codeStandID" gorm:"column:code_stand_id;type:varchar(255);comment:缂栫爜id"` gorm.Model `json:"-"` } @@ -34,9 +36,12 @@ return "sub_order" } -func NewSubOrderSearch() *SubOrderSearch { +func NewSubOrderSearch(db *gorm.DB) *SubOrderSearch { + if db == nil { + db = mysqlx.GetDB() + } return &SubOrderSearch{ - Orm: mysqlx.GetDB(), + Orm: db, } } @@ -45,6 +50,9 @@ if slf.Id != 0 { db = db.Where("id = ?", slf.Id) + } + if slf.Number != "" { + db = db.Where("number = ?", slf.Number) } if len(slf.SearchMap) > 0 { @@ -66,9 +74,16 @@ if key == "member_name" { db = db.Joins("Member").Where("User.username LIKE ?", "%"+v+"%") } - case int: - if key == "master_order_id" { - db = db.Where("master_order_id = ?", v) + case int, float64: + if key == "client_id" || key == "master_order_id" { + db = db.Where(key+" = ?", v) + } + if key == "member_id" { + db = db.Where("sub_order.member_id = ?", v) + } + case []int: + if key == "member_ids" { + db = db.Where("sub_order.member_id in ?", v) } } } @@ -113,6 +128,30 @@ return records, total, err } +func (slf *SubOrderSearch) Count() (int64, error) { + var db = slf.build() + var total int64 + err := db.Count(&total).Error + return total, err +} + +func (slf *SubOrderSearch) MaxAutoIncr() (int, error) { + type Result struct { + Max int + } + + var ( + result Result + db = slf.build() + ) + + err := db.Select("MAX(id) as max").Scan(&result).Error + if err != nil { + return result.Max, fmt.Errorf("max err: %v", err) + } + return result.Max, nil +} + func (slf *SubOrderSearch) SetId(id int) *SubOrderSearch { slf.Id = id return slf @@ -132,3 +171,17 @@ slf.SearchMap = data return slf } +func (slf *SubOrderSearch) SetIds(ids []int) *SubOrderSearch { + slf.Orm = slf.Orm.Where("id in (?)", ids) + return slf +} + +func (slf *SubOrderSearch) SetNumber(number string) *SubOrderSearch { + slf.Number = number + return slf +} + +func (slf *SubOrderSearch) UpdateByMap(data map[string]interface{}) error { + var db = slf.build() + return db.Updates(data).Error +} -- Gitblit v1.8.0