72 lines
2.0 KiB
Go
72 lines
2.0 KiB
Go
|
package routing
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/fasthttp/router"
|
||
|
|
"net/http"
|
||
|
|
)
|
||
|
|
|
||
|
|
type (
|
||
|
|
// RouterContext is a fasthttp request url routing context.
|
||
|
|
RouterContext interface {
|
||
|
|
// UrlResolver returns a UrlResolver.
|
||
|
|
UrlResolver() UrlResolver
|
||
|
|
urlFor() UrlFor
|
||
|
|
// UrlPrefix returns the URL prefix that is added across the entire routing group.
|
||
|
|
UrlPrefix() string
|
||
|
|
withRegister(
|
||
|
|
r *router.Router,
|
||
|
|
parentNames []string,
|
||
|
|
parentPaths []string,
|
||
|
|
middlewares ...Middleware,
|
||
|
|
) RouterRegistry
|
||
|
|
// BuildRouter returns a RouterRegistry for register grouped routing.
|
||
|
|
BuildRouter(errHandleMiddleware Middleware) RouterRegistry
|
||
|
|
}
|
||
|
|
|
||
|
|
routerContextImpl struct {
|
||
|
|
urlPrefix string
|
||
|
|
reverseRouter UrlFor
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
func (ctx *routerContextImpl) UrlResolver() UrlResolver { return ctx.reverseRouter.ToResolver() }
|
||
|
|
func (ctx *routerContextImpl) urlFor() UrlFor { return ctx.reverseRouter }
|
||
|
|
func (ctx *routerContextImpl) UrlPrefix() string { return ctx.urlPrefix }
|
||
|
|
func (ctx *routerContextImpl) withRegister(
|
||
|
|
r *router.Router,
|
||
|
|
parentNames []string,
|
||
|
|
parentPaths []string,
|
||
|
|
middlewares ...Middleware,
|
||
|
|
) RouterRegistry {
|
||
|
|
return &routerRegistryImpl{
|
||
|
|
routerContextImpl: ctx,
|
||
|
|
r: r,
|
||
|
|
middlewares: middlewares,
|
||
|
|
parentNames: parentNames,
|
||
|
|
parentPaths: parentPaths,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
func (c *routerContextImpl) BuildRouter(errHandlemiddleware Middleware) RouterRegistry {
|
||
|
|
r := router.New()
|
||
|
|
r.RedirectTrailingSlash = true
|
||
|
|
r.RedirectFixedPath = true
|
||
|
|
r.HandleOPTIONS = false
|
||
|
|
r.HandleMethodNotAllowed = true
|
||
|
|
if errHandlemiddleware != nil {
|
||
|
|
r.NotFound = JustCode(http.StatusNotFound, errHandlemiddleware)
|
||
|
|
r.MethodNotAllowed = JustCode(http.StatusMethodNotAllowed, errHandlemiddleware)
|
||
|
|
}
|
||
|
|
return &routerRegistryImpl{routerContextImpl: c, r: r}
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewRouterContext returns a RouterContext.
|
||
|
|
func NewRouterContext(
|
||
|
|
urlPrefix string,
|
||
|
|
reverseRouter UrlFor,
|
||
|
|
) (ctx RouterContext) {
|
||
|
|
return &routerContextImpl{
|
||
|
|
urlPrefix: urlPrefix,
|
||
|
|
reverseRouter: reverseRouter,
|
||
|
|
}
|
||
|
|
}
|