feat 签约用户4G接口

2023-09-12 11:11:27 +08:00
parent df5cc36de4
commit fc8f63fca6
4 changed files with 253 additions and 1 deletions
--- a/features/udm_user/repo/repo_udm_sub_user.go
+++ b/features/udm_user/repo/repo_udm_sub_user.go
@@ -209,6 +209,41 @@ func (r *RepoUdmSubUser) Inserts(neID string, subUser model.UdmSubUser, num stri
 	for i := 0; i < numV; i++ {
 		subUser.Imsi = fmt.Sprint(imsiV + i)
 		subUser.Msisdn = fmt.Sprint(msisdnV + i)
+		// IP会自动递增
+		parts := strings.Split(subUser.StaticIp, ".")
+		lastPart := parts[3]
+		lastNum, _ := strconv.Atoi(lastPart)
+		lastNum += i
+		newLastPart := strconv.Itoa(lastNum)
+		parts[3] = newLastPart
+		newIP := strings.Join(parts, ".")
+		subUser.StaticIp = newIP
+		results, err := datasource.DefaultDB().Table("u_sub_user").Insert(subUser)
+		if err == nil {
+			insertNum += results
+		}
+	}
+
+	return insertNum
+}
+
+// Insert4G 批量添加4G用户
+func (r *RepoUdmSubUser) Insert4G(neID string, subUser model.UdmSubUser) int64 {
+	var insertNum int64
+
+	imsiV, err := strconv.Atoi(subUser.Imsi)
+	if err != nil {
+		return 0
+	}
+	numV, err := strconv.Atoi(subUser.SubNum)
+	if err != nil {
+		return 0
+	}
+
+	subUser.NeID = neID
+	for i := 0; i < numV; i++ {
+		subUser.Imsi = fmt.Sprint(imsiV + i)
+
 		results, err := datasource.DefaultDB().Table("u_sub_user").Insert(subUser)
 		if err == nil {
 			insertNum += results
@@ -262,6 +297,78 @@ func (r *RepoUdmSubUser) Update(neID string, authUser model.UdmSubUser) int64 {
 	return results
 }

+// Update4GIP 批量修改4G IP
+func (r *RepoUdmSubUser) Update4GIP(neID string, subUser model.UdmSubUser) int64 {
+	var insertNum int64
+
+	imsiV, err := strconv.Atoi(subUser.Imsi)
+	if err != nil || subUser.StaticIp == "" {
+		return insertNum
+	}
+	numV, err := strconv.Atoi(subUser.SubNum)
+	if err != nil {
+		return insertNum
+	}
+
+	for i := 0; i < numV; i++ {
+		subUser.Imsi = fmt.Sprint(imsiV + i)
+
+		// 查询先
+		var user model.UdmSubUser
+		err := datasource.DefaultDB().Table("u_sub_user").Where("imsi = ? and ne_id = ?", subUser.Imsi, neID).Find(&user)
+		if err == nil {
+			// IP会自动递增
+			parts := strings.Split(subUser.StaticIp, ".")
+			lastPart := parts[3]
+			lastNum, _ := strconv.Atoi(lastPart)
+			lastNum += i
+			newLastPart := strconv.Itoa(lastNum)
+			parts[3] = newLastPart
+			newIP := strings.Join(parts, ".")
+			user.StaticIp = newIP
+			// 更新
+			results, err := datasource.DefaultDB().Table("u_sub_user").Update(user)
+			if err == nil {
+				insertNum += results
+			}
+		}
+	}
+	return insertNum
+}
+
+// UpdateSmData 批量修改sm-data
+func (r *RepoUdmSubUser) UpdateSmData(neID string, subUser model.UdmSubUser) int64 {
+	var insertNum int64
+
+	imsiV, err := strconv.Atoi(subUser.Imsi)
+	if err != nil || subUser.StaticIp == "" {
+		return insertNum
+	}
+	numV, err := strconv.Atoi(subUser.SubNum)
+	if err != nil {
+		return insertNum
+	}
+
+	for i := 0; i < numV; i++ {
+		subUser.Imsi = fmt.Sprint(imsiV + i)
+
+		// 查询先
+		var user model.UdmSubUser
+		err := datasource.DefaultDB().Table("u_sub_user").Where("imsi = ? and ne_id = ?", subUser.Imsi, neID).Find(&user)
+		if err == nil {
+			// IP会自动递增，需提前规划好DNN对应的IP；如dnn不需要绑定IP则不带此字段名
+			// parts := strings.Split(subUser.SmData, "&")
+			user.SmData = subUser.SmData
+			// 更新
+			results, err := datasource.DefaultDB().Table("u_sub_user").Update(user)
+			if err == nil {
+				insertNum += results
+			}
+		}
+	}
+	return insertNum
+}
+
 // Delete 删除实体
 func (r *RepoUdmSubUser) Delete(neID, imsi string) int64 {
 	results, err := datasource.DefaultDB().Table("u_sub_user").Where("imsi = ? and ne_id = ?", imsi, neID).Delete()