diff --git a/src/modules/network_data/service/udm_auth.go b/src/modules/network_data/service/udm_auth.go index 62812a35..481ed649 100644 --- a/src/modules/network_data/service/udm_auth.go +++ b/src/modules/network_data/service/udm_auth.go @@ -49,13 +49,13 @@ func (r *UDMAuthUser) dataByRedis(imsi, neId string) []model.UDMAuthUser { } for k, m := range mkv { - if k == "-" { + if len(k) != 20 { continue } // 跳过-号数据 ausf:360000100000130 - imsi := key[5:] - if strings.Contains(imsi, "-") { + imsi, hasPrefix := strings.CutPrefix(k, "ausf:") + if strings.Contains(imsi, "-") || !hasPrefix { continue } diff --git a/src/modules/network_data/service/udm_sub.go b/src/modules/network_data/service/udm_sub.go index 894dae73..de4e215e 100644 --- a/src/modules/network_data/service/udm_sub.go +++ b/src/modules/network_data/service/udm_sub.go @@ -50,18 +50,18 @@ func (r *UDMSubUser) dataByRedis(imsi, neId string) []model.UDMSubUser { } for k, m := range mkv { - if k == "-" { + if len(k) != 22 { continue } // 跳过-号数据 udm-sd:360000100000130 - imsi := k[7:] - if strings.Contains(imsi, "-") { + imsi, hasPrefix := strings.CutPrefix(k, "udm-sd:") + if strings.Contains(imsi, "-") || !hasPrefix { continue } a := model.UDMSubUser{ - IMSI: key[7:], // udm-sd:360000100000130 + IMSI: imsi, // udm-sd:360000100000130 MSISDN: m["gpsi"], // 8612300000130 NeId: neId, SmfSel: m["smf-sel"], // def_snssai