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