mirror of
https://github.com/gofiber/fiber.git
synced 2025-05-29 19:03:34 +00:00
23 lines
453 B
Go
23 lines
453 B
Go
package middleware
|
|
|
|
import (
|
|
"github.com/gofiber/fiber"
|
|
"github.com/gofiber/utils"
|
|
)
|
|
|
|
// RequestID adds an UUID indentifier to the request
|
|
func RequestID() fiber.Handler {
|
|
return func(ctx *fiber.Ctx) {
|
|
// Get id from request
|
|
rid := ctx.Get(fiber.HeaderXRequestID)
|
|
// Create new UUID if empty
|
|
if len(rid) <= 0 {
|
|
rid = utils.UUID()
|
|
}
|
|
// Set new id to response
|
|
ctx.Set(fiber.HeaderXRequestID, rid)
|
|
// Continue stack
|
|
ctx.Next()
|
|
}
|
|
}
|