256 lines
7.5 KiB
Go
256 lines
7.5 KiB
Go
package api
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"strconv"
|
|
"strings"
|
|
|
|
jwt "github.com/appleboy/gin-jwt/v2"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/golang/protobuf/proto"
|
|
"go.uber.org/zap"
|
|
"joylink.club/bj-rtsts-server/ats/verify/protos/graphicData"
|
|
"joylink.club/bj-rtsts-server/ats/verify/protos/state"
|
|
"joylink.club/bj-rtsts-server/ats/verify/simulation"
|
|
"joylink.club/bj-rtsts-server/ats/verify/simulation/wayside"
|
|
"joylink.club/bj-rtsts-server/dto"
|
|
apiproto "joylink.club/bj-rtsts-server/grpcproto"
|
|
"joylink.club/bj-rtsts-server/service"
|
|
)
|
|
|
|
func InitSimulationRouter(api *gin.RouterGroup, authMiddleware *jwt.GinJWTMiddleware) {
|
|
authed := api.Group("/v1/simulation").Use(authMiddleware.MiddlewareFunc())
|
|
authed.POST("/create", create)
|
|
authed.POST("/destroy/:id", destroy)
|
|
authed.GET("/list", findAllSimulations)
|
|
authed.POST("/check/data", checkSimMapData)
|
|
authed.POST("/train/add", addTrain)
|
|
authed.POST("/train/remove", removeTrain)
|
|
authed.POST("/switch/operation", switchOperation)
|
|
|
|
apiproto.RegisterMsgServer(&apiproto.SimulationServer{})
|
|
}
|
|
|
|
// 创建ATS测试仿真
|
|
//
|
|
// @Summary 创建ATS测试仿真
|
|
//
|
|
// @Security JwtAuth
|
|
//
|
|
// @Description 创建ATS测试仿真
|
|
// @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/create [post]
|
|
func create(c *gin.Context) {
|
|
//user,_:=c.Get(middleware.IdentityKey)
|
|
req := dto.SimulationCreateReqDto{}
|
|
if err := c.ShouldBind(&req); nil != err {
|
|
panic(err)
|
|
}
|
|
zap.S().Debug("创建仿真请求：", req)
|
|
rsp := dto.SimulationCreateRspDto{
|
|
MapId: req.MapId,
|
|
}
|
|
mapData := service.QueryRtssGraphicStorage(req.MapId)
|
|
rsp.SimulationId = simulation.CreateSimulation(int(req.MapId), mapData)
|
|
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")
|
|
zap.S().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.SimulationInfoRepDtoArr
|
|
// @Failure 500 {object} dto.ErrorDto
|
|
// @Router /api/v1/simulation/list [get]
|
|
func findAllSimulations(c *gin.Context) {
|
|
zap.S().Debug("ATS测试仿真-获取ATS测试系统所有仿真实例的基本信息,请求")
|
|
simArr := [...]dto.SimulationInfoRepDto{}
|
|
i := 0
|
|
for v := range simulation.SimulationMap {
|
|
simArr[i] = dto.SimulationInfoRepDto{
|
|
SimulationId: v,
|
|
MapId: strings.Split(v, "_")[0],
|
|
}
|
|
}
|
|
//TODO 后续调用
|
|
c.JSON(http.StatusOK, simArr)
|
|
}
|
|
|
|
// ATS测试仿真地图数据校验
|
|
//
|
|
// @Summary ATS测试仿真地图数据校验
|
|
//
|
|
// @Security JwtAuth
|
|
//
|
|
// @Description ATS测试仿真-添加列车
|
|
// @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) {
|
|
rt := &dto.CheckMapDataReqDto{}
|
|
if err := c.ShouldBind(&rt); nil != err {
|
|
panic(err)
|
|
}
|
|
err := proto.Unmarshal(rt.Data, &graphicData.RtssGraphicStorage{})
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, "参数错误")
|
|
} else {
|
|
c.JSON(http.StatusOK, &dto.CheckMapDataRspDto{Success: true})
|
|
}
|
|
}
|
|
|
|
// 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) {
|
|
//user,_:=c.Get(middleware.IdentityKey)
|
|
req := dto.AddTrainReqDto{}
|
|
if err := c.ShouldBind(&req); err != nil {
|
|
panic(err)
|
|
}
|
|
zap.S().Debug("ATS测试仿真-添加列车,请求：", req)
|
|
deviceMemory := checkDeviceDataAndReturn(req.SimulationId)
|
|
trainMap := deviceMemory.Status.TrainStateMap
|
|
// 获取列车ID
|
|
i := 1
|
|
for {
|
|
if trainMap[strconv.Itoa(i)] == nil {
|
|
break
|
|
} else {
|
|
i++
|
|
}
|
|
}
|
|
id := strconv.Itoa(i)
|
|
rsp := &state.TrainState{
|
|
Id: id,
|
|
HeadLinkId: req.HeadLinkId,
|
|
HeadLinkOffset: req.HeadLinkOffset,
|
|
Up: req.Up,
|
|
}
|
|
trainMap[id] = rsp
|
|
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 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{}
|
|
if err := c.ShouldBind(&rt); err != nil {
|
|
panic(err)
|
|
}
|
|
zap.S().Debug("ATS测试仿真-移除列车,请求：", rt)
|
|
deviceMemory := checkDeviceDataAndReturn(rt.SimulationId)
|
|
delete(deviceMemory.Status.TrainStateMap, rt.TrainId)
|
|
//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"
|
|
// @Param RemoveTrainDto body dto.SwitchOperationReqDto true "ATS测试仿真-操作道岔"
|
|
//
|
|
// @Success 200 {object} string
|
|
// @Failure 500 {object} dto.ErrorDto
|
|
// @Router /api/v1/simulation/switch/operation [post]
|
|
func switchOperation(c *gin.Context) {
|
|
req := &dto.SwitchOperationReqDto{}
|
|
if err := c.ShouldBind(&req); err != nil {
|
|
panic(err)
|
|
}
|
|
deviceMemory := checkDeviceDataAndReturn(req.SimulationId)
|
|
switchInfo := deviceMemory.Status.SwitchStateMap[req.SwitchIndex]
|
|
if switchInfo == nil {
|
|
panic(&dto.ErrorDto{Code: dto.DataNotExist, Message: fmt.Sprintf("仿真[%s]中索引为[%s]的道岔不存在", req.SimulationId, req.SwitchIndex)})
|
|
}
|
|
switchInfo.Normal = req.TurnNormal
|
|
switchInfo.Reverse = req.TurnReverse
|
|
c.JSON(http.StatusOK, "ok")
|
|
}
|
|
|
|
// 获取仿真设备数据并返回
|
|
func checkDeviceDataAndReturn(simId string) *wayside.VerifyMemory {
|
|
deviceMemory := simulation.FindSimulation(simId)
|
|
if deviceMemory == nil {
|
|
panic(&dto.ErrorDto{Code: dto.DataNotExist, Message: fmt.Sprintf("仿真[%s]不存在", simId)})
|
|
}
|
|
return deviceMemory
|
|
}
|