92 lines
2.1 KiB
Go
92 lines
2.1 KiB
Go
package service
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"be.ems/src/framework/vo/result"
|
|
"be.ems/src/modules/ws/model"
|
|
)
|
|
|
|
// 订阅组指定编号为支持服务器向客户端主动推送数据
|
|
const (
|
|
// 组号-其他
|
|
GROUP_OTHER = "0"
|
|
// 组号-跟踪任务网元数据变更 2_traceId
|
|
GROUP_TRACE_NE = "2_"
|
|
// 组号-信令跟踪Packet 4_taskNo
|
|
GROUP_TRACE_PACKET = "4_"
|
|
// 组号-指标通用 10_neType_neId
|
|
GROUP_KPI = "10_"
|
|
// 组号-指标UPF 12_neId
|
|
GROUP_KPI_UPF = "12_"
|
|
// 组号-自定义KPI指标 20_neType_neId
|
|
GROUP_KPI_C = "20_"
|
|
// 组号-IMS_CDR会话事件 1005_neId
|
|
GROUP_IMS_CDR = "1005_"
|
|
// 组号-SMF_CDR会话事件 1006_neId
|
|
GROUP_SMF_CDR = "1006_"
|
|
// 组号-SMSC_CDR会话事件 1007_neId
|
|
GROUP_SMSC_CDR = "1007_"
|
|
// 组号-SGWC_CDR会话事件 1008_neId
|
|
GROUP_SGWC_CDR = "1008_"
|
|
// 组号-AMF_UE会话事件
|
|
GROUP_AMF_UE = "1010"
|
|
// 组号-MME_UE会话事件 1011_neId
|
|
GROUP_MME_UE = "1011_"
|
|
)
|
|
|
|
// 实例化服务层 WSSend 结构体
|
|
var NewWSSend = &WSSend{}
|
|
|
|
// WSSend WebSocket消息发送处理 服务层处理
|
|
type WSSend struct{}
|
|
|
|
// ByClientID 给已知客户端发消息
|
|
func (s *WSSend) ByClientID(clientID string, data any) error {
|
|
v, ok := wsClients.Load(clientID)
|
|
if !ok {
|
|
return fmt.Errorf("no fount client ID: %s", clientID)
|
|
}
|
|
|
|
dataByte, err := json.Marshal(result.OkData(data))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
client := v.(*model.WSClient)
|
|
if len(client.MsgChan) > 90 {
|
|
NewWS.ClientClose(client.ID)
|
|
return fmt.Errorf("msg chan over 90 will close client ID: %s", clientID)
|
|
}
|
|
client.MsgChan <- dataByte
|
|
return nil
|
|
}
|
|
|
|
// ByGroupID 给订阅组的客户端发送消息
|
|
func (s *WSSend) ByGroupID(groupID string, data any) error {
|
|
clientIds, ok := wsGroup.Load(groupID)
|
|
if !ok {
|
|
return fmt.Errorf("no fount Group ID: %s", groupID)
|
|
}
|
|
|
|
// 检查组内是否有客户端
|
|
ids := clientIds.(*[]string)
|
|
if len(*ids) == 0 {
|
|
return fmt.Errorf("no members in the group")
|
|
}
|
|
|
|
// 遍历给客户端发消息
|
|
for _, clientId := range *ids {
|
|
err := s.ByClientID(clientId, map[string]any{
|
|
"groupId": groupID,
|
|
"data": data,
|
|
})
|
|
if err != nil {
|
|
continue
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|