ff67c84f18
All checks were successful
local-test分支打包构建docker并发布运行 / Docker-Build (push) Successful in 4m7s
79 lines
2.3 KiB
Go
79 lines
2.3 KiB
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,
|
|
})
|
|
}
|
|
}
|