From ccc4c924d81c3f8201e7a6c783a9a7148b21670d Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期二, 12 九月 2023 17:43:42 +0800 Subject: [PATCH] 公司增删改查,业务类型增删改查 --- models/warehouse.go | 18 ++++++++++++++++-- 1 files changed, 16 insertions(+), 2 deletions(-) diff --git a/models/warehouse.go b/models/warehouse.go index 42f6408..49a4e9b 100644 --- a/models/warehouse.go +++ b/models/warehouse.go @@ -8,9 +8,10 @@ ) type ( - // Warehouse 閮ㄩ棬淇℃伅 + // Warehouse 浠撳簱 Warehouse struct { WmsModel + Id int `json:"id" gorm:"column:id;primary_key;AUTO_INCREMENT"` Name string `json:"name" gorm:"index;type:varchar(255);not null;comment:浠撳簱鍚嶇О"` //浠撳簱鍚嶇О Active bool `json:"active" gorm:"type:tinyint(1);not null;comment:鏄惁婵�娲�"` //鏄惁鍚敤锛屼紶true灏辫 Code string `json:"code" binding:"required,min=1,max=5" gorm:"index;type:varchar(255);not null;comment:浠撳簱缂栫爜"` //浠撳簱缂栫爜 @@ -19,6 +20,8 @@ ResupplyWhIdsStr string `json:"-" gorm:"column:resupply_wh_ids;type:varchar(255);not null;comment:琛ョ粰鏉ユ簮浠撳簱ID"` //琛ョ粰鏉ユ簮浠撳簱ID ResupplyWhIds []string `json:"resupplyWhIds" gorm:"-"` //琛ョ粰鏉ユ簮浠撳簱ID ResupplyWh []*Warehouse `json:"resupplyWh" gorm:"-"` //琛ョ粰鏉ユ簮浠撳簱 + CompanyId int `json:"companyId" gorm:"type:int;not null;comment:鍏徃id"` + Company Company `json:"company" gorm:"foreignKey:CompanyId"` } WarehouseSearch struct { @@ -28,6 +31,7 @@ PageSize int Keyword string Orm *gorm.DB + Preload bool } ) @@ -79,8 +83,13 @@ return slf } +func (slf *WarehouseSearch) SetPreload(preload bool) *WarehouseSearch { + slf.Preload = preload + return slf +} + func (slf *WarehouseSearch) build() *gorm.DB { - var db = slf.Orm.Table(slf.TableName()) + var db = slf.Orm.Model(&Warehouse{}) if slf.ID != 0 { db = db.Where("id = ?", slf.ID) @@ -93,10 +102,15 @@ if slf.Keyword != "" { db = db.Where("name like ?", fmt.Sprintf("%%%v%%", slf.Keyword)) } + if slf.Name != "" { db = db.Where("name = ?", slf.Name) } + if slf.Preload { + db = db.Preload("Company") + } + return db } -- Gitblit v1.8.0