feat: UE事件AMF/MME区分网元查询

src/modules/ws/processor/ue_connect.go

@@ -8,6 +8,7 @@ import (
 	"be.ems/src/framework/vo/result"
 	neDataModel "be.ems/src/modules/network_data/model"
 	neDataService "be.ems/src/modules/network_data/service"
+	neService "be.ems/src/modules/network_element/service"
 )
 
 // GetUEConnectByAMF 获取UE会话事件-AMF
@@ -20,6 +21,13 @@ func GetUEConnectByAMF(requestID string, data any) ([]byte, error) {
 		return nil, fmt.Errorf("query data structure error")
 	}
 
+	// 查询网元获取IP
+	neInfo := neService.NewNeInfo.SelectNeInfoByNeTypeAndNeID("AMF", query.NeID)
+	if neInfo.NeId != query.NeID || neInfo.IP == "" {
+		return nil, fmt.Errorf("not fount neId info")
+	}
+	query.RmUID = neInfo.RmUID
+
 	rows, total := neDataService.NewUEEventAMF.SelectPage(query)
 	resultByte, err := json.Marshal(result.Ok(map[string]any{
 		"requestId": requestID,
@@ -38,6 +46,13 @@ func GetUEConnectByMME(requestID string, data any) ([]byte, error) {
 		return nil, fmt.Errorf("query data structure error")
 	}
 
+	// 查询网元获取IP
+	neInfo := neService.NewNeInfo.SelectNeInfoByNeTypeAndNeID("MME", query.NeID)
+	if neInfo.NeId != query.NeID || neInfo.IP == "" {
+		return nil, fmt.Errorf("not fount neId info")
+	}
+	query.RmUID = neInfo.RmUID
+
 	rows, total := neDataService.NewUEEventMME.SelectPage(query)
 	resultByte, err := json.Marshal(result.Ok(map[string]any{
 		"requestId": requestID,

src/modules/ws/service/ws_send.go

@@ -30,10 +30,10 @@ const (
 	GROUP_SMSC_CDR = "1007_"
 	// 组号-SGWC_CDR会话事件 1008_neId
 	GROUP_SGWC_CDR = "1008_"
-	// 组号-AMF_UE会话事件
+	// 组号-AMF_UE会话事件 1010_neId
 	GROUP_AMF_UE = "1010"
 	// 组号-MME_UE会话事件 1011_neId
-	GROUP_MME_UE = "1011_"
+	GROUP_MME_UE = "1011"
 )
 
 // 实例化服务层 WSSend 结构体