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]>
43 lines
840 B
Go
43 lines
840 B
Go
package responsetime
|
|
|
|
import (
|
|
"github.com/zap-proto/fiber/v3"
|
|
)
|
|
|
|
// 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
|
|
|
|
// Header is the header key used to set the response time.
|
|
//
|
|
// Optional. Default: "X-Response-Time"
|
|
Header string
|
|
}
|
|
|
|
// ConfigDefault is the default config.
|
|
var ConfigDefault = Config{
|
|
Next: nil,
|
|
Header: fiber.HeaderXResponseTime,
|
|
}
|
|
|
|
// 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
|
|
}
|
|
|
|
return cfg
|
|
}
|