OMC/be.ems
1
0
Fork 0
Code Issues Releases 3 Activity

fix: udm用户信息直接返回后存入数据库

Browse Source
This commit is contained in:
TsMask
2023-11-01 10:39:47 +08:00
parent 928448b548
commit 73f3d550a7
1 changed files with 42 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
features/udm_user/api_udm_user.go
View File

@@ -254,20 +254,18 @@ func (s *UdmUserApi) UdmAuthUserInfo(w http.ResponseWriter, r *http.Request) {
ctx.JSON(w, 200, result.ErrMsg(err.Error())) ctx.JSON(w, 200, result.ErrMsg(err.Error()))
return return
} }
neId = "" // 返回查询的用户信息
var userInfo model.UdmAuthUser userInfo := model.UdmAuthUser{
list := s.authUser.List(model.UdmAuthUser{NeID: neId, Imsi: imsi})
if len(list) > 0 {
userInfo = list[0]
} else {
userInfo = model.UdmAuthUser{
Imsi: imsi, Imsi: imsi,
Amf: data["amf"], Amf: data["amf"],
AlgoIndex: data["algo"], AlgoIndex: data["algo"],
Opc: data["opc"], Opc: data["opc"],
Ki: data["ki"], Ki: data["ki"],
} }
// 查询数据库是否存在并存入
neId = ""
list := s.authUser.List(model.UdmAuthUser{NeID: neId, Imsi: imsi})
if len(list) == 0 {
s.authUser.Insert(neId, userInfo) s.authUser.Insert(neId, userInfo)
} }
ctx.JSON(w, 200, result.OkData(userInfo)) ctx.JSON(w, 200, result.OkData(userInfo))
@@ -655,15 +653,10 @@ func (s *UdmUserApi) UdmSubUserInfo(w http.ResponseWriter, r *http.Request) {
return return
} }
neId = "" // 返回查询的用户信息
var userInfo model.UdmSubUser
list := s.subUser.List(model.UdmSubUser{NeID: neId, Imsi: imsi})
if len(list) > 0 {
userInfo = list[0]
} else {
cnType, _ := strconv.ParseInt(data["CNType"][:4], 0, 64) cnType, _ := strconv.ParseInt(data["CNType"][:4], 0, 64)
rat, _ := strconv.ParseInt(data["RAT"][:4], 0, 64) rat, _ := strconv.ParseInt(data["RAT"][:4], 0, 64)
userInfo = model.UdmSubUser{ userInfo := model.UdmSubUser{
Imsi: imsi, Imsi: imsi,
Msisdn: data["MSISDN"], Msisdn: data["MSISDN"],
Ambr: data["AMBR"], Ambr: data["AMBR"],
@@ -689,6 +682,10 @@ func (s *UdmUserApi) UdmSubUserInfo(w http.ResponseWriter, r *http.Request) {
userInfo.StaticIp = arr[8] userInfo.StaticIp = arr[8]
} }
// 查询数据库是否存在并存入
neId = ""
list := s.subUser.List(model.UdmSubUser{NeID: neId, Imsi: imsi})
if len(list) == 0 {
s.subUser.Insert(neId, userInfo) s.subUser.Insert(neId, userInfo)
} }
ctx.JSON(w, 200, result.OkData(userInfo)) ctx.JSON(w, 200, result.OkData(userInfo))