UDM 用户信息接口添加到路由

This commit is contained in:
TsMask
2023-09-07 19:15:25 +08:00
parent 36549c89d4
commit 7b243bf0dd
6 changed files with 328 additions and 0 deletions

View File

@@ -0,0 +1,90 @@
package service
import (
"fmt"
"strings"
"ems.agt/features/udm_user/model"
"ems.agt/lib/core/redis"
)
// 实例化服务层 ServiceUdmUser 结构体
var NewServiceUdmUser = &ServiceUdmUser{}
// ServiceUdmUser UDM 服务层处理
type ServiceUdmUser struct{}
// phoneImsiList 获取所有imsi
func (r *ServiceUdmUser) phoneImsiList() map[string]string {
phoneKeys := redis.GetKeys("1*********")
phoneAndImsiArr := make(map[string]string, 0)
for _, phone := range phoneKeys {
imsi := redis.Get(phone)
phoneAndImsiArr[phone] = imsi
}
return phoneAndImsiArr
}
// ReadAusfList UDM鉴权用户
func (r *ServiceUdmUser) ReadAusfList() []model.UdmAusfUser {
phoneAndImsiArr := r.phoneImsiList()
user := []model.UdmAusfUser{}
for phone, imsi := range phoneAndImsiArr {
key := fmt.Sprintf("ausf:%s", imsi)
m := redis.GetHash(key)
status := "0"
if _, ok := m["auth_success"]; ok {
status = "1"
}
a := model.UdmAusfUser{
Phone: phone,
Imsi: imsi,
Amf: m["amf"],
Status: status,
Ki: m["ki"],
AlgoIndex: m["algo"],
Opc: m["opc"],
}
user = append(user, a)
}
fmt.Println(len(user))
return user
}
// ReadSdList UDM签约用户
func (r *ServiceUdmUser) ReadSdList() []model.UdmSdUser {
phoneAndImsiArr := r.phoneImsiList()
user := []model.UdmSdUser{}
for phone, imsi := range phoneAndImsiArr {
key := fmt.Sprintf("udm-sd:%s", imsi)
m := redis.GetHash(key)
a := model.UdmSdUser{
Phone: phone,
Imsi: imsi,
Msisdn: m["gpsi"],
SubData: m["sm-dat"],
}
if v, ok := m["am-dat"]; ok {
arr := strings.Split(v, ",")
a.SubUeAMBRTemp = arr[0]
a.SubSNSSAITemp = arr[1]
a.Rai = arr[2]
a.ForbiddenAreasTemp = arr[3]
a.ServiceAreaRestrictTemp = arr[4]
a.CnType = arr[5]
}
if v, ok := m["eps-dat"]; ok {
arr := strings.Split(v, ",")
a.Eps = arr[0]
}
user = append(user, a)
}
fmt.Println(len(user))
return user
}