marge: 合并代码

features/udm_user/api_udm_user.go

@@ -254,20 +254,26 @@ 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{
+		userInfo := model.UdmAuthUser{
 			Imsi:      imsi,
 			Amf:       data["amf"],
 			AlgoIndex: data["algo"],
 			Opc:       data["opc"],
 			Ki:        data["ki"],
 		}
-
 		s.authUser.Insert(neId, userInfo)
 	}
 	ctx.JSON(w, 200, result.OkData(userInfo))
@@ -655,6 +661,7 @@ 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})
@@ -663,7 +670,7 @@ func (s *UdmUserApi) UdmSubUserInfo(w http.ResponseWriter, r *http.Request) {
 	} else {
 		cnType, _ := strconv.ParseInt(data["CNType"][:4], 0, 64)
 		rat, _ := strconv.ParseInt(data["RAT"][:4], 0, 64)
-		userInfo = model.UdmSubUser{
+		userInfo := model.UdmSubUser{
 			Imsi:   imsi,
 			Msisdn: data["MSISDN"],
 			Ambr:   data["AMBR"],
@@ -688,7 +695,6 @@ func (s *UdmUserApi) UdmSubUserInfo(w http.ResponseWriter, r *http.Request) {
 			userInfo.ApnContext = arr[7]
 			userInfo.StaticIp = arr[8]
 		}
-
 		s.subUser.Insert(neId, userInfo)
 	}
 	ctx.JSON(w, 200, result.OkData(userInfo))

features/udm_user/repo/repo_udm_auth_user.go

@@ -177,6 +177,7 @@ func (r *RepoUdmAuthUser) ClearAndInsert(neID string, authArr []model.UdmAuthUse
 		// 调用 InsertMulti 函数将批量数据插入数据库
 		results, err := datasource.DefaultDB().Table("u_auth_user").InsertMulti(batch)
 		if err != nil {
+			log.Errorf("InsertMulti err => %v", err)
 			continue
 		}
 		num += results

features/udm_user/repo/repo_udm_sub_user.go

@@ -196,6 +196,7 @@ func (r *RepoUdmSubUser) ClearAndInsert(neID string, subArr []model.UdmSubUser)
 		// 调用 InsertMulti 函数将批量数据插入数据库
 		results, err := datasource.DefaultDB().Table("u_sub_user").InsertMulti(batch)
 		if err != nil {
+			log.Errorf("InsertMulti err => %v", err)
 			continue
 		}
 		num += results