From 9c8b4d8dc8e2aa07982513c200a37fded076697e Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Sat, 16 Sep 2023 17:54:42 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20UDM=E8=AF=BB=E5=8F=96=E6=95=B0=E6=8D=AEr?=
 =?UTF-8?q?edis=E6=8C=87=E5=AE=9A=E5=BA=93?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../udm_user/service/service_redis_data.go    | 20 +++++++++++++++----
 1 file changed, 16 insertions(+), 4 deletions(-)

diff --git a/features/udm_user/service/service_redis_data.go b/features/udm_user/service/service_redis_data.go
index 8ccc8437..77fac34b 100644
--- a/features/udm_user/service/service_redis_data.go
+++ b/features/udm_user/service/service_redis_data.go
@@ -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,