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

package third_party

import (
	"joylink.club/bj-rtsts-server/third_party/train_pc_sim"
	"log/slog"

	"joylink.club/bj-rtsts-server/dto/state_proto"
	"joylink.club/bj-rtsts-server/third_party/dynamics"
	"joylink.club/bj-rtsts-server/third_party/semi_physical_train"
	"joylink.club/bj-rtsts-server/third_party/tpapi"
)

var tpapiService []tpapi.ThirdPartyApiService

func Init() {
	tpapiService = append(
		tpapiService, dynamics.Default(),
		semi_physical_train.Default(),
		train_pc_sim.Default(),
	)
}

func GetAllServices() []tpapi.ThirdPartyApiService {
	return tpapiService
}

func convertServiceName(name string) state_proto.SimulationThirdPartyApiService_Type {
	switch name {
	case dynamics.Name:
		return state_proto.SimulationThirdPartyApiService_Dynamics
	case semi_physical_train.Name:
		return state_proto.SimulationThirdPartyApiService_SemiPhysicalTrain
	case train_pc_sim.Name:
		return state_proto.SimulationThirdPartyApiService_Train_pc_sim
	default:
		return state_proto.SimulationThirdPartyApiService_Undefined
	}
}

func GetRunningServiceStates() *state_proto.SimulationThirdPartyApiService {
	ss := &state_proto.SimulationThirdPartyApiService{}
	for _, tpas := range tpapiService {
		if tpas.TrueService() {
			collectServiceState(ss, tpas)
		} else {
			trueServices := tpas.FindAppendApiService()
			if trueServices != nil && len(trueServices) > 0 {
				for _, trueService := range trueServices {
					collectServiceState(ss, trueService)
				}
			}

		}

	}
	return ss
}
func collectServiceState(ss *state_proto.SimulationThirdPartyApiService, service tpapi.ThirdPartyApiService) {
	t := convertServiceName(service.Name())
	if t == state_proto.SimulationThirdPartyApiService_Undefined {
		slog.Error("未知的第三方接口服务类型", "name", service.Name())
		return
	}
	switch service.State() {
	case tpapi.ThirdPartyState_Normal:
		ss.States = append(ss.States, &state_proto.SimulationThirdPartyApiServiceState{
			Type:        t,
			ServiceName: service.ServiceDesc(),
			State:       state_proto.SimulationThirdPartyApiService_Normal,
		})
	case tpapi.ThirdPartyState_Broken:
		ss.States = append(ss.States, &state_proto.SimulationThirdPartyApiServiceState{
			Type:        t,
			ServiceName: service.ServiceDesc(),
			State:       state_proto.SimulationThirdPartyApiService_Error,
		})
	}
}