60 lines
1.4 KiB
Go
60 lines
1.4 KiB
Go
package apiproto
|
|
|
|
import (
|
|
"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}-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
|
|
}
|
|
simulation := simulation.FindSimulation(simId)
|
|
if simulation == nil {
|
|
return nil
|
|
}
|
|
data, err := proto.Marshal(simulation.GetAllState())
|
|
if err != nil {
|
|
panic(dto.ErrorDto{Code: dto.DataOperationError, Message: err.Error()})
|
|
}
|
|
return data
|
|
}
|
|
|
|
// 定时发送数据
|
|
func (t *SimulationServer) onTick() []TopicMsg {
|
|
simArr := simulation.GetSimulationArr()
|
|
msgArr := make([]TopicMsg, len(simArr))
|
|
for i, v := range simArr {
|
|
channelName := handlerChannelName(v.SimulationId, t.getChannelName())
|
|
b, err := proto.Marshal(v.GetAllState())
|
|
if err != nil {
|
|
panic(dto.ErrorDto{Code: dto.DataOperationError, Message: err.Error()})
|
|
}
|
|
msgArr[i] = TopicMsg{channalName: channelName, data: b}
|
|
}
|
|
return msgArr
|
|
}
|
|
|
|
// 处理订阅通道名称
|
|
func handlerChannelName(sid string, format string) string {
|
|
return strings.Replace(format, "{sid}", sid, 1)
|
|
}
|