72 lines
1.7 KiB
Go
72 lines
1.7 KiB
Go
package apiproto
|
|
|
|
import (
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"google.golang.org/protobuf/proto"
|
|
"joylink.club/bj-rtsts-server/ats/verify/simulation"
|
|
"joylink.club/bj-rtsts-server/dto"
|
|
)
|
|
|
|
type SimulationServer struct{}
|
|
|
|
// 返回通道格式
|
|
func (t *SimulationServer) getChannelName() string {
|
|
return "simulation-{sid}_{mid}-devices-status"
|
|
}
|
|
|
|
// 消息运行间隔
|
|
func (t *SimulationServer) getInterval() time.Duration {
|
|
return 200 * time.Millisecond
|
|
}
|
|
|
|
// 返回所有数据
|
|
func (t *SimulationServer) allMsgData(params map[string]string) []byte {
|
|
simId := params["SID"]
|
|
if simId == "" {
|
|
return nil
|
|
}
|
|
mId := params["MID"]
|
|
if mId == "" {
|
|
return nil
|
|
}
|
|
simulation := simulation.FindSimulation(simId)
|
|
if simulation == nil {
|
|
return nil
|
|
}
|
|
mapId, _ := strconv.Atoi(mId)
|
|
data, err := proto.Marshal(simulation.GetAllState(int32(mapId)))
|
|
if err != nil {
|
|
panic(dto.ErrorDto{Code: dto.DataOperationError, Message: err.Error()})
|
|
}
|
|
return data
|
|
}
|
|
|
|
// 定时发送数据
|
|
func (t *SimulationServer) onTick() []TopicMsg {
|
|
simArr := simulation.GetSimulationArr()
|
|
var msgArr []TopicMsg
|
|
for _, v := range simArr {
|
|
for _, mapId := range v.MapIds {
|
|
idStr := strconv.Itoa(int(mapId))
|
|
channelName := handlerChannelName(v.SimulationId, idStr, t.getChannelName())
|
|
b, err := proto.Marshal(v.GetAllState(mapId))
|
|
if err != nil {
|
|
panic(dto.ErrorDto{Code: dto.DataOperationError, Message: err.Error()})
|
|
}
|
|
msgArr = append(msgArr, TopicMsg{channalName: channelName, data: b})
|
|
}
|
|
}
|
|
return msgArr
|
|
}
|
|
|
|
// 处理订阅通道名称
|
|
func handlerChannelName(sid, mapId string, format string) string {
|
|
var channelName string
|
|
channelName = strings.Replace(format, "{sid}", sid, 1)
|
|
channelName = strings.Replace(channelName, "{mid}", mapId, 1)
|
|
return channelName
|
|
}
|