Files
be.ems/features/udm_user/service/service_redis_data.go
2023-09-08 11:15:11 +08:00

84 lines
1.7 KiB
Go

package service
import (
"fmt"
"strings"
"ems.agt/features/udm_user/model"
"ems.agt/lib/core/redis"
)
// phoneImsiList 获取所有imsi
func 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
}
// redisUdmAuthUserList UDM鉴权用户
func redisUdmAuthUserList() []model.UdmAuthUser {
phoneAndImsiArr := phoneImsiList()
user := []model.UdmAuthUser{}
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.UdmAuthUser{
Msisdn: phone,
Imsi: imsi,
Amf: m["amf"],
Status: status,
Ki: m["ki"],
AlgoIndex: m["algo"],
Opc: m["opc"],
}
user = append(user, a)
}
return user
}
// redisUdmSubUserList UDM签约用户
func redisUdmSubUserList() []model.UdmSubUser {
phoneAndImsiArr := phoneImsiList()
user := []model.UdmSubUser{}
for phone, imsi := range phoneAndImsiArr {
key := fmt.Sprintf("udm-sd:%s", imsi)
m := redis.GetHash(key)
a := model.UdmSubUser{
Imsi: imsi,
Msisdn: m["gpsi"],
SubData: m["sm-dat"],
}
if a.Msisdn == "" {
a.Msisdn = phone
}
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)
}
return user
}