fix: 网元模块（查询网元，UDM用户信息）

src/modules/network_element/service/udm_sub.impl.go

@@ -0,0 +1,191 @@
+package service
+
+import (
+	"fmt"
+	"strings"
+
+	"ems.agt/src/framework/redis"
+	"ems.agt/src/modules/network_element/model"
+	"ems.agt/src/modules/network_element/repository"
+)
+
+// 实例化服务层 UDMSubImpl 结构体
+var NewUDMSubImpl = &UDMSubImpl{
+	udmSubRepository: repository.NewUDMSubImpl,
+}
+
+// UDM签约信息 服务层处理
+type UDMSubImpl struct {
+	// UDM签约信息数据信息
+	udmSubRepository repository.IUDMSub
+}
+
+// subDataByRedis UDM签约用户
+func (r *UDMSubImpl) subDataByRedis(imsi, neID string) []model.UDMSub {
+	arr := []model.UDMSub{}
+	udmsdArr, err := redis.GetKeys("udmuser", fmt.Sprintf("udm-sd:%s", imsi))
+	if err != nil {
+		return arr
+	}
+	for _, key := range udmsdArr {
+		m, err := redis.GetHash("udmuser", key)
+		if err != nil {
+			continue
+		}
+
+		a := model.UDMSub{
+			Imsi:   key[7:],
+			Msisdn: m["gpsi"], // 46003550072 strings.TrimPrefix(m["gpsi"], "86"),
+			SmfSel: m["smf-sel"],
+			SmData: m["sm-dat"], // 1-000001&cmnet&ims&3gnet
+			NeID:   neID,
+		}
+
+		// def_ambr,def_nssai,0,def_arfb,def_sar,3,1,12000,1,1000,0,1,-
+		if v, ok := m["am-dat"]; ok {
+			arr := strings.Split(v, ",")
+			a.Ambr = arr[0]
+			a.Nssai = arr[1]
+			a.Rat = arr[2]
+			a.Arfb = arr[3]
+			a.Sar = arr[4]
+			a.Cn = arr[5]
+		}
+		// 1,64,24,65,def_eps,1,2,010200000000,-
+		if v, ok := m["eps-dat"]; ok {
+			arr := strings.Split(v, ",")
+			// 跳过非常规数据
+			if len(arr) > 9 {
+				continue
+			}
+			a.EpsDat = v
+			a.EpsFlag = arr[0]
+			a.EpsOdb = arr[1]
+			a.HplmnOdb = arr[2]
+			a.Ard = arr[3]
+			a.Epstpl = arr[4]
+			a.ContextId = arr[5]
+			a.ApnContext = arr[7]
+			// [6] 是不要的，导入和导出不用
+			a.StaticIp = arr[8]
+		}
+
+		arr = append(arr, a)
+	}
+	return arr
+}
+
+// Save UDM签约用户-获取redis全部保存数据库
+func (r *UDMSubImpl) Save(neID string) int64 {
+	var num int64 = 0
+	subArr := r.subDataByRedis("*", neID)
+	// 有数据才清空
+	if len(subArr) == 0 {
+		return num
+	}
+	go r.udmSubRepository.ClearAndInsert(neID, subArr)
+	return int64(len(subArr))
+}
+
+// Page  UDM签约用户-分页查询数据库
+func (r *UDMSubImpl) Page(query map[string]any) map[string]any {
+	return r.udmSubRepository.SelectPage(query)
+}
+
+// List  UDM签约用户-查询数据库
+func (r *UDMSubImpl) List(subUser model.UDMSub) []model.UDMSub {
+	return r.udmSubRepository.SelectList(subUser)
+}
+
+// Insert UDM签约用户-新增单个
+// imsi长度15，ki长度32，opc长度0或者32
+func (r *UDMSubImpl) Insert(neID string, subUser model.UDMSub) int64 {
+	authArr := r.subDataByRedis(subUser.Imsi, neID)
+	if len(authArr) > 0 {
+		r.udmSubRepository.Delete(neID, subUser.Imsi)
+		r.udmSubRepository.Inserts(authArr)
+	}
+	return 0
+}
+
+// Insert UDM签约用户-批量添加
+func (r *UDMSubImpl) Inserts(neID string, subUser model.UDMSub, num string) int64 {
+	startIMSI := subUser.Imsi
+	startIMSI = startIMSI[:len(startIMSI)-len(num)] + "*"
+	// keys udm-sd:4600001000004*
+	subArr := r.subDataByRedis(startIMSI, neID)
+	if len(subArr) > 0 {
+		for _, v := range subArr {
+			r.udmSubRepository.Delete(neID, v.Imsi)
+		}
+		return r.udmSubRepository.Inserts(subArr)
+	}
+	return 0
+}
+
+// InsertCSV UDM签约用户-批量添加
+func (r *UDMSubImpl) InsertCSV(neID string, data []map[string]string) int64 {
+	prefixes := make(map[string]struct{})
+	for _, v := range data {
+		imsi := v["imsi"]
+		prefix := imsi[:len(imsi)-4]
+		prefixes[prefix] = struct{}{}
+	}
+	// 分组插入
+	var num int64 = 0
+	for prefix := range prefixes {
+		r.udmSubRepository.DeletePrefixImsi(neID, prefix)
+		subArr := r.subDataByRedis(prefix+"*", neID)
+		if len(subArr) > 0 {
+			num += r.udmSubRepository.Inserts(subArr)
+		}
+	}
+	return num
+}
+
+// InsertTxt UDM签约用户-批量添加
+func (r *UDMSubImpl) InsertTxt(neID string, data [][]string) int64 {
+	prefixes := make(map[string]struct{})
+	for _, v := range data {
+		imsi := v[0]
+		prefix := imsi[:len(imsi)-4]
+		prefixes[prefix] = struct{}{}
+	}
+	// 分组插入
+	var num int64 = 0
+	for prefix := range prefixes {
+		r.udmSubRepository.DeletePrefixImsi(neID, prefix)
+		subArr := r.subDataByRedis(prefix+"*", neID)
+		if len(subArr) > 0 {
+			num += r.udmSubRepository.Inserts(subArr)
+		}
+	}
+	return num
+}
+
+// Insert UDM签约用户-修改更新
+func (r *UDMSubImpl) Update(neID string, subUser model.UDMSub) int64 {
+	authArr := r.subDataByRedis(subUser.Imsi, neID)
+	if len(authArr) > 0 {
+		r.udmSubRepository.Delete(neID, subUser.Imsi)
+		return r.udmSubRepository.Inserts(authArr)
+	}
+	return 0
+}
+
+// Insert UDM签约用户-删除单个
+func (r *UDMSubImpl) Delete(neID, imsi string) int64 {
+	return r.udmSubRepository.Delete(neID, imsi)
+}
+
+// Insert UDM签约用户-删除范围
+func (r *UDMSubImpl) Deletes(neID, imsi, num string) int64 {
+	prefix := imsi[:len(imsi)-len(num)]
+	// keys udm-sd:4600001000004*
+	authArr := r.subDataByRedis(prefix+"*", neID)
+	if len(authArr) > 0 {
+		r.udmSubRepository.DeletePrefixImsi(neID, prefix)
+		return r.udmSubRepository.Inserts(authArr)
+	}
+	return 0
+}