forked from ebhomengo/niki
130 lines
4.9 KiB
Go
130 lines
4.9 KiB
Go
|
// SPDX-License-Identifier: MIT
|
||
|
// SPDX-FileCopyrightText: © 2015 LabStack LLC and Echo contributors
|
||
|
|
||
|
package echo
|
||
|
|
||
|
import (
|
||
|
"net/http"
|
||
|
)
|
||
|
|
||
|
// Group is a set of sub-routes for a specified route. It can be used for inner
|
||
|
// routes that share a common middleware or functionality that should be separate
|
||
|
// from the parent echo instance while still inheriting from it.
|
||
|
type Group struct {
|
||
|
common
|
||
|
host string
|
||
|
prefix string
|
||
|
middleware []MiddlewareFunc
|
||
|
echo *Echo
|
||
|
}
|
||
|
|
||
|
// Use implements `Echo#Use()` for sub-routes within the Group.
|
||
|
func (g *Group) Use(middleware ...MiddlewareFunc) {
|
||
|
g.middleware = append(g.middleware, middleware...)
|
||
|
if len(g.middleware) == 0 {
|
||
|
return
|
||
|
}
|
||
|
// group level middlewares are different from Echo `Pre` and `Use` middlewares (those are global). Group level middlewares
|
||
|
// are only executed if they are added to the Router with route.
|
||
|
// So we register catch all route (404 is a safe way to emulate route match) for this group and now during routing the
|
||
|
// Router would find route to match our request path and therefore guarantee the middleware(s) will get executed.
|
||
|
g.RouteNotFound("", NotFoundHandler)
|
||
|
g.RouteNotFound("/*", NotFoundHandler)
|
||
|
}
|
||
|
|
||
|
// CONNECT implements `Echo#CONNECT()` for sub-routes within the Group.
|
||
|
func (g *Group) CONNECT(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||
|
return g.Add(http.MethodConnect, path, h, m...)
|
||
|
}
|
||
|
|
||
|
// DELETE implements `Echo#DELETE()` for sub-routes within the Group.
|
||
|
func (g *Group) DELETE(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||
|
return g.Add(http.MethodDelete, path, h, m...)
|
||
|
}
|
||
|
|
||
|
// GET implements `Echo#GET()` for sub-routes within the Group.
|
||
|
func (g *Group) GET(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||
|
return g.Add(http.MethodGet, path, h, m...)
|
||
|
}
|
||
|
|
||
|
// HEAD implements `Echo#HEAD()` for sub-routes within the Group.
|
||
|
func (g *Group) HEAD(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||
|
return g.Add(http.MethodHead, path, h, m...)
|
||
|
}
|
||
|
|
||
|
// OPTIONS implements `Echo#OPTIONS()` for sub-routes within the Group.
|
||
|
func (g *Group) OPTIONS(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||
|
return g.Add(http.MethodOptions, path, h, m...)
|
||
|
}
|
||
|
|
||
|
// PATCH implements `Echo#PATCH()` for sub-routes within the Group.
|
||
|
func (g *Group) PATCH(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||
|
return g.Add(http.MethodPatch, path, h, m...)
|
||
|
}
|
||
|
|
||
|
// POST implements `Echo#POST()` for sub-routes within the Group.
|
||
|
func (g *Group) POST(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||
|
return g.Add(http.MethodPost, path, h, m...)
|
||
|
}
|
||
|
|
||
|
// PUT implements `Echo#PUT()` for sub-routes within the Group.
|
||
|
func (g *Group) PUT(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||
|
return g.Add(http.MethodPut, path, h, m...)
|
||
|
}
|
||
|
|
||
|
// TRACE implements `Echo#TRACE()` for sub-routes within the Group.
|
||
|
func (g *Group) TRACE(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||
|
return g.Add(http.MethodTrace, path, h, m...)
|
||
|
}
|
||
|
|
||
|
// Any implements `Echo#Any()` for sub-routes within the Group.
|
||
|
func (g *Group) Any(path string, handler HandlerFunc, middleware ...MiddlewareFunc) []*Route {
|
||
|
routes := make([]*Route, len(methods))
|
||
|
for i, m := range methods {
|
||
|
routes[i] = g.Add(m, path, handler, middleware...)
|
||
|
}
|
||
|
return routes
|
||
|
}
|
||
|
|
||
|
// Match implements `Echo#Match()` for sub-routes within the Group.
|
||
|
func (g *Group) Match(methods []string, path string, handler HandlerFunc, middleware ...MiddlewareFunc) []*Route {
|
||
|
routes := make([]*Route, len(methods))
|
||
|
for i, m := range methods {
|
||
|
routes[i] = g.Add(m, path, handler, middleware...)
|
||
|
}
|
||
|
return routes
|
||
|
}
|
||
|
|
||
|
// Group creates a new sub-group with prefix and optional sub-group-level middleware.
|
||
|
func (g *Group) Group(prefix string, middleware ...MiddlewareFunc) (sg *Group) {
|
||
|
m := make([]MiddlewareFunc, 0, len(g.middleware)+len(middleware))
|
||
|
m = append(m, g.middleware...)
|
||
|
m = append(m, middleware...)
|
||
|
sg = g.echo.Group(g.prefix+prefix, m...)
|
||
|
sg.host = g.host
|
||
|
return
|
||
|
}
|
||
|
|
||
|
// File implements `Echo#File()` for sub-routes within the Group.
|
||
|
func (g *Group) File(path, file string) {
|
||
|
g.file(path, file, g.GET)
|
||
|
}
|
||
|
|
||
|
// RouteNotFound implements `Echo#RouteNotFound()` for sub-routes within the Group.
|
||
|
//
|
||
|
// Example: `g.RouteNotFound("/*", func(c echo.Context) error { return c.NoContent(http.StatusNotFound) })`
|
||
|
func (g *Group) RouteNotFound(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||
|
return g.Add(RouteNotFound, path, h, m...)
|
||
|
}
|
||
|
|
||
|
// Add implements `Echo#Add()` for sub-routes within the Group.
|
||
|
func (g *Group) Add(method, path string, handler HandlerFunc, middleware ...MiddlewareFunc) *Route {
|
||
|
// Combine into a new slice to avoid accidentally passing the same slice for
|
||
|
// multiple routes, which would lead to later add() calls overwriting the
|
||
|
// middleware from earlier calls.
|
||
|
m := make([]MiddlewareFunc, 0, len(g.middleware)+len(middleware))
|
||
|
m = append(m, g.middleware...)
|
||
|
m = append(m, middleware...)
|
||
|
return g.echo.add(g.host, method, g.prefix+path, handler, m...)
|
||
|
}
|