mirror of
https://github.com/gofiber/fiber.git
synced 2025-04-27 13:14:31 +00:00
8c3f81e2b7
* Use composites for internal structures. Fix alignment of structures across Fiber * Update struct alignment in test files * Enable alignment check with govet * Fix ctx autoformat unit-test * Revert app Config struct. Add betteralign to Makefile * Disable comment on alert since it wont work for forks * Update benchmark.yml * Update benchmark.yml * Remove warning from using positional fields * Update router.go
55 lines
1.2 KiB
Go
55 lines
1.2 KiB
Go
package requestid
|
|
|
|
import (
|
|
"github.com/gofiber/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.UUID
|
|
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 fast UUID generator which will expose the number of
|
|
// requests made to the server. To conceal this value for better
|
|
// privacy, use the "utils.UUIDv4" generator.
|
|
var ConfigDefault = Config{
|
|
Next: nil,
|
|
Header: fiber.HeaderXRequestID,
|
|
Generator: utils.UUID,
|
|
}
|
|
|
|
// 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
|
|
}
|