Adding upstream version 2.52.6.
Signed-off-by: Daniel Baumann <daniel@debian.org>
This commit is contained in:
parent
a960158181
commit
6d002e9543
441 changed files with 95392 additions and 0 deletions
33
middleware/requestid/requestid.go
Normal file
33
middleware/requestid/requestid.go
Normal file
|
@ -0,0 +1,33 @@
|
|||
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)
|
||||
if rid == "" {
|
||||
rid = 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()
|
||||
}
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue