From 56e2431f4f61fd492230ddc10be1f0beace4172f Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Tue, 4 Mar 2025 16:25:07 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20UDM=E7=94=A8=E6=88=B7=E6=95=B0=E6=8D=AE?=
 =?UTF-8?q?=E5=8A=A0=E8=BD=BD=E9=94=99=E8=AF=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/modules/network_data/service/udm_auth.go | 10 +++++++---
 src/modules/network_data/service/udm_sub.go  | 16 +++++++++++++---
 2 files changed, 20 insertions(+), 6 deletions(-)

diff --git a/src/modules/network_data/service/udm_auth.go b/src/modules/network_data/service/udm_auth.go
index a951c765..62812a35 100644
--- a/src/modules/network_data/service/udm_auth.go
+++ b/src/modules/network_data/service/udm_auth.go
@@ -43,9 +43,13 @@ func (r *UDMAuthUser) dataByRedis(imsi, neId string) []model.UDMAuthUser {
 	if err != nil {
 		return arr
 	}
-	for _, key := range ausfArr {
-		m, err := redis.GetHash(source, key)
-		if err != nil {
+	mkv, err := redis.GetHashBatch(source, ausfArr)
+	if err != nil {
+		return arr
+	}
+
+	for k, m := range mkv {
+		if k == "-" {
 			continue
 		}
 
diff --git a/src/modules/network_data/service/udm_sub.go b/src/modules/network_data/service/udm_sub.go
index cb0184f8..894dae73 100644
--- a/src/modules/network_data/service/udm_sub.go
+++ b/src/modules/network_data/service/udm_sub.go
@@ -44,9 +44,19 @@ func (r *UDMSubUser) dataByRedis(imsi, neId string) []model.UDMSubUser {
 	if err != nil {
 		return arr
 	}
-	for _, key := range udmsdArr {
-		m, err := redis.GetHash(source, key)
-		if err != nil {
+	mkv, err := redis.GetHashBatch(source, udmsdArr)
+	if err != nil {
+		return arr
+	}
+
+	for k, m := range mkv {
+		if k == "-" {
+			continue
+		}
+
+		// 跳过-号数据 udm-sd:360000100000130
+		imsi := k[7:]
+		if strings.Contains(imsi, "-") {
 			continue
 		}