OMC/be.ems
1
0
Fork 0
Code Issues Releases 3 Activity

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

Browse Source
This commit is contained in:
TsMask
2023-11-20 18:49:31 +08:00
parent 03b404ed18
commit 0c3cd1b31f
21 changed files with 2225 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/modules/network_element/service/ne_info.go Normal file
View File

@@ -0,0 +1,9 @@
package service
import "ems.agt/src/modules/network_element/model"
// 网元信息 服务层接口
type INeInfo interface {
// SelectNeInfoByNeTypeAndNeID 通过ne_type和ne_id查询网元信息
SelectNeInfoByNeTypeAndNeID(neType, neID string) model.NeInfo
}

22
src/modules/network_element/service/ne_info.impl.go Normal file
View File

@@ -0,0 +1,22 @@
package service
import (
"ems.agt/src/modules/network_element/model"
"ems.agt/src/modules/network_element/repository"
)
// 实例化服务层 NeInfoImpl 结构体
var NewNeInfoImpl = &NeInfoImpl{
NeInfoRepository: repository.NewNeInfoImpl,
}
// 网元信息 服务层处理
type NeInfoImpl struct {
// 网元信息数据信息
NeInfoRepository repository.INeInfo
}
// SelectNeInfoByNeTypeAndNeID 通过ne_type和ne_id查询网元信息
func (r *NeInfoImpl) SelectNeInfoByNeTypeAndNeID(neType, neID string) model.NeInfo {
return r.NeInfoRepository.SelectNeInfoByNeTypeAndNeID(neType, neID)
}

37
src/modules/network_element/service/udm_auth.go Normal file
View File

@@ -0,0 +1,37 @@
package service
import "ems.agt/src/modules/network_element/model"
// UDM鉴权信息 服务层接口
type IUDMAuth interface {
// Save UDM鉴权用户-获取redis全部保存数据库
Save(neID string) int64
// Page UDM鉴权用户-分页查询数据库
Page(query map[string]any) map[string]any
// List UDM鉴权用户-查询数据库
List(authUser model.UDMAuth) []model.UDMAuth
// Insert UDM鉴权用户-新增单个
// imsi长度15，ki长度32，opc长度0或者32
Insert(neID string, authUser model.UDMAuth) int64
// Insert UDM鉴权用户-批量添加
Inserts(neID string, authUser model.UDMAuth, num string) int64
// InsertCSV UDM鉴权用户-批量添加
InsertCSV(neID string, data []map[string]string) int64
// InsertTxt UDM鉴权用户-批量添加
InsertTxt(neID string, data [][]string) int64
// Insert UDM鉴权用户-修改更新
Update(neID string, authUser model.UDMAuth) int64
// Insert UDM鉴权用户-删除单个
Delete(neID, imsi string) int64
// Insert UDM鉴权用户-删除范围
Deletes(neID, imsi, num string) int64
}

177
src/modules/network_element/service/udm_auth.impl.go Normal file
View File

@@ -0,0 +1,177 @@
package service
import (
"fmt"
"strings"
"ems.agt/src/framework/redis"
"ems.agt/src/modules/network_element/model"
"ems.agt/src/modules/network_element/repository"
)
// 实例化服务层 UDMAuthImpl 结构体
var NewUDMAuthImpl = &UDMAuthImpl{
udmAuthRepository: repository.NewUDMAuthImpl,
}
// UDM鉴权信息 服务层处理
type UDMAuthImpl struct {
// UDM鉴权信息数据信息
udmAuthRepository repository.IUDMAuth
}
// authDataByRedis UDM鉴权用户
func (r *UDMAuthImpl) authDataByRedis(imsi, neID string) []model.UDMAuth {
arr := []model.UDMAuth{}
ausfArr, err := redis.GetKeys("udmuser", fmt.Sprintf("ausf:%s", imsi))
if err != nil {
return arr
}
for _, key := range ausfArr {
m, err := redis.GetHash("udmuser", key)
if err != nil {
continue
}
// 跳过-号数据
imsi := key[5:]
if strings.Contains(imsi, "-") {
continue
}
status := "0"
if _, ok := m["auth_success"]; ok {
status = "1"
}
amf := ""
if v, ok := m["amf"]; ok {
amf = strings.Replace(v, "\r\n", "", 1)
}
a := model.UDMAuth{
Imsi: imsi,
Amf: amf,
Status: status,
Ki: m["ki"],
AlgoIndex: m["algo"],
Opc: m["opc"],
NeID: neID,
}
arr = append(arr, a)
}
return arr
}
// Save UDM鉴权用户-获取redis全部保存数据库
func (r *UDMAuthImpl) Save(neID string) int64 {
var num int64 = 0
authArr := r.authDataByRedis("*", neID)
// 有数据才清空
if len(authArr) == 0 {
return num
}
go r.udmAuthRepository.ClearAndInsert(neID, authArr)
return int64(len(authArr))
}
// Page UDM鉴权用户-分页查询数据库
func (r *UDMAuthImpl) Page(query map[string]any) map[string]any {
return r.udmAuthRepository.SelectPage(query)
}
// List UDM鉴权用户-查询数据库
func (r *UDMAuthImpl) List(authUser model.UDMAuth) []model.UDMAuth {
return r.udmAuthRepository.SelectList(authUser)
}
// Insert UDM鉴权用户-新增单个
// imsi长度15，ki长度32，opc长度0或者32
func (r *UDMAuthImpl) Insert(neID string, authUser model.UDMAuth) int64 {
authArr := r.authDataByRedis(authUser.Imsi, neID)
if len(authArr) > 0 {
r.udmAuthRepository.Delete(neID, authUser.Imsi)
r.udmAuthRepository.Inserts(authArr)
}
return 0
}
// Insert UDM鉴权用户-批量添加
func (r *UDMAuthImpl) Inserts(neID string, authUser model.UDMAuth, num string) int64 {
startIMSI := authUser.Imsi
startIMSI = startIMSI[:len(startIMSI)-len(num)] + "*"
// keys udm-sd:4600001000004*
authArr := r.authDataByRedis(startIMSI, neID)
if len(authArr) > 0 {
for _, v := range authArr {
r.udmAuthRepository.Delete(neID, v.Imsi)
}
return r.udmAuthRepository.Inserts(authArr)
}
return 0
}
// InsertCSV UDM鉴权用户-批量添加
func (r *UDMAuthImpl) 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.udmAuthRepository.DeletePrefixImsi(neID, prefix)
subArr := r.authDataByRedis(prefix+"*", neID)
if len(subArr) > 0 {
num += r.udmAuthRepository.Inserts(subArr)
}
}
return num
}
// InsertTxt UDM鉴权用户-批量添加
func (r *UDMAuthImpl) 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.udmAuthRepository.DeletePrefixImsi(neID, prefix)
subArr := r.authDataByRedis(prefix+"*", neID)
if len(subArr) > 0 {
num += r.udmAuthRepository.Inserts(subArr)
}
}
return num
}
// Insert UDM鉴权用户-修改更新
func (r *UDMAuthImpl) Update(neID string, authUser model.UDMAuth) int64 {
authArr := r.authDataByRedis(authUser.Imsi, neID)
if len(authArr) > 0 {
r.udmAuthRepository.Delete(neID, authUser.Imsi)
return r.udmAuthRepository.Inserts(authArr)
}
return 0
}
// Insert UDM鉴权用户-删除单个
func (r *UDMAuthImpl) Delete(neID, imsi string) int64 {
return r.udmAuthRepository.Delete(neID, imsi)
}
// Insert UDM鉴权用户-删除范围
func (r *UDMAuthImpl) Deletes(neID, imsi, num string) int64 {
prefix := imsi[:len(imsi)-len(num)]
// keys udm-sd:4600001000004*
authArr := r.authDataByRedis(prefix+"*", neID)
if len(authArr) > 0 {
r.udmAuthRepository.DeletePrefixImsi(neID, prefix)
return r.udmAuthRepository.Inserts(authArr)
}
return 0
}

37
src/modules/network_element/service/udm_sub.go Normal file
View File

@@ -0,0 +1,37 @@
package service
import "ems.agt/src/modules/network_element/model"
// UDM签约用户信息 服务层接口
type IUDMSub interface {
// Save UDM签约用户-获取redis全部保存数据库
Save(neID string) int64
// Page UDM签约用户-分页查询数据库
Page(query map[string]any) map[string]any
// List UDM签约用户-查询数据库
List(subUser model.UDMSub) []model.UDMSub
// Insert UDM签约用户-新增单个
// imsi长度15，ki长度32，opc长度0或者32
Insert(neID string, subUser model.UDMSub) int64
// Insert UDM签约用户-批量添加
Inserts(neID string, subUser model.UDMSub, num string) int64
// InsertCSV UDM签约用户-批量添加
InsertCSV(neID string, data []map[string]string) int64
// InsertTxt UDM签约用户-批量添加
InsertTxt(neID string, data [][]string) int64
// Insert UDM签约用户-修改更新
Update(neID string, subUser model.UDMSub) int64
// Insert UDM签约用户-删除单个
Delete(neID, imsi string) int64
// Insert UDM签约用户-删除范围
Deletes(neID, imsi, num string) int64
}

191
src/modules/network_element/service/udm_sub.impl.go Normal file
View File

@@ -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
}