rts-sim-testing-service/grpcproto/simulation_server.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) 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
}