mirror of
https://github.com/zap-proto/fiber.git
synced 2026-10-30 15:06:23 +00:00
zeekayandHanzo Dev
f60e254acb
Fork of gofiber/fiber v3 based on the v3.2.0 tag. Rewrites all self-imports github.com/gofiber/fiber/v3 -> github.com/zap-proto/fiber/v3. Upstream gofiber/schema and gofiber/utils/v2 dependencies are left untouched. Co-authored-by: Hanzo Dev <[email protected]>
53 lines
1.2 KiB
Go
53 lines
1.2 KiB
Go
package requestid
|
|
|
|
import (
|
|
"github.com/zap-proto/fiber/v3"
|
|
"github.com/gofiber/utils/v2"
|
|
)
|
|
|
|
// Config defines the config for middleware.
|
|
type Config struct {
|
|
// Next defines a function to skip this middleware when returned true.
|
|
//
|
|
// Optional. Default: nil
|
|
Next func(c fiber.Ctx) bool
|
|
|
|
// Generator defines a function to generate the unique identifier.
|
|
//
|
|
// Optional. Default: utils.SecureToken
|
|
Generator func() string
|
|
|
|
// Header is the header key where to get/set the unique request ID
|
|
//
|
|
// Optional. Default: "X-Request-ID"
|
|
Header string
|
|
}
|
|
|
|
// ConfigDefault is the default config
|
|
// It uses a secure token generator for better privacy and security.
|
|
var ConfigDefault = Config{
|
|
Next: nil,
|
|
Header: fiber.HeaderXRequestID,
|
|
Generator: utils.SecureToken,
|
|
}
|
|
|
|
// Helper function to set default values
|
|
func configDefault(config ...Config) Config {
|
|
// Return default config if nothing provided
|
|
if len(config) < 1 {
|
|
return ConfigDefault
|
|
}
|
|
|
|
// Override default config
|
|
cfg := config[0]
|
|
|
|
// Set default values
|
|
if cfg.Header == "" {
|
|
cfg.Header = ConfigDefault.Header
|
|
}
|
|
if cfg.Generator == nil {
|
|
cfg.Generator = ConfigDefault.Generator
|
|
}
|
|
return cfg
|
|
}
|