mirror of
https://github.com/zap-proto/fiber.git
synced 2026-10-30 15:06:23 +00:00
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]>
30 lines
496 B
Go
30 lines
496 B
Go
package responsetime
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/zap-proto/fiber/v3"
|
|
)
|
|
|
|
// 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()
|
|
}
|
|
|
|
start := time.Now()
|
|
|
|
err := c.Next()
|
|
|
|
c.Set(cfg.Header, time.Since(start).String())
|
|
|
|
return err
|
|
}
|
|
}
|