🚀 [Feature]: middleware/requestid: don't call "Generator" func on existing request ID header (#2371)

call uuid generator only if rid is empty
2023-03-15 16:45:42 +08:00 · 2023-03-15 16:45:42 +08:00 · 56839b433e
parent d7b36cde54
commit 56839b433e
1 changed files with 4 additions and 1 deletions
--- a/middleware/requestid/requestid.go
+++ b/middleware/requestid/requestid.go
@ -16,7 +16,10 @@ func New(config ...Config) fiber.Handler {
 			return c.Next()
 		}
 		// Get id from request, else we generate one
-		rid := c.Get(cfg.Header, cfg.Generator())
+		rid := c.Get(cfg.Header)
+		if rid == "" {
+			rid = cfg.Generator()
+		}

 		// Set new id to response header
 		c.Set(cfg.Header, rid)