pref: 优化批量导入数据库

features/udm_user/repo/repo_udm_sub_user.go

@@ -213,8 +213,7 @@ func (r *RepoUdmSubUser) ClearAndInsert(neID string, subArr []model.UdmSubUser)
 }
 
 // Insert 新增实体
-func (r *RepoUdmSubUser) Insert(neID string, subUser model.UdmSubUser) int64 {
-	subUser.NeID = neID
+func (r *RepoUdmSubUser) Insert(subUser model.UdmSubUser) int64 {
 	results, err := datasource.DefaultDB().Table("u_sub_user").Insert(subUser)
 	if err != nil {
 		return results
@@ -223,57 +222,25 @@ func (r *RepoUdmSubUser) Insert(neID string, subUser model.UdmSubUser) int64 {
 }
 
 // Insert 批量添加
-func (r *RepoUdmSubUser) Inserts(neID string, subUser model.UdmSubUser, num string) int64 {
-	var insertNum int64
-
-	imsiVlen := len(subUser.Imsi)
-	imsiV, err := strconv.Atoi(subUser.Imsi)
-	if err != nil {
-		return 0
-	}
-	msisdnVlen := len(subUser.Msisdn)
-	msisdnV, err := strconv.Atoi(subUser.Msisdn)
-	if err != nil {
-		return 0
-	}
-
-	numV, err := strconv.Atoi(num)
-	if err != nil {
-		return 0
-	}
-
-	subUser.NeID = neID
-	for i := 0; i < numV; i++ {
-		msisdn := fmt.Sprint(msisdnV + i)
-		if len(msisdn) < msisdnVlen {
-			msisdn = fmt.Sprintf("%0*s", msisdnVlen, msisdn)
+func (r *RepoUdmSubUser) Inserts(subUser []model.UdmSubUser) int64 {
+	var num int64
+	n := len(subUser)
+	batchSize := 5000
+	for i := 0; i < n; i += batchSize {
+		end := i + batchSize
+		if end > n {
+			end = n
 		}
-		subUser.Msisdn = msisdn
+		batch := subUser[i:end]
 
-		imsi := fmt.Sprint(imsiV + i)
-		if len(imsi) < imsiVlen {
-			imsi = fmt.Sprintf("%0*s", imsiVlen, imsi)
-		}
-		subUser.Imsi = imsi
-
-		// IP会自动递增
-		if subUser.StaticIp != "" {
-			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
+		// 调用 InsertMulti 函数将批量数据插入数据库
+		results, err := datasource.DefaultDB().Table("u_sub_user").InsertMulti(batch)
+		if err != nil {
+			continue
 		}
+		num += results
 	}
-
-	return insertNum
+	return num
 }
 
 // Insert4G 批量添加4G用户