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