From e5df488268e29b272932e6cc1d2b1e7034590ba0 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期三, 12 六月 2024 11:40:45 +0800
Subject: [PATCH] weedfs包放到项目里

---
 models/location_product_amount.go |   40 ++++++++++++++++++++++++++++++++++++++--
 1 files changed, 38 insertions(+), 2 deletions(-)

diff --git a/models/location_product_amount.go b/models/location_product_amount.go
index 4f1af2b..efa1205 100644
--- a/models/location_product_amount.go
+++ b/models/location_product_amount.go
@@ -36,6 +36,7 @@
 		LocationIds []int
 		ProductIds  []string
 		Ids         []int
+		Query       string
 	}
 
 	LocationProductAmountWithOperation struct {
@@ -48,11 +49,12 @@
 		Amount                  decimal.Decimal            `json:"amount" gorm:"column:amount"`
 		Unit                    string                     `json:"unit" gorm:"column:unit"`
 		CreateDate              string                     `json:"createDate" gorm:"column:create_date"`
-		AdjustAmount            decimal.Decimal            `json:"adjustAmount" gorm:"column:adjust_amount"`
-		DifferenceAmount        decimal.Decimal            `json:"differenceAmount" gorm:"-"`
+		AdjustAmount            decimal.Decimal            `json:"adjustAmount" gorm:"column:adjust_amount"` //宸��
+		DifferenceAmount        decimal.Decimal            `json:"differenceAmount" gorm:"-"`                //璁℃暟鏁伴噺
 		OperationId             int                        `json:"operationId" gorm:"column:operation_id"`
 		Status                  constvar.OperationStatus   `json:"status" gorm:"status"`
 		BaseOperationType       constvar.BaseOperationType `json:"baseOperationType" gorm:"base_operation_type"`
+		Weight                  decimal.Decimal            `gorm:"type:decimal(20,3);comment:閲嶉噺" json:"weight"` //閲嶉噺
 	}
 )
 
@@ -119,6 +121,11 @@
 	return slf
 }
 
+func (slf *LocationProductAmountSearch) SetQuery(query string) *LocationProductAmountSearch {
+	slf.Query = query
+	return slf
+}
+
 func (slf *LocationProductAmountSearch) build() *gorm.DB {
 	var db = slf.Orm.Model(&LocationProductAmount{})
 
@@ -161,6 +168,10 @@
 
 	if len(slf.Ids) > 0 {
 		db = db.Where("id in (?)", slf.Ids)
+	}
+
+	if slf.Query != "" {
+		db = db.Where(slf.Query)
 	}
 
 	return db
@@ -242,6 +253,9 @@
 		db      = slf.build()
 	)
 
+	if slf.PageNum*slf.PageSize > 0 {
+		db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize)
+	}
 	if err := db.Find(&records).Error; err != nil {
 		return records, fmt.Errorf("find records err: %v", err)
 	}
@@ -278,3 +292,25 @@
 
 	return nil
 }
+
+func (slf *LocationProductAmountSearch) GroupCount(field string) ([]*GroupCount, error) {
+	var (
+		db     = slf.build()
+		result = make([]*GroupCount, 0)
+	)
+	if err := db.Select("count(*) as total, " + field + " as class").Group(field).Scan(&result).Error; err != nil {
+		return nil, fmt.Errorf("select group err: %v", err)
+	}
+	return result, nil
+}
+
+func (slf *LocationProductAmountSearch) GroupSum(groupField string, sumField string) ([]*GroupSum, error) {
+	var (
+		db     = slf.build()
+		result = make([]*GroupSum, 0)
+	)
+	if err := db.Select("sum(" + sumField + ") as sum, " + groupField + " as class").Group(groupField).Scan(&result).Error; err != nil {
+		return nil, fmt.Errorf("select group err: %v", err)
+	}
+	return result, nil
+}

--
Gitblit v1.8.0