feat: UDM用户数据添加imsi拓展信息关联

src/modules/network_data/service/udm_auth.go

@@ -1,28 +1,172 @@
 package service
 
-import "be.ems/src/modules/network_data/model"
+import (
+	"fmt"
+	"strconv"
+	"strings"
 
-// UDM鉴权信息 服务层接口
-type IUDMAuth interface {
-	// ResetData 重置鉴权用户数据，清空数据库重新同步Redis数据
-	ResetData(neId string) int64
+	"be.ems/src/framework/redis"
+	"be.ems/src/modules/network_data/model"
+	"be.ems/src/modules/network_data/repository"
+)
 
-	// SelectPage 分页查询数据库
-	SelectPage(query map[string]any) map[string]any
-
-	// SelectList 查询数据库
-	SelectList(u model.UDMAuth) []model.UDMAuth
-
-	// Insert 从数据中读取后删除imsi再存入数据库
-	// imsi长度15，ki长度32，opc长度0或者32
-	Insert(neId string, u model.UDMAuth) int64
-
-	// InsertData 导入文件数据 dataType目前两种：txt/csv
-	InsertData(neId, dataType string, data any) int64
-
-	// Delete 删除单个不重新加载
-	Delete(neID, imsi string) int64
-
-	// LoadData 删除范围后重新加载 num表示imsi后几位
-	LoadData(neID, imsi, num string) int64
+// 实例化服务层 UDMAuthUser 结构体
+var NewUDMAuthUser = &UDMAuthUser{
+	udmAuthRepository: repository.NewUDMAuthUser,
+}
+
+// UDM鉴权信息 服务层处理
+type UDMAuthUser struct {
+	// UDM鉴权信息数据信息
+	udmAuthRepository *repository.UDMAuthUser
+}
+
+// dataByRedis UDM鉴权用户 db:0 中 ausf:*
+func (r *UDMAuthUser) dataByRedis(imsi, neId string) []model.UDMAuthUser {
+	arr := []model.UDMAuthUser{}
+	key := fmt.Sprintf("ausf:%s", imsi)
+	ausfArr, err := redis.GetKeys("udmuser", key)
+	if err != nil {
+		return arr
+	}
+	for _, key := range ausfArr {
+		m, err := redis.GetHash("udmuser", key)
+		if err != nil {
+			continue
+		}
+
+		// 跳过-号数据 ausf:360000100000130
+		imsi := key[5:]
+		if strings.Contains(imsi, "-") {
+			continue
+		}
+
+		amf := ""
+		if v, ok := m["amf"]; ok {
+			amf = strings.Replace(v, "\r\n", "", 1)
+		}
+		a := model.UDMAuthUser{
+			IMSI:      imsi,
+			Amf:       amf,
+			Status:    "1", // 默认给1
+			Ki:        m["ki"],
+			AlgoIndex: m["algo"],
+			Opc:       m["opc"],
+			NeId:      neId,
+		}
+		arr = append(arr, a)
+	}
+	return arr
+}
+
+// ResetData 重置鉴权用户数据，清空数据库重新同步Redis数据
+func (r *UDMAuthUser) ResetData(neId string) int64 {
+	authArr := r.dataByRedis("*", neId)
+	// 数据清空后添加
+	go r.udmAuthRepository.ClearAndInsert(neId, authArr)
+	return int64(len(authArr))
+}
+
+// ParseInfo 解析单个用户imsi鉴权信息 data从命令MML得到的结果
+func (r *UDMAuthUser) ParseInfo(imsi, neId string, data map[string]string) model.UDMAuthUser {
+	u := model.UDMAuthUser{
+		IMSI:      imsi,
+		NeId:      neId,
+		Amf:       data["amf"],
+		Status:    "1",
+		Ki:        data["ki"],
+		AlgoIndex: data["algo"],
+		Opc:       data["opc"],
+	}
+
+	// 补充用户ID用于更新
+	auth := r.udmAuthRepository.SelectByIMSIAndNeID(imsi, neId)
+	if auth.IMSI == imsi {
+		u.ID = auth.ID
+	}
+	return u
+}
+
+// SelectPage 分页查询数据库
+func (r *UDMAuthUser) SelectPage(query map[string]any) map[string]any {
+	return r.udmAuthRepository.SelectPage(query)
+}
+
+// SelectList 查询数据库
+func (r *UDMAuthUser) SelectList(u model.UDMAuthUser) []model.UDMAuthUser {
+	return r.udmAuthRepository.SelectList(u)
+}
+
+// Insert 从数据中读取后删除imsi再存入数据库
+// imsi长度15，ki长度32，opc长度0或者32
+func (r *UDMAuthUser) Insert(neId string, u model.UDMAuthUser) int64 {
+	uArr := r.dataByRedis(u.IMSI, neId)
+	if len(uArr) > 0 {
+		r.udmAuthRepository.Delete(u.IMSI, neId)
+		return r.udmAuthRepository.Inserts(uArr)
+	}
+	return 0
+}
+
+// InsertData 导入文件数据 dataType目前两种：txt/csv
+func (r *UDMAuthUser) InsertData(neId, dataType string, data any) int64 {
+	// imsi截取前缀,重新获取部分数据
+	prefixes := make(map[string]struct{})
+
+	if dataType == "csv" {
+		for _, v := range data.([]map[string]string) {
+			imsi := v["imsi"]
+			if len(imsi) < 6 {
+				continue
+			}
+			prefix := imsi[:len(imsi)-4]
+			prefixes[prefix] = struct{}{}
+		}
+	}
+	if dataType == "txt" {
+		for _, v := range data.([][]string) {
+			imsi := v[0]
+			if len(imsi) < 6 {
+				continue
+			}
+			prefix := imsi[:len(imsi)-4]
+			prefixes[prefix] = struct{}{}
+		}
+	}
+
+	// 根据前缀重新加载插入
+	var num int64 = 0
+	for prefix := range prefixes {
+		// 直接删除前缀的记录
+		r.udmAuthRepository.DeletePrefixByIMSI(neId, prefix)
+		// keys ausf:4600001000004*
+		arr := r.dataByRedis(prefix+"*", neId)
+		if len(arr) > 0 {
+			num += r.udmAuthRepository.Inserts(arr)
+		}
+	}
+	return num
+}
+
+// Delete 删除单个不重新加载
+func (r *UDMAuthUser) Delete(imsi, neId string) int64 {
+	return r.udmAuthRepository.Delete(imsi, neId)
+}
+
+// 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
+		}
+		r.udmAuthRepository.Inserts(arr)
+	}
 }