From 7bed630958e25d632903e19a57c1735b9dc9d8d8 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期三, 17 四月 2024 13:38:34 +0800
Subject: [PATCH] 增加钉钉机器人报警功能

---
 models/location_product_amount.go |   33 +++++++++++++++++++++++++++++++++
 1 files changed, 33 insertions(+), 0 deletions(-)

diff --git a/models/location_product_amount.go b/models/location_product_amount.go
index 4f1af2b..447b469 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 {
@@ -53,6 +54,7 @@
 		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
@@ -278,3 +289,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