39 lines
866 B
Go
39 lines
866 B
Go
package service
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"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) error {
|
|
if reqMsg.RequestID == "" {
|
|
return fmt.Errorf("message requestId is required")
|
|
}
|
|
switch reqMsg.Type {
|
|
case "ps":
|
|
res, err := processor.GetProcessData(reqMsg.RequestID, reqMsg.Data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
client.MsgChan <- res
|
|
case "net":
|
|
res, err := processor.GetNetConnections(reqMsg.RequestID, reqMsg.Data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
client.MsgChan <- res
|
|
default:
|
|
return fmt.Errorf("message type not supported")
|
|
}
|
|
return nil
|
|
}
|