mirror of
https://github.com/therootcompany/golib.git
synced 2025-10-07 01:28:19 +00:00
feat(http/middleware): add idiomatic http-compatible middleware
This commit is contained in:
parent
193c8f65cf
commit
4c26dd6e2a
55
http/middleware/middleware.go
Normal file
55
http/middleware/middleware.go
Normal file
@ -0,0 +1,55 @@
|
|||||||
|
// Copyright 2025 AJ ONeal <aj@therootcompany.com> (https://therootcompany.com)
|
||||||
|
//
|
||||||
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
||||||
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||||
|
// file, You can obtain one at https://mozilla.org/MPL/2.0/.
|
||||||
|
//
|
||||||
|
// SPDX-License-Identifier: MPL-2.0
|
||||||
|
|
||||||
|
package middleware
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
"slices"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Middleware receives and returns and http.HandlerFunc
|
||||||
|
type Middleware func(http.HandlerFunc) http.HandlerFunc
|
||||||
|
|
||||||
|
// MiddlewareChain enables inline chaining
|
||||||
|
type MiddlewareChain struct {
|
||||||
|
middlewares []Middleware
|
||||||
|
}
|
||||||
|
|
||||||
|
// New creates a reusable MiddlewareChain with 0 or more middleware
|
||||||
|
func New(middlewares ...Middleware) MiddlewareChain {
|
||||||
|
return MiddlewareChain{middlewares: middlewares}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Use appends additional middleware to the chain
|
||||||
|
func (c MiddlewareChain) Use(middlewares ...Middleware) MiddlewareChain {
|
||||||
|
newMiddlewares := make([]Middleware, len(c.middlewares), len(c.middlewares)+len(middlewares))
|
||||||
|
copy(newMiddlewares, c.middlewares)
|
||||||
|
newMiddlewares = append(newMiddlewares, middlewares...)
|
||||||
|
|
||||||
|
return MiddlewareChain{middlewares: newMiddlewares}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Handle composes middleware with the final handler
|
||||||
|
func (c MiddlewareChain) Handle(handler http.HandlerFunc) http.HandlerFunc {
|
||||||
|
if handler == nil {
|
||||||
|
panic("mw.New(...).Use(...).Handle(-->this<--) requires a handler")
|
||||||
|
}
|
||||||
|
|
||||||
|
middlewares := make([]Middleware, len(c.middlewares))
|
||||||
|
copy(middlewares, c.middlewares)
|
||||||
|
slices.Reverse(middlewares)
|
||||||
|
|
||||||
|
// Apply middleware in forward order
|
||||||
|
result := handler
|
||||||
|
for _, m := range middlewares {
|
||||||
|
result = m(result)
|
||||||
|
}
|
||||||
|
|
||||||
|
return result
|
||||||
|
}
|
Loading…
x
Reference in New Issue
Block a user