rts-sim-testing-service/third_party/semi_physical_train/semi_physical_train.go

package semi_physical_train

import (
	"fmt"

	"joylink.club/bj-rtsts-server/config"
	"joylink.club/bj-rtsts-server/third_party/message"
	"joylink.club/bj-rtsts-server/third_party/udp"
)

// 半实物仿真列车通信接口
type SemiPhysicalTrain interface {
	// 启动半实物仿真消息处理
	Start(manager SemiPhysicalMessageManager)
	// 停止半实物仿真消息处理
	Stop()
	// 发送列车控制消息
	SendTrainControlMessage(info *message.DynamicsTrainInfo)
}

type SemiPhysicalMessageManager interface {
	// 处理半实物仿真列车控制消息
	HandleSemiPhysicalTrainControlMsg(b []byte)
}

type semiPhysicalTrainImpl struct {
	trainControlUdpServer   udp.UdpServer
	trainSpeedInfoUdpClient udp.UdpClient

	manager SemiPhysicalMessageManager
}

func (s *semiPhysicalTrainImpl) handleTrainControlMsg(b []byte) {
	handler := s.manager
	if handler != nil {
		handler.HandleSemiPhysicalTrainControlMsg(b)
	}
}

func (s *semiPhysicalTrainImpl) Start(manager SemiPhysicalMessageManager) {
	s.manager = manager
}

func (s *semiPhysicalTrainImpl) Stop() {
	s.manager = nil
}

func (s *semiPhysicalTrainImpl) SendTrainControlMessage(info *message.DynamicsTrainInfo) {
	sendMsg := &message.TrainSpeedMsg{}
	sendMsg.DynamicsDecode(info)
	s.trainSpeedInfoUdpClient.Send(sendMsg.Encode())
}

func newSemiPhysicalTrain() SemiPhysicalTrain {
	s := &semiPhysicalTrainImpl{
		trainSpeedInfoUdpClient: udp.NewClient(fmt.Sprintf("%v:%v", config.Config.Vobc.Ip, config.Config.Vobc.RemotePort)),
	}
	s.trainControlUdpServer = udp.NewServer(fmt.Sprintf(":%d", config.Config.Vobc.LocalPort), s.handleTrainControlMsg)
	s.trainControlUdpServer.Listen()
	return s
}

var _default SemiPhysicalTrain

func Default() SemiPhysicalTrain {
	if !config.Config.Vobc.Open {
		panic("半实物仿真接口模块未开启")
	}
	return _default
}

func Init() {
	if !config.Config.Vobc.Open {
		return
	}
	_default = newSemiPhysicalTrain()
}