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

package api

// 列车相关的关系管理
import (
	"log/slog"
	"net/http"
	"strconv"

	jwt "github.com/appleboy/gin-jwt/v2"
	"github.com/gin-gonic/gin"
	"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"
	"joylink.club/bj-rtsts-server/sys_error"
)

func InitTrainManageRouter(api *gin.RouterGroup, authMiddleware *jwt.GinJWTMiddleware) {
	authed := api.Group("/v1/trainManage").Use(authMiddleware.MiddlewareFunc(), middleware.PermissMiddleware)
	authed.GET("/paging", pageQueryTrainInfo)
	authed.GET("/list", queryTrainInfoList)
	authed.POST("", createTrainInfo)
	authed.GET("/:id", queryTrainInfo)
	authed.PUT("/:id", updateTrainInfo)
	authed.DELETE("/:id", deleteTrainInfo)
}

// 分页查询列车列表
//
//	@Summary		分页查询列车信息列表
//
//	@Security		JwtAuth
//
//	@Description	可以通过列车名称过滤，分页查询列车信息列表
//	@Tags			列车管理Api
//	@Accept			json
//	@Produce		json
//	@Param			pageTrainInfoReqDto	query		dto.PageTrainInfoReqDto	true	"列车查询条件带分页信息"
//	@Success		200	{object}	dto.PageDto
//	@Failure		401	{object}	dto.ErrorDto
//	@Failure		404	{object}	dto.ErrorDto
//	@Failure		500	{object}	dto.ErrorDto
//	@Router			/api/v1/trainManage/paging [get]
func pageQueryTrainInfo(c *gin.Context) {
	req := dto.PageTrainInfoReqDto{}
	if err := c.ShouldBind(&req); err != nil {
		panic(sys_error.New("查询失败，参数格式错误", err))
	}
	c.JSON(http.StatusOK, service.PageTrainInfoByPublished(&req))
}

// 查询列车列表
//
//	@Summary		查询列车信息列表
//
//	@Security		JwtAuth
//
//	@Description	可以通过列车名称过滤，查询列车信息列表
//	@Tags			列车管理Api
//	@Accept			json
//	@Produce		json
//	@Param			trainInfoReqDto	query		dto.TrainInfoReqDto	true	"列车查询条件"
//	@Success		200	{object}	dto.TrainInfoDto
//	@Failure		401	{object}	dto.ErrorDto
//	@Failure		404	{object}	dto.ErrorDto
//	@Failure		500	{object}	dto.ErrorDto
//	@Router			/api/v1/trainManage/list [get]
func queryTrainInfoList(c *gin.Context) {
	req := dto.TrainInfoReqDto{}
	if err := c.ShouldBind(&req); err != nil {
		panic(sys_error.New("查询失败，参数格式错误", err))
	}
	c.JSON(http.StatusOK, service.ListTrainByProject(&req))
}

// 创建列车
//
//	@Summary		创建列车
//
//	@Security		JwtAuth
//
//	@Description	创建列车数据
//	@Tags			列车管理Api
//	@Accept			json
//	@Produce		json
//	@Param			trainInfoDto	body	dto.TrainInfoDto	true	"创建的列车信息"
//	@Success		200	{object}	nil
//	@Failure		401	{object}	dto.ErrorDto
//	@Failure		404	{object}	dto.ErrorDto
//	@Failure		500	{object}	dto.ErrorDto
//	@Router			/api/v1/trainManage [post]
func createTrainInfo(c *gin.Context) {
	req := dto.TrainInfoDto{}
	if err := c.ShouldBind(&req); err != nil {
		panic(sys_error.New("保存失败，参数格式错误", err))
	}
	user, _ := c.Get(middleware.IdentityKey)
	slog.Debug("发布图形数据", user)
	c.JSON(http.StatusOK, service.CreateTrain(&req, user.(*model.User)))
}

// 查询列车详情
//
//	@Summary		查询列车详情
//
//	@Security		JwtAuth
//
//	@Description	查询列车详情
//	@Tags			列车管理Api
//	@Accept			json
//	@Produce		json
//	@Param			id	path		int	true	"列车ID"
//	@Success		200	{object}    dto.TrainInfoDto
//	@Failure		401	{object}	dto.ErrorDto
//	@Failure		404	{object}	dto.ErrorDto
//	@Failure		500	{object}	dto.ErrorDto
//	@Router			/api/v1/trainManage/{id} [get]
func queryTrainInfo(c *gin.Context) {
	id, exist := c.Params.Get("id")
	if !exist {
		panic(sys_error.New("查询失败，缺少主键"))
	}
	int64Id, _ := strconv.ParseInt(id, 10, 64)
	c.JSON(http.StatusOK, service.QueryTrain(int32(int64Id)))
}

// 修改列车信息
//
//	@Summary		修改列车信息
//
//	@Security		JwtAuth
//
//	@Description	修改列车信息
//	@Tags			列车管理Api
//	@Accept			json
//	@Produce		json
//	@Param			id	path		int	true	"列车ID"
//	@Param			trainInfoDto	query	dto.TrainInfoDto	true	"修改的列车信息"
//	@Success		200	{object}	nil
//	@Failure		401	{object}	dto.ErrorDto
//	@Failure		404	{object}	dto.ErrorDto
//	@Failure		500	{object}	dto.ErrorDto
//	@Router			/api/v1/trainManage/{id} [put]
func updateTrainInfo(c *gin.Context) {
	id, exist := c.Params.Get("id")
	if !exist {
		panic(sys_error.New("更新失败，缺少主键"))
	}
	req := dto.TrainInfoDto{}
	if err := c.ShouldBind(&req); err != nil {
		panic(sys_error.New("更新失败，参数格式错误", err))
	}
	int64Id, _ := strconv.ParseInt(id, 10, 64)
	user, _ := c.Get(middleware.IdentityKey)
	slog.Debug("发布图形数据", user)
	c.JSON(http.StatusOK, service.UpdateTrain(int32(int64Id), &req, user.(*model.User)))
}

// 删除列车数据
//
//	@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/trainManage/{id} [delete]
func deleteTrainInfo(c *gin.Context) {
	idStr := c.Param("id")
	id, err := strconv.Atoi(idStr)
	if err != nil {
		panic(sys_error.New("删除失败，缺少主键"))
	}
	service.DeletePublishedById(int32(id))
	c.JSON(http.StatusOK, true)
}