From bedfbcbe3af9635d15d0cafece60fbd311159fe5 Mon Sep 17 00:00:00 2001
From: liujiandao <274878379@qq.com>
Date: 星期三, 10 四月 2024 13:51:08 +0800
Subject: [PATCH] 字段修改

---
 models/dict.go |   20 +++++++++++++++-----
 1 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/models/dict.go b/models/dict.go
index 7e682d8..fd32dd3 100644
--- a/models/dict.go
+++ b/models/dict.go
@@ -19,11 +19,12 @@
 
 	DictSearch struct {
 		Dict
-		Order    string
-		PageNum  int
-		PageSize int
-		Orm      *gorm.DB
-		Keyword  string
+		Order     string
+		PageNum   int
+		PageSize  int
+		Orm       *gorm.DB
+		Keyword   string
+		DictTypes []constvar.DictType
 	}
 )
 
@@ -75,6 +76,11 @@
 	return slf
 }
 
+func (slf *DictSearch) SetDictTypes(dts []constvar.DictType) *DictSearch {
+	slf.DictTypes = dts
+	return slf
+}
+
 func (slf *DictSearch) build() *gorm.DB {
 	var db = slf.Orm.Table(slf.TableName())
 
@@ -103,6 +109,10 @@
 		db = db.Where("name like ? or number like ?", kw, kw)
 	}
 
+	if len(slf.DictTypes) > 0 {
+		db = db.Where("dict_type in (?)", slf.DictTypes)
+	}
+
 	return db
 }
 

--
Gitblit v1.8.0