From 68b7014ed48e5d6af9d3fdc6831bd287fb63754b Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Thu, 6 Mar 2025 11:16:12 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20UDM=E6=95=B0=E6=8D=AEIMSI=E5=8B=BE?=
 =?UTF-8?q?=E9=80=89=E5=AF=BC=E5=87=BA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/modules/network_data/repository/udm_auth.go | 5 ++++-
 src/modules/network_data/repository/udm_sub.go  | 5 ++++-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/modules/network_data/repository/udm_auth.go b/src/modules/network_data/repository/udm_auth.go
index ffb882b2..c8f4d29d 100644
--- a/src/modules/network_data/repository/udm_auth.go
+++ b/src/modules/network_data/repository/udm_auth.go
@@ -2,6 +2,7 @@ package repository
 
 import (
 	"fmt"
+	"strings"
 
 	"be.ems/src/framework/database/db"
 	"be.ems/src/framework/logger"
@@ -36,7 +37,9 @@ func (r *UDMAuthUser) SelectPage(query map[string]string) (int64, []model.UDMAut
 		tx = tx.Where("ne_id = ?", v)
 	}
 	if v, ok := query["imsis"]; ok && v != "" {
-		tx = tx.Where("imsi in ?", v)
+		arr := strings.Split(v, ",")
+		tx = tx.Where("imsi in ?", arr)
+		query["pageSize"] = fmt.Sprint(len(arr))
 	}
 
 	var total int64 = 0
diff --git a/src/modules/network_data/repository/udm_sub.go b/src/modules/network_data/repository/udm_sub.go
index 375e610e..bd1b97a0 100644
--- a/src/modules/network_data/repository/udm_sub.go
+++ b/src/modules/network_data/repository/udm_sub.go
@@ -2,6 +2,7 @@ package repository
 
 import (
 	"fmt"
+	"strings"
 
 	"be.ems/src/framework/database/db"
 	"be.ems/src/framework/logger"
@@ -39,7 +40,9 @@ func (r *UDMSubUser) SelectPage(query map[string]string) (int64, []model.UDMSubU
 		tx = tx.Where("ne_id =?", v)
 	}
 	if v, ok := query["imsis"]; ok && v != "" {
-		tx = tx.Where("imsi in ?", v)
+		arr := strings.Split(v, ",")
+		tx = tx.Where("imsi in ?", arr)
+		query["pageSize"] = fmt.Sprint(len(arr))
 	}
 
 	var total int64 = 0