From b3a318b7f707ca49fa9127881bbe709654eaa761 Mon Sep 17 00:00:00 2001
From: liujiandao <274878379@qq.com>
Date: 星期三, 27 九月 2023 17:27:51 +0800
Subject: [PATCH] 报表功能开发

---
 models/material.go |   19 ++++++++++++++-----
 1 files changed, 14 insertions(+), 5 deletions(-)

diff --git a/models/material.go b/models/material.go
index df61270..f22f5b9 100644
--- a/models/material.go
+++ b/models/material.go
@@ -77,11 +77,12 @@
 		//MaterialTypes   []constvar.MaterialType
 		Keyword string
 		//SetTemplateType constvar.SetTemplateType
-		Order    string
-		PageNum  int
-		PageSize int
-		Ids      []string
-		Orm      *gorm.DB
+		Order       string
+		PageNum     int
+		PageSize    int
+		Ids         []string
+		Orm         *gorm.DB
+		CategoryIds []int
 	}
 
 	IdAndName struct {
@@ -157,6 +158,11 @@
 	return slf
 }
 
+func (slf *MaterialSearch) SetCategoryIds(ids []int) *MaterialSearch {
+	slf.CategoryIds = ids
+	return slf
+}
+
 //
 //func (slf *MaterialSearch) SetSetTemplateType(setType constvar.SetTemplateType) *MaterialSearch {
 //	slf.SetTemplateType = setType
@@ -230,6 +236,9 @@
 	if slf.CategoryId > 0 {
 		db = db.Where("category_id = ?", slf.CategoryId)
 	}
+	if len(slf.CategoryIds) > 0 {
+		db = db.Where("category_id in ?", slf.CategoryIds)
+	}
 
 	return db
 }

--
Gitblit v1.8.0