rts-sim-testing-service/api/simulation.go

356 lines
11 KiB
Go
Raw Normal View History

2023-07-28 14:36:16 +08:00
package api
import (
2023-07-31 17:27:05 +08:00
"fmt"
"log/slog"
2023-07-28 14:36:16 +08:00
"net/http"
"sort"
2023-07-31 17:27:05 +08:00
"strconv"
2023-07-28 14:36:16 +08:00
jwt "github.com/appleboy/gin-jwt/v2"
"github.com/gin-gonic/gin"
2023-07-31 17:27:05 +08:00
"github.com/golang/protobuf/proto"
"joylink.club/bj-rtsts-server/ats/verify/protos/graphicData"
"joylink.club/bj-rtsts-server/ats/verify/protos/state"
2023-07-31 08:41:42 +08:00
"joylink.club/bj-rtsts-server/ats/verify/simulation"
2023-08-01 14:54:11 +08:00
"joylink.club/bj-rtsts-server/ats/verify/simulation/wayside/memory"
2023-08-31 14:06:53 +08:00
"joylink.club/bj-rtsts-server/config"
2023-07-28 14:36:16 +08:00
"joylink.club/bj-rtsts-server/dto"
2023-10-13 17:43:48 +08:00
"joylink.club/bj-rtsts-server/dto/request_proto"
2023-07-31 08:41:42 +08:00
apiproto "joylink.club/bj-rtsts-server/grpcproto"
2023-08-30 09:28:21 +08:00
"joylink.club/bj-rtsts-server/middleware"
"joylink.club/bj-rtsts-server/service"
2023-07-28 14:36:16 +08:00
)
func InitSimulationRouter(api *gin.RouterGroup, authMiddleware *jwt.GinJWTMiddleware) {
2023-08-30 09:28:21 +08:00
authed := api.Group("/v1/simulation").Use(authMiddleware.MiddlewareFunc(), middleware.PermissMiddleware)
authed.POST("/createByProject", createByProjectId)
2023-07-28 17:24:58 +08:00
authed.POST("/destroy/:id", destroy)
authed.GET("/list", findAllSimulations)
authed.POST("/check/data", checkSimMapData)
2023-07-31 17:27:05 +08:00
authed.POST("/train/add", addTrain)
authed.POST("/train/remove", removeTrain)
authed.POST("/switch/operation", switchOperation)
2023-08-31 14:06:53 +08:00
authed.GET("/getDataChannelName", getDataChannelName)
2023-10-07 17:48:39 +08:00
authed.POST("/relay/operation", relayOperation)
2023-10-13 14:41:55 +08:00
authed.POST("/signal/operation", signalOperation)
2023-07-31 08:41:42 +08:00
// 初始化地图信息
initPublishMapInfo()
2023-09-21 16:22:55 +08:00
apiproto.RegisterMsgServer(&apiproto.SimulationServer{})
2023-08-31 10:30:44 +08:00
apiproto.RegisterMsgServer(&apiproto.MemoryChangeServer{SimulationMap: make(map[string]*state.SimulationStatus)})
2023-07-28 14:36:16 +08:00
}
func initPublishMapInfo() {
2023-08-01 17:16:16 +08:00
mapArr, err := service.ListAllPublishedGi()
if err != nil {
panic("地图加载出错")
}
sort.SliceStable(mapArr, func(i, j int) bool { return mapArr[i].Type != 0 })
for _, v := range mapArr {
memory.PublishMapVerifyStructure(v)
}
}
// 创建ATS测试仿真通过项目ID
//
// @Summary 创建ATS测试仿真
//
// @Security JwtAuth
//
// @Description 创建ATS测试仿真通过项目ID
// @Tags ATS测试仿真Api
// @Accept json
// @Produce json
// @Param Authorization header string true "JWT Token"
// @Param SimulationCreateReqDto body dto.SimulationCreateReqDto true "创建仿真请求"
// @Success 200 {object} dto.SimulationCreateRspDto
// @Failure 500 {object} dto.ErrorDto
// @Router /api/v1/simulation/createByProject [post]
func createByProjectId(c *gin.Context) {
req := dto.SimulationCreateReqDto{}
if err := c.ShouldBind(&req); nil != err {
2023-08-30 18:05:11 +08:00
panic(dto.ErrorDto{Code: dto.ArgumentParseError, Message: err.Error()})
}
2023-08-31 16:16:18 +08:00
mapInfos := service.QueryProjectPublishedGi(req.ProjectId)
if len(mapInfos) == 0 {
panic(dto.ErrorDto{Code: dto.DataNotExist, Message: "项目未关联地图"})
}
2023-09-19 11:02:50 +08:00
mapIds := make([]int32, len(mapInfos))
for i, mapInfo := range mapInfos {
mapIds[i] = mapInfo.ID
}
rsp := dto.SimulationCreateRspDto{ProjectId: req.ProjectId, MapId: mapIds[0], MapIds: mapIds}
rsp.SimulationId = simulation.CreateSimulation(req.ProjectId, mapIds)
2023-07-28 14:36:16 +08:00
c.JSON(http.StatusOK, &rsp)
}
// ATS仿真销毁
//
// @Summary ATS仿真销毁
//
// @Security JwtAuth
//
// @Description ATS测试仿真-添加列车
// @Tags ATS测试仿真Api
// @Accept json
// @Produce json
// @Param Authorization header string true "JWT Token"
// @Param id path int true "仿真id"
// @Success 200 {object} string
// @Failure 500 {object} dto.ErrorDto
// @Router /api/v1/simulation/destroy/{id} [post]
func destroy(c *gin.Context) {
simId := c.Param("id")
slog.Debug("ATS测试仿真-ATS仿真销毁 请求:", simId)
simulation.DestroySimulation(simId)
c.JSON(http.StatusOK, "ok")
}
// 获取ATS测试系统所有仿真实例的基本信息
//
// @Summary 获取ATS测试系统所有仿真实例的基本信息
//
// @Security JwtAuth
//
// @Description 获取ATS测试系统所有仿真实例的基本信息
// @Tags ATS测试仿真Api
// @Accept json
// @Produce json
// @Param Authorization header string true "JWT Token"
// @Success 200 {object} dto.SimulationInfoRspDtoArr
// @Failure 500 {object} dto.ErrorDto
// @Router /api/v1/simulation/list [get]
func findAllSimulations(c *gin.Context) {
2023-08-01 14:54:11 +08:00
c.JSON(http.StatusOK, simulation.ListAllSimulations())
}
// ATS测试仿真地图数据校验
//
// @Summary ATS测试仿真地图数据校验
//
// @Security JwtAuth
//
// @Description 地图数据校验
// @Tags ATS测试仿真Api
// @Accept json
// @Produce json
// @Param Authorization header string true "JWT Token"
// @Param RemoveTrainDto body dto.CheckMapDataReqDto true "ATS测试仿真-地图数据"
//
// @Success 200 {object} dto.CheckMapDataRspDto
// @Failure 500 {object} dto.ErrorDto
// @Router /api/v1/simulation/check/data [post]
func checkSimMapData(c *gin.Context) {
2023-07-31 17:27:05 +08:00
rt := &dto.CheckMapDataReqDto{}
if err := c.ShouldBind(&rt); nil != err {
2023-08-30 18:05:11 +08:00
panic(dto.ErrorDto{Code: dto.ArgumentParseError, Message: err.Error()})
2023-07-31 17:27:05 +08:00
}
err := proto.Unmarshal(rt.Data, &graphicData.RtssGraphicStorage{})
if err != nil {
c.JSON(http.StatusInternalServerError, "参数错误")
2023-08-31 16:16:18 +08:00
return
}
2023-08-31 16:16:18 +08:00
c.JSON(http.StatusOK, &dto.CheckMapDataRspDto{Success: true})
}
2023-07-31 17:27:05 +08:00
// ATS测试仿真-添加列车
//
// @Summary ATS测试仿真-添加列车
//
// @Security JwtAuth
//
// @Description ATS测试仿真-添加列车
// @Tags ATS测试仿真Api
// @Accept json
// @Produce json
// @Param Authorization header string true "JWT Token"
// @Param AddTrainReqDto body dto.AddTrainReqDto true "ATS测试仿真-添加列车"
// @Success 200 {object} dto.AddTrainRspDto
// @Failure 500 {object} dto.ErrorDto
// @Router /api/v1/simulation/train/add [post]
func addTrain(c *gin.Context) {
req := dto.AddTrainReqDto{}
if err := c.ShouldBind(&req); err != nil {
2023-08-30 18:05:11 +08:00
panic(dto.ErrorDto{Code: dto.ArgumentParseError, Message: err.Error()})
2023-07-31 17:27:05 +08:00
}
simulation := checkDeviceDataAndReturn(req.SimulationId)
2023-08-22 16:44:34 +08:00
id := getAddTrainPrimaryKey(simulation)
if id == -1 {
c.JSON(http.StatusBadRequest, "已存在在线列车")
2023-08-31 16:16:18 +08:00
return
2023-07-31 17:27:05 +08:00
}
2023-08-31 16:16:18 +08:00
rsp := &state.TrainState{
Id: strconv.Itoa(id),
HeadDeviceId: req.Id,
HeadOffset: req.HeadOffset,
DevicePort: req.DevicePort,
RunDirection: req.RunDirection,
TrainLength: req.TrainLength,
}
2023-09-21 17:57:50 +08:00
memory.AddTrainState(simulation, rsp, req.MapId)
2023-08-31 16:16:18 +08:00
c.JSON(http.StatusOK, &rsp)
2023-07-31 17:27:05 +08:00
}
2023-07-28 16:14:31 +08:00
// ATS测试仿真-移除列车
//
2023-07-28 16:14:31 +08:00
// @Summary ATS测试仿真-移除列车
//
// @Security JwtAuth
//
2023-07-28 16:14:31 +08:00
// @Description ATS测试仿真-移除列车
// @Tags ATS测试仿真Api
// @Accept json
// @Produce json
// @Param Authorization header string true "JWT Token"
// @Param RemoveTrainDto body dto.RemoveTrainDto true "ATS测试仿真-移除列车"
//
// @Success 200 {object} string
// @Failure 500 {object} dto.ErrorDto
// @Router /api/v1/simulation/train/remove [post]
func removeTrain(c *gin.Context) {
rt := &dto.RemoveTrainDto{}
2023-07-31 17:27:05 +08:00
if err := c.ShouldBind(&rt); err != nil {
2023-08-30 18:05:11 +08:00
panic(dto.ErrorDto{Code: dto.ArgumentParseError, Message: err.Error()})
}
slog.Debug("ATS测试仿真-移除列车,请求:", rt)
simulation := checkDeviceDataAndReturn(rt.SimulationId)
memory.RemoveTrainState(simulation, rt.TrainId)
2023-07-31 17:27:05 +08:00
//TODO 后续调用列车删除操作
c.JSON(http.StatusOK, "ok")
}
// 获取ATS测试-操作道岔
//
// @Summary 获取ATS测试-操作道岔
//
// @Security JwtAuth
//
// @Description ATS测试-操作道岔
// @Tags ATS测试仿真Api
// @Accept json
// @Produce json
// @Param Authorization header string true "JWT Token"
2023-10-16 09:05:37 +08:00
// @Param TurnoutOperationReq body request_proto.TurnoutOperationReq true "ATS测试仿真-操作道岔"
//
// @Success 200 {object} string
// @Failure 500 {object} dto.ErrorDto
// @Router /api/v1/simulation/switch/operation [post]
func switchOperation(c *gin.Context) {
2023-10-13 17:43:48 +08:00
req := &request_proto.TurnoutOperationReq{}
2023-07-31 17:27:05 +08:00
if err := c.ShouldBind(&req); err != nil {
2023-08-30 18:05:11 +08:00
panic(dto.ErrorDto{Code: dto.ArgumentParseError, Message: err.Error()})
2023-07-31 17:27:05 +08:00
}
simulation := checkDeviceDataAndReturn(req.SimulationId)
slog.Info("传入状态参数", req)
2023-10-13 17:43:48 +08:00
memory.HandleTurnoutOperation(simulation, req)
// memory.ChangeTurnoutState(simulation, &state.SwitchState{
// Id: req.DeviceId,
// Normal: req.TurnNormal,
// Reverse: req.TurnReverse,
// }, req.MapId)
2023-07-31 17:27:05 +08:00
c.JSON(http.StatusOK, "ok")
}
2023-10-13 14:41:55 +08:00
// ATS测试-信号机操作
//
// @Summary ATS测试-信号机操作
//
// @Security JwtAuth
//
// @Description ATS测试-信号机操作
// @Tags ATS测试仿真Api
// @Accept json
// @Produce json
// @Param Authorization header string true "JWT Token"
// @Param SignalOperationReqDto body dto.SignalOperationReqDto true "ATS测试仿真-操作信号机"
//
// @Success 200 {object} string
// @Failure 500 {object} dto.ErrorDto
// @Router /api/v1/simulation/signal/operation [post]
func signalOperation(c *gin.Context) {
req := &dto.SignalOperationReqDto{}
if err := c.ShouldBind(&req); err != nil {
panic(dto.ErrorDto{Code: dto.ArgumentParseError, Message: err.Error()})
}
2023-10-16 10:37:59 +08:00
simulation := checkDeviceDataAndReturn(req.SimulationId)
2023-10-13 14:41:55 +08:00
slog.Info("传入状态参数", req)
2023-10-13 17:44:17 +08:00
memory.ChangeSignalState(simulation, req.MapId, req.DeviceId, req.Aspect)
2023-10-13 14:41:55 +08:00
c.JSON(http.StatusOK, "ok")
}
2023-08-31 14:06:53 +08:00
// 获取仿真信息更新通道名称
//
// @Summary 获取仿真信息更新通道名称
//
// @Security JwtAuth
//
// @Description 获取仿真信息更新通道名称
// @Tags ATS测试仿真Api
// @Accept json
// @Produce json
// @Param Authorization header string true "JWT Token"
//
// @Success 200 {object} string
// @Failure 500 {object} dto.ErrorDto
// @Router /api/v1/simulation/getDataChannelName [get]
func getDataChannelName(c *gin.Context) {
c.JSON(http.StatusOK, config.SimulationId_prefix+apiproto.MemoryChangeServerSuffix)
}
2023-10-07 17:48:39 +08:00
// 获取ATS测试-操作继电器
//
// @Summary 获取ATS测试-操作继电器
//
// @Security JwtAuth
//
// @Description ATS测试-操作继电器
// @Tags ATS测试仿真Api
// @Accept json
// @Produce json
// @Param Authorization header string true "JWT Token"
// @Param RelayOperationReqDto body dto.RelayOperationReqDto true "ATS测试仿真-操作继电器"
//
// @Success 200 {object} string
// @Failure 500 {object} dto.ErrorDto
// @Router /api/v1/simulation/relay/operation [post]
func relayOperation(c *gin.Context) {
req := &dto.RelayOperationReqDto{}
if err := c.ShouldBind(&req); err != nil {
panic(dto.ErrorDto{Code: dto.ArgumentParseError, Message: err.Error()})
}
simulation := checkDeviceDataAndReturn(req.SimulationId)
slog.Info("传入状态参数", req)
2023-10-07 17:51:00 +08:00
memory.ChangeRelayState(simulation, req.MapId, req.Id, req.Td)
2023-10-07 17:48:39 +08:00
c.JSON(http.StatusOK, "ok")
}
2023-07-31 17:27:05 +08:00
// 获取仿真设备数据并返回
2023-08-01 14:54:11 +08:00
func checkDeviceDataAndReturn(simId string) *memory.VerifySimulation {
2023-07-31 17:27:05 +08:00
deviceMemory := simulation.FindSimulation(simId)
if deviceMemory == nil {
panic(&dto.ErrorDto{Code: dto.DataNotExist, Message: fmt.Sprintf("仿真[%s]不存在", simId)})
}
2023-07-31 17:27:05 +08:00
return deviceMemory
}
2023-08-22 16:44:34 +08:00
// 获取列车主键
func getAddTrainPrimaryKey(simulation *memory.VerifySimulation) int {
trainMap := &simulation.Memory.Status.TrainStateMap
// 获取列车ID
i := 1
for {
t, ok := trainMap.Load(strconv.Itoa(i))
2023-08-31 16:16:18 +08:00
if !ok {
break
}
ts := t.(*state.TrainState)
if ts.Show {
i = -1
2023-08-22 16:44:34 +08:00
break
}
2023-08-31 16:16:18 +08:00
i++
2023-08-22 16:44:34 +08:00
}
return i
}