From 88485221a0f214514c53e97373c7b46e35aa08c1 Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Tue, 29 Oct 2024 11:02:44 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20UDM=E9=89=B4=E6=9D=83=E7=AD=BE=E7=BA=A6?=
 =?UTF-8?q?imsi=E5=A4=9A=E6=9F=A5=E8=AF=A2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/modules/network_data/repository/udm_auth.go | 7 +++++++
 src/modules/network_data/repository/udm_sub.go  | 7 +++++++
 2 files changed, 14 insertions(+)

diff --git a/src/modules/network_data/repository/udm_auth.go b/src/modules/network_data/repository/udm_auth.go
index b1429a4b..582e9573 100644
--- a/src/modules/network_data/repository/udm_auth.go
+++ b/src/modules/network_data/repository/udm_auth.go
@@ -74,6 +74,13 @@ func (r *UDMAuthUser) SelectPage(query map[string]any) map[string]any {
 		conditions = append(conditions, "ne_id = ?")
 		params = append(params, v)
 	}
+	if v, ok := query["imsis"]; ok && v != "" {
+		placeholder := repo.KeyPlaceholderByQuery(len(v.([]any)))
+		conditions = append(conditions, fmt.Sprintf("imsi in (%s)", placeholder))
+		for _, v := range v.([]any) {
+			params = append(params, v.(string))
+		}
+	}
 
 	// 构建查询条件语句
 	whereSql := ""
diff --git a/src/modules/network_data/repository/udm_sub.go b/src/modules/network_data/repository/udm_sub.go
index d1025a33..b283a1cd 100644
--- a/src/modules/network_data/repository/udm_sub.go
+++ b/src/modules/network_data/repository/udm_sub.go
@@ -110,6 +110,13 @@ func (r *UDMSubUser) SelectPage(query map[string]any) map[string]any {
 		conditions = append(conditions, "ne_id = ?")
 		params = append(params, v)
 	}
+	if v, ok := query["imsis"]; ok && v != "" {
+		placeholder := repo.KeyPlaceholderByQuery(len(v.([]any)))
+		conditions = append(conditions, fmt.Sprintf("imsi in (%s)", placeholder))
+		for _, v := range v.([]any) {
+			params = append(params, v.(string))
+		}
+	}
 
 	// 构建查询条件语句
 	whereSql := ""