100 lines
2.6 KiB
Go
100 lines
2.6 KiB
Go
package server
|
|
|
|
import (
|
|
"github.com/labstack/echo-contrib/echoprometheus"
|
|
"github.com/labstack/echo/v4"
|
|
echoMiddleware "github.com/labstack/echo/v4/middleware"
|
|
"leafdev.top/Leaf/api-platform/internal/base/conf"
|
|
httpHandlers "leafdev.top/Leaf/api-platform/internal/handler/http"
|
|
"leafdev.top/Leaf/api-platform/internal/handler/http/response"
|
|
"leafdev.top/Leaf/api-platform/internal/router"
|
|
"leafdev.top/Leaf/api-platform/pkg/consts"
|
|
"net/http"
|
|
)
|
|
|
|
type HttpServer struct {
|
|
config *conf.Config
|
|
Echo *echo.Echo
|
|
apiRouter *router.Api
|
|
swaggerRouter *router.SwaggerRouter
|
|
middleware *httpHandlers.Middleware
|
|
}
|
|
|
|
// NewHTTPServer new http server.
|
|
func NewHTTPServer(
|
|
config *conf.Config,
|
|
apiRouter *router.Api,
|
|
swaggerRouter *router.SwaggerRouter,
|
|
middleware *httpHandlers.Middleware,
|
|
) *HttpServer {
|
|
|
|
e := echo.New()
|
|
e.Use(echoMiddleware.Recover())
|
|
e.Use(middleware.Logger.Handler())
|
|
e.Use(middleware.JSONResponse.Handler())
|
|
|
|
return &HttpServer{
|
|
config: config,
|
|
Echo: e,
|
|
apiRouter: apiRouter,
|
|
swaggerRouter: swaggerRouter,
|
|
middleware: middleware,
|
|
}
|
|
}
|
|
|
|
func (hs *HttpServer) AllowAllCors() {
|
|
var defaultCORSConfig = echoMiddleware.CORSConfig{
|
|
AllowOrigins: []string{"*"},
|
|
AllowHeaders: []string{"Origin", "Content-Length", "Content-Type", "X-Requested-With", "X-Auth-Token", "Authorization"},
|
|
AllowCredentials: true,
|
|
AllowMethods: []string{http.MethodGet, http.MethodHead, http.MethodPut, http.MethodPatch, http.MethodPost, http.MethodDelete},
|
|
MaxAge: 12 * 60,
|
|
}
|
|
|
|
hs.Echo.Use(echoMiddleware.CORSWithConfig(defaultCORSConfig))
|
|
}
|
|
|
|
func (hs *HttpServer) BizRouter() *echo.Echo {
|
|
hs.AllowAllCors()
|
|
|
|
rootGroup := hs.Echo.Group("")
|
|
|
|
// swagger
|
|
hs.swaggerRouter.Register(rootGroup)
|
|
|
|
apiV1 := rootGroup.Group("/api/v1")
|
|
{
|
|
//apiV1.Use(corsMiddleWare)
|
|
apiV1.Use(hs.middleware.JSONResponse.Handler())
|
|
apiV1.Use(hs.middleware.Auth.Handler())
|
|
hs.apiRouter.InitApiRouter(apiV1)
|
|
}
|
|
|
|
apiV1NoAuth := rootGroup.Group("/api/v1")
|
|
{
|
|
//apiV1.Use(corsMiddleWare)
|
|
hs.apiRouter.InitNoAuthApiRouter(apiV1NoAuth)
|
|
}
|
|
|
|
hs.Echo.RouteNotFound("/*", func(ctx echo.Context) error {
|
|
return response.Ctx(ctx).Status(http.StatusNotFound).Error(consts.ErrPageNotFound).Send()
|
|
})
|
|
|
|
return hs.Echo
|
|
}
|
|
|
|
func (hs *HttpServer) MetricRouter() *echo.Echo {
|
|
e := echo.New()
|
|
e.Use(echoMiddleware.Recover())
|
|
|
|
metricGroup := e.Group("")
|
|
// prometheus
|
|
metricGroup.Use(echoprometheus.NewMiddleware(hs.config.App.Name))
|
|
|
|
metricGroup.GET("/metrics", echoprometheus.NewHandler())
|
|
|
|
metricGroup.GET("/healthz", func(ctx echo.Context) error { return ctx.String(200, "OK") })
|
|
|
|
return e
|
|
}
|