57 lines
1.5 KiB
Go
57 lines
1.5 KiB
Go
package service
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"ems.agt/src/framework/logger"
|
|
"ems.agt/src/framework/vo/result"
|
|
"ems.agt/src/modules/ws/model"
|
|
"ems.agt/src/modules/ws/processor"
|
|
)
|
|
|
|
// 实例化服务层 WSReceiveImpl 结构体
|
|
var NewWSReceiveImpl = &WSReceiveImpl{}
|
|
|
|
// WSReceiveImpl WebSocket消息接收处理 服务层处理
|
|
type WSReceiveImpl struct{}
|
|
|
|
// Receive 接收处理
|
|
func (s *WSReceiveImpl) Receive(client *model.WSClient, reqMsg model.WSRequest) {
|
|
if reqMsg.RequestID == "" {
|
|
msg := "message requestId is required"
|
|
logger.Warnf("ws ReceiveMessage UID %s err: %s", client.BindUid, msg)
|
|
msgByte, _ := json.Marshal(result.ErrMsg(msg))
|
|
client.MsgChan <- msgByte
|
|
return
|
|
}
|
|
|
|
var resByte []byte
|
|
var err error
|
|
|
|
switch reqMsg.Type {
|
|
case "ps":
|
|
resByte, err = processor.GetProcessData(reqMsg.RequestID, reqMsg.Data)
|
|
case "net":
|
|
resByte, err = processor.GetNetConnections(reqMsg.RequestID, reqMsg.Data)
|
|
case "cdr":
|
|
resByte, err = processor.GetCDRConnect(reqMsg.RequestID, reqMsg.Data)
|
|
case "ue":
|
|
resByte, err = processor.GetUEConnect(reqMsg.RequestID, reqMsg.Data)
|
|
case "upf_tf":
|
|
resByte, err = processor.GetUPFTotalFlow(reqMsg.RequestID, reqMsg.Data)
|
|
case "ne_state":
|
|
resByte, err = processor.GetNeState(reqMsg.RequestID, reqMsg.Data)
|
|
default:
|
|
err = fmt.Errorf("message type not supported")
|
|
}
|
|
|
|
if err != nil {
|
|
logger.Warnf("ws ReceiveMessage UID %s err: %s", client.BindUid, err.Error())
|
|
msgByte, _ := json.Marshal(result.ErrMsg(err.Error()))
|
|
client.MsgChan <- msgByte
|
|
return
|
|
}
|
|
client.MsgChan <- resByte
|
|
}
|