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)) }