fix: UDM读取数据redis指定库

This commit is contained in:
TsMask
2023-09-16 17:54:42 +08:00
parent 03c73db8c2
commit 9c8b4d8dc8

View File

@@ -10,10 +10,16 @@ import (
// phoneImsiList 获取所有imsi
func phoneImsiList() map[string]string {
phoneKeys := redis.GetKeys("1*********")
phoneAndImsiArr := make(map[string]string, 0)
phoneKeys, err := redis.GetKeys("udmuser", "1*********")
if err != nil {
return phoneAndImsiArr
}
for _, phone := range phoneKeys {
imsi := redis.Get(phone)
imsi, err := redis.Get("udmuser", phone)
if err != nil {
continue
}
phoneAndImsiArr[phone] = imsi
}
return phoneAndImsiArr
@@ -25,7 +31,10 @@ func redisUdmAuthUserList() []model.UdmAuthUser {
user := []model.UdmAuthUser{}
for phone, imsi := range phoneAndImsiArr {
key := fmt.Sprintf("ausf:%s", imsi)
m := redis.GetHash(key)
m, err := redis.GetHash("udmuser", key)
if err != nil {
continue
}
status := "0"
if _, ok := m["auth_success"]; ok {
@@ -51,7 +60,10 @@ func redisUdmSubUserList() []model.UdmSubUser {
user := []model.UdmSubUser{}
for phone, imsi := range phoneAndImsiArr {
key := fmt.Sprintf("udm-sd:%s", imsi)
m := redis.GetHash(key)
m, err := redis.GetHash("udmuser", key)
if err != nil {
continue
}
a := model.UdmSubUser{
Imsi: imsi,