fxi: 鉴权用户KI/OPC信息显示

features/udm_user/api_udm_user.go

@@ -254,7 +254,19 @@ func (s *UdmUserApi) UdmAuthUserInfo(w http.ResponseWriter, r *http.Request) {
 		ctx.JSON(w, 200, result.ErrMsg(err.Error()))
 		return
 	}
+
+	// 查询数据库是否存在并存入
+	neId = ""
+	var userInfo model.UdmAuthUser
+	list := s.authUser.List(model.UdmAuthUser{NeID: neId, Imsi: imsi})
+	if len(list) > 0 {
+		userInfo = list[0]
 		// 返回查询的用户信息
+		userInfo.Amf = data["amf"]
+		userInfo.AlgoIndex = data["algo"]
+		userInfo.Opc = data["opc"]
+		userInfo.Ki = data["ki"]
+	} else {
 		userInfo := model.UdmAuthUser{
 			Imsi:      imsi,
 			Amf:       data["amf"],
@@ -262,10 +274,6 @@ func (s *UdmUserApi) UdmAuthUserInfo(w http.ResponseWriter, r *http.Request) {
 			Opc:       data["opc"],
 			Ki:        data["ki"],
 		}
-	// 查询数据库是否存在并存入
-	neId = ""
-	list := s.authUser.List(model.UdmAuthUser{NeID: neId, Imsi: imsi})
-	if len(list) == 0 {
 		s.authUser.Insert(neId, userInfo)
 	}
 	ctx.JSON(w, 200, result.OkData(userInfo))
@@ -653,7 +661,13 @@ func (s *UdmUserApi) UdmSubUserInfo(w http.ResponseWriter, r *http.Request) {
 		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)
 		rat, _ := strconv.ParseInt(data["RAT"][:4], 0, 64)
 		userInfo := model.UdmSubUser{
@@ -681,11 +695,6 @@ func (s *UdmUserApi) UdmSubUserInfo(w http.ResponseWriter, r *http.Request) {
 			userInfo.ApnContext = arr[7]
 			userInfo.StaticIp = arr[8]
 		}
-
-	// 查询数据库是否存在并存入
-	neId = ""
-	list := s.subUser.List(model.UdmSubUser{NeID: neId, Imsi: imsi})
-	if len(list) == 0 {
 		s.subUser.Insert(neId, userInfo)
 	}
 	ctx.JSON(w, 200, result.OkData(userInfo))