244 lines
7.4 KiB
Go
244 lines
7.4 KiB
Go
package api
|
|
|
|
import (
|
|
"net/http"
|
|
"strconv"
|
|
|
|
jwt "github.com/appleboy/gin-jwt/v2"
|
|
"github.com/gin-gonic/gin"
|
|
"go.uber.org/zap"
|
|
"google.golang.org/protobuf/proto"
|
|
"joylink.club/bj-rtsts-server/ats/verify/protos/graphicData"
|
|
"joylink.club/bj-rtsts-server/ats/verify/simulation/wayside/memory"
|
|
"joylink.club/bj-rtsts-server/db/model"
|
|
"joylink.club/bj-rtsts-server/dto"
|
|
"joylink.club/bj-rtsts-server/middleware"
|
|
"joylink.club/bj-rtsts-server/service"
|
|
)
|
|
|
|
func InitDraftingRouter(api *gin.RouterGroup, authMiddleware *jwt.GinJWTMiddleware) {
|
|
authed := api.Group("/v1/drafting").Use(authMiddleware.MiddlewareFunc(), middleware.PermissMiddleware)
|
|
authed.GET("/paging", pageQueryDrafting)
|
|
authed.POST("", createDrafting)
|
|
authed.POST("/:id/saveAs", saveAsDrafting)
|
|
authed.GET("/:id", queryDraftingInfo)
|
|
authed.PUT("/:id", updateDraftingInfo)
|
|
authed.DELETE("/:id", deleteDrafting)
|
|
authed.POST("/calculatelink", generateCalculateLinkData)
|
|
}
|
|
|
|
// 分页查询草稿
|
|
//
|
|
// @Summary 分页查询草稿
|
|
//
|
|
// @Security JwtAuth
|
|
//
|
|
// @Description 可以通过草稿名称过滤，分页查询草稿
|
|
// @Tags 草稿Api
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param PageDraftingReqDto query dto.PageDraftingReqDto true "草稿查询条件带分页信息"
|
|
// @Success 200 {object} dto.PageDto
|
|
// @Failure 401 {object} dto.ErrorDto
|
|
// @Failure 404 {object} dto.ErrorDto
|
|
// @Failure 500 {object} dto.ErrorDto
|
|
// @Router /api/v1/drafting/paging [get]
|
|
func pageQueryDrafting(c *gin.Context) {
|
|
user, _ := c.Get(middleware.IdentityKey)
|
|
zap.S().Debug("分页查询草稿", user)
|
|
req := dto.PageDraftingReqDto{}
|
|
if err := c.ShouldBind(&req); err != nil {
|
|
panic(dto.ErrorDto{Code: dto.ArgumentParseError, Message: err.Error()})
|
|
}
|
|
zap.S().Debug("分页查草稿参数", req)
|
|
page, err := service.PageDraftingQuery(&req)
|
|
if err != nil {
|
|
panic(dto.ErrorDto{Code: dto.DataOperationError, Message: err.Error()})
|
|
}
|
|
c.JSON(http.StatusOK, page)
|
|
}
|
|
|
|
// 创建草稿
|
|
//
|
|
// @Summary 创建草稿
|
|
//
|
|
// @Security JwtAuth
|
|
//
|
|
// @Description 创建草稿数据
|
|
// @Tags 草稿Api
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param draftingDto query dto.DraftingDto true "创建的草稿信息"
|
|
// @Success 200 {object} nil
|
|
// @Failure 401 {object} dto.ErrorDto
|
|
// @Failure 404 {object} dto.ErrorDto
|
|
// @Failure 500 {object} dto.ErrorDto
|
|
// @Router /api/v1/drafting [post]
|
|
func createDrafting(c *gin.Context) {
|
|
user, _ := c.Get(middleware.IdentityKey)
|
|
createId := user.(*model.User).ID
|
|
req := dto.DraftingDto{}
|
|
if err := c.ShouldBind(&req); err != nil {
|
|
panic(dto.ErrorDto{Code: dto.ArgumentParseError, Message: err.Error()})
|
|
}
|
|
zap.S().Debug("保存数据", req)
|
|
data, err := service.CreateDrafting(createId, &req)
|
|
if err != nil {
|
|
panic(dto.ErrorDto{Code: dto.DataOperationError, Message: err.Error()})
|
|
}
|
|
c.JSON(http.StatusOK, data)
|
|
}
|
|
|
|
// 草稿另存为
|
|
//
|
|
// @Summary 草稿另存为
|
|
//
|
|
// @Security JwtAuth
|
|
//
|
|
// @Description 草稿另存为
|
|
// @Tags 草稿Api
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param id path int true "源草稿id"
|
|
// @Param draftingDto query dto.DraftingDto true "另存为草稿信息"
|
|
// @Success 200 {object} nil
|
|
// @Failure 401 {object} dto.ErrorDto
|
|
// @Failure 404 {object} dto.ErrorDto
|
|
// @Failure 500 {object} dto.ErrorDto
|
|
// @Router /api/v1/drafting/{id}/saveAs [post]
|
|
func saveAsDrafting(c *gin.Context) {
|
|
id, exist := c.Params.Get("id")
|
|
if !exist {
|
|
panic(dto.ErrorDto{Code: dto.ArgumentParseError, Message: "必要参数id不存在"})
|
|
}
|
|
zap.S().Debug("传入参数id为" + id)
|
|
user, _ := c.Get(middleware.IdentityKey)
|
|
createrId := user.(*model.User).ID
|
|
req := dto.DraftingDto{}
|
|
if err := c.ShouldBind(&req); err != nil {
|
|
panic(dto.ErrorDto{Code: dto.ArgumentParseError, Message: err.Error()})
|
|
}
|
|
int64Id, _ := strconv.ParseInt(id, 10, 64)
|
|
newData, err := service.SaveAsDrafting(createrId, int32(int64Id), &req)
|
|
if err != nil {
|
|
panic(dto.ErrorDto{Code: dto.DataOperationError, Message: err.Error()})
|
|
}
|
|
c.JSON(http.StatusOK, newData)
|
|
}
|
|
|
|
// 查询草稿详情
|
|
//
|
|
// @Summary 查询草稿详情
|
|
//
|
|
// @Security JwtAuth
|
|
//
|
|
// @Description 查询草稿详情
|
|
// @Tags 草稿Api
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param id path int true "草稿ID"
|
|
// @Success 200 {object} model.Drafting
|
|
// @Failure 401 {object} dto.ErrorDto
|
|
// @Failure 404 {object} dto.ErrorDto
|
|
// @Failure 500 {object} dto.ErrorDto
|
|
// @Router /api/v1/drafting/{id} [get]
|
|
func queryDraftingInfo(c *gin.Context) {
|
|
id, exist := c.Params.Get("id")
|
|
if !exist {
|
|
panic(dto.ErrorDto{Code: dto.ArgumentParseError, Message: "必要参数id不存在"})
|
|
}
|
|
zap.S().Debug("传入参数id为" + id)
|
|
int64Id, _ := strconv.ParseInt(id, 10, 64)
|
|
c.JSON(http.StatusOK, service.QueryDrafting(int32(int64Id)))
|
|
}
|
|
|
|
// 修改草稿信息
|
|
//
|
|
// @Summary 修改草稿信息
|
|
//
|
|
// @Security JwtAuth
|
|
//
|
|
// @Description 修改草稿信息
|
|
// @Tags 草稿Api
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param id path int true "草稿ID"
|
|
// @Param draftingDto query dto.DraftingDto true "修改的草稿信息"
|
|
// @Success 200 {object} nil
|
|
// @Failure 401 {object} dto.ErrorDto
|
|
// @Failure 404 {object} dto.ErrorDto
|
|
// @Failure 500 {object} dto.ErrorDto
|
|
// @Router /api/v1/drafting/{id} [put]
|
|
func updateDraftingInfo(c *gin.Context) {
|
|
id, exist := c.Params.Get("id")
|
|
if !exist {
|
|
panic(dto.ErrorDto{Code: dto.ArgumentParseError, Message: "必要参数id不存在"})
|
|
}
|
|
zap.S().Debug("传入参数id为" + id)
|
|
req := dto.DraftingDto{}
|
|
if err := c.ShouldBind(&req); err != nil {
|
|
panic(dto.ErrorDto{Code: dto.ArgumentParseError, Message: err.Error()})
|
|
}
|
|
int64Id, _ := strconv.ParseInt(id, 10, 64)
|
|
result := service.UpdateDrafting(int32(int64Id), &req)
|
|
if !result {
|
|
panic(dto.ErrorDto{Code: dto.DataOperationError, Message: "保存出错"})
|
|
}
|
|
c.JSON(http.StatusOK, result)
|
|
}
|
|
|
|
// 删除草稿数据
|
|
//
|
|
// @Summary 删除草稿数据
|
|
//
|
|
// @Security JwtAuth
|
|
//
|
|
// @Description 删除草稿数据
|
|
// @Tags 草稿Api
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param id path int true "草稿ID"
|
|
// @Success 200 {object} nil
|
|
// @Failure 401 {object} dto.ErrorDto
|
|
// @Failure 404 {object} dto.ErrorDto
|
|
// @Failure 500 {object} dto.ErrorDto
|
|
// @Router /api/v1/drafting/{id} [delete]
|
|
func deleteDrafting(c *gin.Context) {
|
|
user, _ := c.Get(middleware.IdentityKey)
|
|
zap.S().Debug("id删除草稿的图形数据", user)
|
|
idStr := c.Param("id")
|
|
id, err := strconv.Atoi(idStr)
|
|
if err != nil {
|
|
panic(dto.ErrorDto{Code: dto.ArgumentParseError, Message: err.Error()})
|
|
}
|
|
zap.S().Debug("id查询草稿的图形数据", id)
|
|
service.DeleteDraftingById(id)
|
|
c.JSON(http.StatusOK, true)
|
|
}
|
|
|
|
// 根据地图数据新生成计算的link信息
|
|
//
|
|
// @Summary 根据地图数据新生成计算的link信息
|
|
//
|
|
// @Security JwtAuth
|
|
//
|
|
// @Description 根据地图数据新生成计算的link信息
|
|
// @Tags 草稿Api
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param DraftingMapDataDto query dto.DraftingMapDataDto true "地图信息"
|
|
// @Success 200 {object} nil
|
|
// @Failure 401 {object} dto.ErrorDto
|
|
// @Failure 404 {object} dto.ErrorDto
|
|
// @Failure 500 {object} dto.ErrorDto
|
|
// @Router /api/v1/drafting/calculatelink [post]
|
|
func generateCalculateLinkData(c *gin.Context) {
|
|
req := dto.DraftingMapDataDto{}
|
|
if err := c.ShouldBind(&req); err != nil {
|
|
panic(dto.ErrorDto{Code: dto.ArgumentParseError, Message: err.Error()})
|
|
}
|
|
gd := &graphicData.RtssGraphicStorage{}
|
|
proto.Unmarshal(req.Proto, gd)
|
|
c.JSON(http.StatusOK, memory.BuildCalculateLinkData(gd))
|
|
}
|