52 lines
1.6 KiB
Go
52 lines
1.6 KiB
Go
package message_server
|
|
|
|
import (
|
|
"time"
|
|
|
|
"joylink.club/bj-rtsts-server/message_server/ms_api"
|
|
"joylink.club/bj-rtsts-server/mqtt"
|
|
"joylink.club/bj-rtsts-server/ts/protos/graphicData"
|
|
"joylink.club/bj-rtsts-server/ts/protos/state"
|
|
"joylink.club/bj-rtsts-server/ts/simulation/wayside/memory"
|
|
"joylink.club/ecs"
|
|
"joylink.club/rtsssimulation/component"
|
|
"joylink.club/rtsssimulation/entity"
|
|
)
|
|
|
|
func NewRccMs(vs *memory.VerifySimulation, mapId int32) ms_api.MsgTask {
|
|
return ms_api.NewScheduleTask("继电器状态", func() error {
|
|
relayStates, err := collectRelayState(vs.World, mapId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
ststes := &state.PushedDevicesStatus{
|
|
All: true,
|
|
AllStatus: &state.AllDevicesStatus{
|
|
ReplyState: relayStates,
|
|
},
|
|
}
|
|
mqtt.GetMsgClient().PubRCCState(vs.SimulationId, mapId, ststes)
|
|
return nil
|
|
}, 200*time.Millisecond)
|
|
}
|
|
|
|
// 获取仿真地图的继电器状态，前端推送
|
|
func collectRelayState(world ecs.World, mapId int32) ([]*state.ReplyState, error) {
|
|
// 获取本地图下的继电器信息
|
|
uidMap := memory.QueryMapUidMapByType(mapId, &graphicData.Relay{})
|
|
var replyStateArr []*state.ReplyState
|
|
for _, u := range uidMap {
|
|
entry, ok := entity.GetEntityByUid(world, u.Uid)
|
|
if !ok {
|
|
// 暂时注释，很多继电器都没初始化
|
|
//return nil, fmt.Errorf("继电器实体不存在: World id=%d, uid=%s", r.vs.World.Id(), u.Uid)
|
|
continue
|
|
}
|
|
if entry.HasComponent(component.RelayTag) {
|
|
bit := component.BitStateType.Get(entry)
|
|
replyStateArr = append(replyStateArr, &state.ReplyState{Id: u.CommonId, Xh: bit.Val})
|
|
}
|
|
}
|
|
return replyStateArr, nil
|
|
}
|