diff options
Diffstat (limited to 'internal/router')
-rw-r--r-- | internal/router/attach.go | 7 | ||||
-rw-r--r-- | internal/router/router.go | 11 |
2 files changed, 11 insertions, 7 deletions
diff --git a/internal/router/attach.go b/internal/router/attach.go index 88364831a..7c20b33d8 100644 --- a/internal/router/attach.go +++ b/internal/router/attach.go @@ -39,3 +39,10 @@ func (r *router) AttachMiddleware(middleware gin.HandlerFunc) { func (r *router) AttachNoRouteHandler(handler gin.HandlerFunc) { r.engine.NoRoute(handler) } + +// AttachGroup attaches the given handlers into a group with the given relativePath as +// base path for that group. It then returns the *gin.RouterGroup so that the caller +// can add any extra middlewares etc specific to that group, as desired. +func (r *router) AttachGroup(relativePath string, handlers ...gin.HandlerFunc) *gin.RouterGroup { + return r.engine.Group(relativePath, handlers...) +} diff --git a/internal/router/router.go b/internal/router/router.go index 3917de314..5eb4cb222 100644 --- a/internal/router/router.go +++ b/internal/router/router.go @@ -47,8 +47,10 @@ type Router interface { AttachMiddleware(handler gin.HandlerFunc) // Attach 404 NoRoute handler AttachNoRouteHandler(handler gin.HandlerFunc) - // Add Gin StaticFS handler - AttachStaticFS(relativePath string, fs http.FileSystem) + // Attach a router group, and receive that group back. + // More middlewares and handlers can then be attached on + // the group by the caller. + AttachGroup(path string, handlers ...gin.HandlerFunc) *gin.RouterGroup // Start the router Start() // Stop the router @@ -62,11 +64,6 @@ type router struct { certManager *autocert.Manager } -// Add Gin StaticFS handler -func (r *router) AttachStaticFS(relativePath string, fs http.FileSystem) { - r.engine.StaticFS(relativePath, fs) -} - // Start starts the router nicely. It will serve two handlers if letsencrypt is enabled, and only the web/API handler if letsencrypt is not enabled. func (r *router) Start() { // listen is the server start function, by |