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