mirror of https://github.com/gofiber/fiber.git
31 lines
635 B
Go
31 lines
635 B
Go
package requestid
|
|
|
|
import (
|
|
"github.com/gofiber/fiber/v2"
|
|
)
|
|
|
|
// New creates a new middleware handler
|
|
func New(config ...Config) fiber.Handler {
|
|
// Set default config
|
|
cfg := configDefault(config...)
|
|
|
|
// Return new handler
|
|
return func(c *fiber.Ctx) error {
|
|
// Don't execute middleware if Next returns true
|
|
if cfg.Next != nil && cfg.Next(c) {
|
|
return c.Next()
|
|
}
|
|
// Get id from request, else we generate one
|
|
rid := c.Get(cfg.Header, cfg.Generator())
|
|
|
|
// Set new id to response header
|
|
c.Set(cfg.Header, rid)
|
|
|
|
// Add the request ID to locals
|
|
c.Locals(cfg.ContextKey, rid)
|
|
|
|
// Continue stack
|
|
return c.Next()
|
|
}
|
|
}
|