feat: WS模块支持mme用户时间查询

This commit is contained in:
TsMask
2024-06-15 17:24:27 +08:00
parent ff4511f236
commit f3bc87a5f2
3 changed files with 23 additions and 1 deletions

View File

@@ -16,7 +16,7 @@ func GetUEConnectByAMF(requestID string, data any) ([]byte, error) {
var query neDataModel.UEEventAMFQuery
err := json.Unmarshal(msgByte, &query)
if err != nil {
logger.Warnf("ws processor GetUEConnect err: %s", err.Error())
logger.Warnf("ws processor GetUEConnectByAMF err: %s", err.Error())
return nil, fmt.Errorf("query data structure error")
}
@@ -27,3 +27,21 @@ func GetUEConnectByAMF(requestID string, data any) ([]byte, error) {
}))
return resultByte, err
}
// GetUEConnectByMME 获取UE会话事件-MME
func GetUEConnectByMME(requestID string, data any) ([]byte, error) {
msgByte, _ := json.Marshal(data)
var query neDataModel.UEEventMMEQuery
err := json.Unmarshal(msgByte, &query)
if err != nil {
logger.Warnf("ws processor GetUEConnectByMME err: %s", err.Error())
return nil, fmt.Errorf("query data structure error")
}
dataMap := neDataService.NewUEEventMMEImpl.SelectPage(query)
resultByte, err := json.Marshal(result.Ok(map[string]any{
"requestId": requestID,
"data": dataMap,
}))
return resultByte, err
}

View File

@@ -74,6 +74,8 @@ func (s *WSReceiveImpl) AsyncReceive(client *model.WSClient, reqMsg model.WSRequ
resByte, err = processor.GetCDRConnectBySMF(reqMsg.RequestID, reqMsg.Data)
case "amf_ue":
resByte, err = processor.GetUEConnectByAMF(reqMsg.RequestID, reqMsg.Data)
case "mme_ue":
resByte, err = processor.GetUEConnectByMME(reqMsg.RequestID, reqMsg.Data)
case "upf_tf":
resByte, err = processor.GetUPFTotalFlow(reqMsg.RequestID, reqMsg.Data)
case "ne_state":

View File

@@ -22,6 +22,8 @@ const (
GROUP_SMF_CDR = "1006"
// 组号-AMF_UE会话事件
GROUP_AMF_UE = "1010"
// 组号-MME_UE会话事件
GROUP_MME_UE = "1011"
)
// 实例化服务层 WSSendImpl 结构体