46 lines
1.4 KiB
Go
46 lines
1.4 KiB
Go
package fi
|
||
|
||
import (
|
||
"fmt"
|
||
"joylink.club/ecs"
|
||
"joylink.club/rtsssimulation/component"
|
||
"joylink.club/rtsssimulation/component/component_proto"
|
||
"joylink.club/rtsssimulation/entity"
|
||
)
|
||
|
||
// UpdateSignalLightFault 设置信号机灯故障
|
||
//
|
||
// light : 灯泡
|
||
// faultType ：故障类型
|
||
// set : true-设置故障，false-取消故障
|
||
func UpdateSignalLightFault(w ecs.World, signalId string, light component_proto.Light_Color, faultType component_proto.Signal_Fault, set bool) error {
|
||
r := <-ecs.Request[ecs.EmptyType](w, func() ecs.Result[ecs.EmptyType] {
|
||
wd := entity.GetWorldData(w)
|
||
signalEntry, ok := wd.EntityMap[signalId]
|
||
if ok {
|
||
switch faultType {
|
||
case component_proto.Signal_DS: //断丝故障
|
||
{
|
||
lightEntry := component.SignalLightsType.Get(signalEntry).GetLightByColor(light)
|
||
if lightEntry == nil {
|
||
return ecs.NewErrResult(fmt.Errorf("信号机[%s]没有色灯[%d]", signalId, light))
|
||
}
|
||
if set {
|
||
if !lightEntry.HasComponent(component.LightFaultDsType) {
|
||
lightEntry.AddComponent(component.LightFaultDsType)
|
||
}
|
||
} else {
|
||
if lightEntry.HasComponent(component.LightFaultDsType) {
|
||
lightEntry.RemoveComponent(component.LightFaultDsType)
|
||
}
|
||
}
|
||
}
|
||
default:
|
||
return ecs.NewErrResult(fmt.Errorf("信号机[%s]不支持该故障类型[%s]", signalId, faultType))
|
||
}
|
||
}
|
||
return ecs.NewOkEmptyResult()
|
||
})
|
||
return r.Err
|
||
}
|