fix: UDM批量操作加载数据根据前缀重载

This commit is contained in:
TsMask
2025-10-11 11:42:26 +08:00
parent b401748dac
commit 8f7636bc2a
5 changed files with 53 additions and 73 deletions

View File

@@ -2,7 +2,6 @@ package service
import (
"fmt"
"strconv"
"strings"
"time"
@@ -176,18 +175,13 @@ func (r *UDMAuthUser) Delete(imsi, neId string) int64 {
// LoadData 重新加载从imsi开始num的数据
func (r *UDMAuthUser) LoadData(neId, imsi, num string) {
startIMSI, _ := strconv.ParseInt(imsi, 10, 64)
subNum, _ := strconv.ParseInt(num, 10, 64)
var i int64
for i = 0; i < subNum; i++ {
keyIMSI := fmt.Sprintf("%015d", startIMSI+i)
// 删除原数据
r.udmAuthRepository.Delete(keyIMSI, neId)
// 加载数据
arr := r.dataByRedis(keyIMSI, neId)
if len(arr) < 1 {
continue
}
// 直接删除前缀的记录
index := len(imsi) - len(num) - 1
prefix := imsi[:index]
r.udmAuthRepository.DeletePrefixByIMSI(prefix, neId)
// 加载数据
arr := r.dataByRedis(prefix+"*", neId)
if len(arr) > 0 {
r.udmAuthRepository.Inserts(arr)
}
}