potato/routers/router.go

45 lines
1002 B
Go

package routers
import (
"github.com/gin-gonic/gin"
_ "github.com/swaggo/gin-swagger"
ginSwagger "github.com/swaggo/gin-swagger"
"github.com/swaggo/gin-swagger/swaggerFiles"
_ "github.com/swaggo/gin-swagger/swaggerFiles"
v1 "github.com/viletyy/potato/controller/api/v1"
_ "github.com/viletyy/potato/docs"
"github.com/viletyy/potato/middleware/cors"
"github.com/viletyy/potato/middleware/jwt"
"github.com/viletyy/potato/pkg/setting"
)
var (
Engine = gin.Default()
V1RouterGroup = Engine.Group("../api/v1")
)
func InitRouter() *gin.Engine {
Engine.Use(gin.Logger())
Engine.Use(gin.Recovery())
gin.SetMode(setting.RunMode)
Engine.Use(cors.CORSMiddleware())
Engine.GET("/api/v1/auth", v1.GetUserAuth)
Engine.GET("/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler))
V1InitModule()
return Engine
}
func V1InitModule() {
V1RouterGroup.Use(jwt.JWT())
users := V1RouterGroup.Group("users")
users.GET("", v1.GetUsers)
users.POST("", v1.AddUser)
V1InitBasicRouter()
}