From 89356af2c7dd33fe7ce25ad3de1fb576a3db068b Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Fri, 22 Dec 2023 12:23:08 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20UDM=E7=94=A8=E6=88=B7=E8=8E=B7=E5=8F=96?=
 =?UTF-8?q?=E6=9F=A5=E8=AF=A2=E4=B8=BA=E7=A9=BA=E5=80=BC=E6=83=85=E5=86=B5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/modules/network_element/controller/udm_auth.go   | 5 +++++
 src/modules/network_element/controller/udm_sub.go    | 5 +++++
 src/modules/network_element/service/udm_auth.impl.go | 6 +-----
 src/modules/network_element/service/udm_sub.impl.go  | 6 +-----
 4 files changed, 12 insertions(+), 10 deletions(-)

diff --git a/src/modules/network_element/controller/udm_auth.go b/src/modules/network_element/controller/udm_auth.go
index d1acc52e..45929fe8 100644
--- a/src/modules/network_element/controller/udm_auth.go
+++ b/src/modules/network_element/controller/udm_auth.go
@@ -89,6 +89,11 @@ func (s *UDMAuthController) Info(c *gin.Context) {
 		return
 	}
 
+	if len(data) == 0 {
+		c.JSON(200, result.ErrMsg("No Auth Data"))
+		return
+	}
+
 	// 查询数据库是否存在并存入
 	neId = ""
 	var userInfo model.UDMAuth
diff --git a/src/modules/network_element/controller/udm_sub.go b/src/modules/network_element/controller/udm_sub.go
index df738622..40909313 100644
--- a/src/modules/network_element/controller/udm_sub.go
+++ b/src/modules/network_element/controller/udm_sub.go
@@ -90,6 +90,11 @@ func (s *UDMSubController) Info(c *gin.Context) {
 		return
 	}
 
+	if len(data) == 0 {
+		c.JSON(200, result.ErrMsg("No Subs Data"))
+		return
+	}
+
 	// 解析返回的数据
 	cnType, _ := strconv.ParseInt(data["CNType"][:4], 0, 64)
 	rat, _ := strconv.ParseInt(data["RAT"][:4], 0, 64)
diff --git a/src/modules/network_element/service/udm_auth.impl.go b/src/modules/network_element/service/udm_auth.impl.go
index 75a051a1..cd1ac30c 100644
--- a/src/modules/network_element/service/udm_auth.impl.go
+++ b/src/modules/network_element/service/udm_auth.impl.go
@@ -62,12 +62,8 @@ func (r *UDMAuthImpl) authDataByRedis(imsi, neID string) []model.UDMAuth {
 
 // Save UDM鉴权用户-获取redis全部保存数据库
 func (r *UDMAuthImpl) Save(neID string) int64 {
-	var num int64 = 0
 	authArr := r.authDataByRedis("*", neID)
-	// 有数据才清空
-	if len(authArr) == 0 {
-		return num
-	}
+	// 数据清空后添加
 	go r.udmAuthRepository.ClearAndInsert(neID, authArr)
 	return int64(len(authArr))
 }
diff --git a/src/modules/network_element/service/udm_sub.impl.go b/src/modules/network_element/service/udm_sub.impl.go
index 45fc5cd4..3bc27c59 100644
--- a/src/modules/network_element/service/udm_sub.impl.go
+++ b/src/modules/network_element/service/udm_sub.impl.go
@@ -78,12 +78,8 @@ func (r *UDMSubImpl) subDataByRedis(imsi, neID string) []model.UDMSub {
 
 // Save UDM签约用户-获取redis全部保存数据库
 func (r *UDMSubImpl) Save(neID string) int64 {
-	var num int64 = 0
 	subArr := r.subDataByRedis("*", neID)
-	// 有数据才清空
-	if len(subArr) == 0 {
-		return num
-	}
+	// 数据清空后添加
 	go r.udmSubRepository.ClearAndInsert(neID, subArr)
 	return int64(len(subArr))
 }