🚀 [Feature]: middleware/requestid: don't call "Generator" func on exi…

…sting request ID header (#2371) call uuid generator only if rid is empty
gofiber · Mar 15, 2023 · 56839b4 · 56839b4
1 parent d7b36cd
commit 56839b4
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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)