OMC/be.ems
1
0
Fork 0
Code Issues Releases 3 Activity

feat: ws支持upf_tf类型信息获取UPF-总流量数

Browse Source
This commit is contained in:
TsMask
2024-01-29 15:48:01 +08:00
parent f1b17e00ab
commit 2223828b2b
2 changed files with 59 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
src/modules/ws/service/ws_receive.impl.go
View File

@@ -18,33 +18,28 @@ func (s *WSReceiveImpl) Receive(client *model.WSClient, reqMsg model.WSRequest)
if reqMsg.RequestID == "" {
return fmt.Errorf("message requestId is required")
}
var resByte []byte
var err error
switch reqMsg.Type {
case "ps":
res, err := processor.GetProcessData(reqMsg.RequestID, reqMsg.Data)
if err != nil {
return err
}
client.MsgChan <- res
resByte, err = processor.GetProcessData(reqMsg.RequestID, reqMsg.Data)
case "net":
res, err := processor.GetNetConnections(reqMsg.RequestID, reqMsg.Data)
if err != nil {
return err
}
client.MsgChan <- res
resByte, err = processor.GetNetConnections(reqMsg.RequestID, reqMsg.Data)
case "cdr":
res, err := processor.GetCDRConnect(reqMsg.RequestID, reqMsg.Data)
if err != nil {
return err
}
client.MsgChan <- res
resByte, err = processor.GetCDRConnect(reqMsg.RequestID, reqMsg.Data)
case "ue":
res, err := processor.GetUEConnect(reqMsg.RequestID, reqMsg.Data)
if err != nil {
return err
}
client.MsgChan <- res
resByte, err = processor.GetUEConnect(reqMsg.RequestID, reqMsg.Data)
case "upf_tf":
resByte, err = processor.GetUPFTotalFlow(reqMsg.RequestID, reqMsg.Data)
default:
return fmt.Errorf("message type not supported")
}
if err != nil {
return err
}
client.MsgChan <- resByte
return nil
}