rts-sim-testing-service/message_server/simulation_state_ms.go

package message_server

import (
	"fmt"
	"joylink.club/bj-rtsts-server/dto/state_proto"

	"joylink.club/bj-rtsts-server/message_server/ms_api"
	"joylink.club/bj-rtsts-server/mqtt"
	"joylink.club/bj-rtsts-server/ts/simulation/wayside/memory"
	"joylink.club/ecs"
)

func NewStateMs(vs *memory.VerifySimulation) ms_api.MsgTask {
	return ms_api.NewMonitorTask(fmt.Sprintf("仿真[%s]状态", vs.SimulationId), func() {
		ecs.WorldStateChangeEvent.Subscribe(vs.World, func(_ ecs.World, e ecs.WorldStateChange) {
			s := &state_proto.SimulationStatus{SimulationId: vs.SimulationId}
			switch e.NewState {
			case ecs.WorldClosed:
				s.State = state_proto.SimulationStatus_DESTROY
			case ecs.WorldError:
				s.State = state_proto.SimulationStatus_ERROR
			case ecs.WorldPause:
				s.State = state_proto.SimulationStatus_PAUSE
			default:
				return
			}
			mqtt.GetMsgClient().PubSimulationState(vs.SimulationId, s)
		})
	})
}