From ea02a2a9a8219d44b103f0a31cf9fc81ff8495e1 Mon Sep 17 00:00:00 2001
From: liujiandao <274878379@qq.com>
Date: 星期六, 23 九月 2023 16:57:48 +0800
Subject: [PATCH] 产品列表添加类型名称

---
 models/location.go |   98 ++++++++++++++++++++++++++++++++++++++++---------
 1 files changed, 80 insertions(+), 18 deletions(-)

diff --git a/models/location.go b/models/location.go
index 0146ca0..394e0bb 100644
--- a/models/location.go
+++ b/models/location.go
@@ -11,16 +11,21 @@
 	// Location 浣嶇疆
 	Location 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:浣嶇疆鍚嶇О"`     //浣嶇疆鍚嶇О
-		ParentId          int                   `json:"parentId" gorm:"type:int;not null"`                             //涓婄骇id
-		CompanyId         int                   `json:"companyId" gorm:"type:int;not null"`                            //鍏徃id
-		Company           Company               `json:"company" gorm:"foreignKey:CompanyId"`                           //鍏徃
-		Type              constvar.LocationType `json:"type" gorm:"type:tinyint;not null;comment:浣嶇疆绫诲瀷"`                //浣嶇疆绫诲瀷
-		CountFrequency    int                   `json:"countFrequency" gorm:"type:tinyint;not null;comment:鐩樼偣棰戠巼锛堝ぉ锛�"`   //鐩樼偣棰戠巼锛堝ぉ锛�
-		IsScrapLocation   bool                  `json:"isScrapLocation" gorm:"type:tinyint;not null;comment:鏄惁鎶ュ簾浣嶇疆"`   //鏄惁鎶ュ簾浣嶇疆
-		IsReturnLocation  bool                  `json:"isReturnLocation" gorm:"type:tinyint;not null;comment:鏄惁閫�璐т綅缃�"`  //鏄惁閫�璐т綅缃�
-		ReplenishLocation bool                  `json:"replenishLocation" gorm:"type:tinyint;not null;comment:鏄惁琛ュ厖浣嶇疆"` //鏄惁琛ュ厖浣嶇疆
+		Id        int    `json:"id" gorm:"column:id;primary_key;AUTO_INCREMENT"`
+		Name      string `json:"name" gorm:"index;type:varchar(255);not null;comment:浣嶇疆鍚嶇О"` //浣嶇疆鍚嶇О
+		ParentId  string `json:"parentId" gorm:"type:varchar(255)"`                         //涓婄骇id
+		CompanyId int    `json:"companyId" gorm:"type:int"`                                 //鍏徃id
+		//Company              Company                       `json:"company" gorm:"foreignKey:CompanyId"`                       //鍏徃
+		Type                 constvar.LocationType         `json:"type" gorm:"type:int(11);comment:浣嶇疆绫诲瀷"`                 //浣嶇疆绫诲瀷
+		CountFrequency       int                           `json:"countFrequency" gorm:"type:tinyint;comment:鐩樼偣棰戠巼锛堝ぉ锛�"`    //鐩樼偣棰戠巼锛堝ぉ锛�
+		IsScrapLocation      bool                          `json:"isScrapLocation" gorm:"type:tinyint;comment:鏄惁鎶ュ簾浣嶇疆"`    //鏄惁鎶ュ簾浣嶇疆
+		IsReturnLocation     bool                          `json:"isReturnLocation" gorm:"type:tinyint;comment:鏄惁閫�璐т綅缃�"`   //鏄惁閫�璐т綅缃�
+		ReplenishLocation    bool                          `json:"replenishLocation" gorm:"type:tinyint;comment:鏄惁琛ュ厖浣嶇疆"`  //鏄惁琛ュ厖浣嶇疆
+		ForceRemovalStrategy constvar.ForceRemovalStrategy `json:"forceRemovalStrategy" gorm:"type:tinyint;comment:涓嬫灦绛栫暐"` //涓嬫灦绛栫暐
+		Notes                string                        `json:"notes" gorm:"type:varchar(255);comment:澶栭儴澶囨敞"`           //澶栭儴澶囨敞
+		RecentlyCount        string                        `json:"recentlyCount" gorm:"type:varchar(255);comment:鏈�杩戠洏鐐�"`   //鏈�杩戠洏鐐�
+		NextCount            string                        `json:"nextCount" gorm:"type:varchar(255);comment:涓嬫鐩樼偣"`       //涓嬫鐩樼偣
+		JointName            string                        `json:"jointName" gorm:"-"`                                    //鎷兼帴鍚嶇О
 	}
 
 	LocationSearch struct {
@@ -31,11 +36,12 @@
 		Keyword  string
 		Orm      *gorm.DB
 		Preload  bool
+		Parents  []string
 	}
 )
 
 func (slf *Location) TableName() string {
-	return "location"
+	return "wms_location"
 }
 
 func NewLocationSearch() *LocationSearch {
@@ -57,8 +63,12 @@
 	return slf
 }
 
-func (slf *LocationSearch) SetID(id uint) *LocationSearch {
-	slf.ID = id
+func (slf *LocationSearch) SetID(ID int) *LocationSearch {
+	slf.Id = ID
+	return slf
+}
+func (slf *LocationSearch) SetParents(ids []string) *LocationSearch {
+	slf.Parents = ids
 	return slf
 }
 
@@ -77,11 +87,26 @@
 	return slf
 }
 
-func (slf *LocationSearch) build() *gorm.DB {
-	var db = slf.Orm.Model(&Location{})
+func (slf *LocationSearch) SetType(_type int) *LocationSearch {
+	slf.Type = constvar.LocationType(_type)
+	return slf
+}
 
-	if slf.ID != 0 {
-		db = db.Where("id = ?", slf.ID)
+func (slf *LocationSearch) SetParentId(parentId string) *LocationSearch {
+	slf.ParentId = parentId
+	return slf
+}
+
+func (slf *LocationSearch) SetCompanyId(companyId int) *LocationSearch {
+	slf.CompanyId = companyId
+	return slf
+}
+
+func (slf *LocationSearch) build() *gorm.DB {
+	var db = slf.Orm.Table(slf.TableName())
+
+	if slf.Id != 0 {
+		db = db.Where("id = ?", slf.Id)
 	}
 
 	if slf.Order != "" {
@@ -96,6 +121,20 @@
 		db = db.Where("name = ?", slf.Name)
 	}
 
+	if slf.Type != 0 {
+		db = db.Where("type=?", slf.Type)
+	}
+
+	if slf.ParentId != "" {
+		db = db.Where("parent_id=?", slf.ParentId)
+	}
+	if slf.CompanyId != 0 {
+		db = db.Where("company_id=?", slf.CompanyId)
+	}
+	if len(slf.Parents) != 0 {
+		db = db.Where("parent_id in (?)", slf.Parents)
+	}
+
 	return db
 }
 
@@ -108,6 +147,16 @@
 	}
 
 	return nil
+}
+
+func (slf *LocationSearch) CreateReturnId(record *Location) (int, error) {
+	var db = slf.build()
+
+	if err := db.Create(record).Error; err != nil {
+		return 0, err
+	}
+
+	return record.Id, nil
 }
 
 // CreateBatch 鎵归噺鎻掑叆
@@ -157,7 +206,7 @@
 
 func (slf *LocationSearch) Delete() error {
 	var db = slf.build()
-	return db.Delete(&Location{}).Error
+	return db.Unscoped().Delete(&Location{}).Error
 }
 
 func (slf *LocationSearch) First() (*Location, error) {
@@ -246,3 +295,16 @@
 
 	return records, nil
 }
+
+func (slf *LocationSearch) FindAll() ([]*Location, error) {
+	var (
+		records = make([]*Location, 0)
+		db      = slf.build()
+	)
+	err := db.Find(&records)
+	if err != nil {
+		fmt.Println(err)
+		return records, fmt.Errorf("func FindAll err: %v", err)
+	}
+	return records, nil
+}

--
Gitblit v1.8.0