fix: UDM数据来源通过网元主机的Redis客户端进行连接

src/modules/network_data/service/udm_auth.go

@@ -8,6 +8,7 @@ import (
 	"be.ems/src/framework/redis"
 	"be.ems/src/modules/network_data/model"
 	"be.ems/src/modules/network_data/repository"
+	neService "be.ems/src/modules/network_element/service"
 )
 
 // 实例化服务层 UDMAuthUser 结构体
@@ -25,12 +26,25 @@ type UDMAuthUser struct {
 func (r *UDMAuthUser) dataByRedis(imsi, neId string) []model.UDMAuthUser {
 	arr := []model.UDMAuthUser{}
 	key := fmt.Sprintf("ausf:%s", imsi)
-	ausfArr, err := redis.GetKeys("udmuser", key)
+	source := fmt.Sprintf("UDM_%s", neId)
+
+	// 网元主机的Redis客户端
+	redisClient, err := neService.NewNeInfo.NeRunRedisClient("UDM", neId)
+	if err != nil {
+		return arr
+	}
+	defer func() {
+		redisClient.Close()
+		redis.ConnectPush(source, nil)
+	}()
+	redis.ConnectPush(source, redisClient.Client)
+
+	ausfArr, err := redis.GetKeys(source, key)
 	if err != nil {
 		return arr
 	}
 	for _, key := range ausfArr {
-		m, err := redis.GetHash("udmuser", key)
+		m, err := redis.GetHash(source, key)
 		if err != nil {
 			continue
 		}