confrontation-training/initialize/router.go

package initialize

import (
	"confrontation-training/api/gateway"
	"confrontation-training/api/question"
	"confrontation-training/global"
	_ "confrontation-training/global"
	"confrontation-training/middleware"
	"confrontation-training/models"
	"fmt"
	"github.com/gin-gonic/gin"
	swaggerFiles "github.com/swaggo/files"
	ginSwagger "github.com/swaggo/gin-swagger"
	"net/http"
)

func Router() {

	//初始化表格
	initTableErr := global.Db.AutoMigrate(&models.User{}, &models.Question{})
	if initTableErr != nil {
		fmt.Printf("初始化表格异常：%s", initTableErr.Error())
		return
	}
	gin.SetMode(gin.ReleaseMode)
	engine := gin.Default()
	//解决跨域
	engine.Use(middleware.Cors())

	//404 NOT FOUND
	engine.NoRoute(func(context *gin.Context) {
		context.String(http.StatusNotFound, "404 not found")
	})

	//swag
	engine.GET("/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler))
	engine.Static("/file", "./files")
	engine.GET("/ws", WsHandler)

	//group
	v1 := engine.Group("/v1")
	user := v1.Group("/user")
	user.POST("/login", gateway.GetUser().UserLogin)

	//user.Use(middleware.JwtAuth())
	user.POST("/register", gateway.GetUser().UseRegister)
	user.POST("/reset", gateway.GetUser().ResetPassword)
	user.POST("/find", gateway.GetUser().UserList)
	user.POST("/change/password", gateway.GetUser().ModePass)
	device := v1.Group("/device")
	device.GET("/scan", gateway.ScanDevice)
	device.GET("/connection", gateway.ConnectDevice)
	device.POST("/write/data/", gateway.WriteData)
	device.GET("/open/notify/", gateway.OpenNotify)
	device.GET("/:mac/stop/trans/", gateway.StopTrans)
	device.GET("/:mac/stop/collect/", gateway.StopCollect)
	device.GET("/:mac/disconnect/", gateway.Disconnect)
	chat := v1.Group("/chat")
	chat.GET("/get/question/", question.GetQuestionService().GetQuestion)
	chat.POST("/create/question/", question.GetQuestionService().CreateQuestion)
	chat.GET("/get/answer/", question.GteAnswerService().CreateAnswer)

	//
	//record := v1.Group("/record")
	//record.POST("/create", api.GetRecord().SaveRecord)
	//record.POST("/find", api.GetRecord().RecordList)
	//record.GET("/:recordId/find", api.GetRecord().GetRecordById)
	err := engine.Run(fmt.Sprintf(":%s", global.Config.Server.Port))
	if err != nil {
		fmt.Printf("%s", err.Error())
		return
	}
}