rts-sim-testing-service/ts/simulation/wayside/memory/wayside_memory_ckm.go

package memory

import (
	"fmt"
	"joylink.club/bj-rtsts-server/dto/data_proto"
	"joylink.club/bj-rtsts-server/dto/request_proto"
	"joylink.club/bj-rtsts-server/sys_error"
	appcomponent "joylink.club/bj-rtsts-server/ts/simulation/app_component"
	"joylink.club/ecs"
	"joylink.club/rtsssimulation/component"
	"joylink.club/rtsssimulation/entity"
	"unsafe"
)

func CkmOperation(simulation *VerifySimulation, req *request_proto.CkmOperationReq) *sys_error.BusinessError {
	switch req.Operation {
	case request_proto.Ckm_SetParams:
		return setParam(simulation, req)
	default:
		panic(fmt.Sprintf("未知的屏蔽门操作:%s", req.Operation))
	}
	return nil
}

func setParam(simulation *VerifySimulation, req *request_proto.CkmOperationReq) *sys_error.BusinessError {
	uid := QueryUidByMidAndComId(req.MapId, req.DeviceId, &data_proto.GarageDoor{})
	result := <-ecs.Request[ecs.EmptyType](simulation.World, func() ecs.Result[ecs.EmptyType] {
		wd := entity.GetWorldData(simulation.World)
		entry, ok := wd.EntityMap[uid]
		if ok {
			entry.AddComponent(appcomponent.CkmParamType, unsafe.Pointer(req.Param))
			if req.Param.Fault == request_proto.Ckm_FA_State_Loss {
				setRelayStateLoss(entry)
			} else {
				clearRelayStateLoss(entry)
			}
		} else {
			return ecs.NewErrResult(fmt.Errorf("未找到id=%s的实体", uid))
		}
		return ecs.NewOkEmptyResult()
	})
	if result.Err != nil {
		return sys_error.New("车库门设置参数失败", result.Err)
	} else {
		return nil
	}
}

func setRelayStateLoss(entry *ecs.Entry) {
	if entry.HasComponent(component.CkmCircuitType) {
		ckmCircuit := component.CkmCircuitType.Get(entry)
		for _, relayEntry := range ckmCircuit.RelayList() {
			if relayEntry != nil {
				relayEntry.AddComponent(appcomponent.RelayStateLossTag)
			}
		}
	}
}

func clearRelayStateLoss(entry *ecs.Entry) {
	if entry.HasComponent(component.CkmCircuitType) {
		ckmCircuit := component.CkmCircuitType.Get(entry)
		for _, relayEntry := range ckmCircuit.RelayList() {
			if relayEntry != nil {
				relayEntry.RemoveComponent(appcomponent.RelayStateLossTag)
			}
		}
	}
}