100 lines
2.8 KiB
Go
100 lines
2.8 KiB
Go
|
package routing
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/fasthttp/router"
|
||
|
|
"github.com/valyala/fasthttp"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
type (
|
||
|
|
// RouterRegistry is a fasthttp request url routing builder.
|
||
|
|
RouterRegistry interface {
|
||
|
|
RouterContext
|
||
|
|
// Name returns the current group name.
|
||
|
|
Name() string
|
||
|
|
// ToContext returns the current group name.
|
||
|
|
ToContext() RouterContext
|
||
|
|
// Register is a registration method for http request.
|
||
|
|
Register(
|
||
|
|
name string, path string, params []string,
|
||
|
|
handler Router,
|
||
|
|
middlewares []Middleware,
|
||
|
|
methods ...string,
|
||
|
|
)
|
||
|
|
// RegisterNested is a registration method for http request with a router generator.
|
||
|
|
RegisterNested(
|
||
|
|
name string, path string, params []string,
|
||
|
|
routerGenerator NestedRouter,
|
||
|
|
middlewares []Middleware,
|
||
|
|
methods ...string,
|
||
|
|
)
|
||
|
|
// Wrap returns a child RouterRegistry with specified name and path.
|
||
|
|
Wrap(name string, path string, middlewares ...Middleware) RouterRegistry
|
||
|
|
// Handler is a handler method that process incoming requests.
|
||
|
|
// It implements the Router interface.
|
||
|
|
Handler(ctx *fasthttp.RequestCtx)
|
||
|
|
}
|
||
|
|
routerRegistryImpl struct {
|
||
|
|
*routerContextImpl
|
||
|
|
name string
|
||
|
|
r *router.Router
|
||
|
|
middlewares []Middleware
|
||
|
|
parentNames []string
|
||
|
|
parentPaths []string
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
func (r *routerRegistryImpl) Handler(ctx *fasthttp.RequestCtx) {
|
||
|
|
r.r.Handler(ctx)
|
||
|
|
}
|
||
|
|
func (r *routerRegistryImpl) Name() string { return r.name }
|
||
|
|
func (r *routerRegistryImpl) ToContext() RouterContext { return r.routerContextImpl }
|
||
|
|
|
||
|
|
func (r *routerRegistryImpl) Register(
|
||
|
|
name string, path string, params []string,
|
||
|
|
handler Router,
|
||
|
|
middlewares []Middleware,
|
||
|
|
methods ...string,
|
||
|
|
) {
|
||
|
|
|
||
|
|
fullPath := r.reverseRouter.MustAddGr(name, path, r.parentNames, r.parentPaths, params...)
|
||
|
|
mixedRouter := ApplyMiddlware(handler, append(r.middlewares, middlewares...)...)
|
||
|
|
for _, method := range methods {
|
||
|
|
r.r.Handle(method, fullPath, mixedRouter)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *routerRegistryImpl) RegisterNested(
|
||
|
|
name string, path string, params []string,
|
||
|
|
routerGenerator NestedRouter,
|
||
|
|
middlewares []Middleware,
|
||
|
|
methods ...string,
|
||
|
|
) {
|
||
|
|
r.Register(name, path, params, routerGenerator(), middlewares, methods...)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *routerRegistryImpl) Wrap(name string, path string, middlewares ...Middleware) RouterRegistry {
|
||
|
|
var (
|
||
|
|
newName, newPath []string
|
||
|
|
)
|
||
|
|
if len(name) > 0 {
|
||
|
|
newName = append(r.parentNames, name)
|
||
|
|
} else {
|
||
|
|
newName = r.parentNames
|
||
|
|
}
|
||
|
|
if len(path) > 0 {
|
||
|
|
newPath = append(r.parentPaths, path)
|
||
|
|
} else {
|
||
|
|
newPath = r.parentPaths
|
||
|
|
}
|
||
|
|
return &routerRegistryImpl{
|
||
|
|
routerContextImpl: r.routerContextImpl,
|
||
|
|
r: r.r,
|
||
|
|
middlewares: append(r.middlewares, middlewares...),
|
||
|
|
name: strings.Join(newName, "."),
|
||
|
|
parentNames: newName,
|
||
|
|
parentPaths: newPath,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
func (r *routerRegistryImpl) String() string { return r.urlFor().String() }
|