Add RegisterWithOpts

LKaemmerling · Nov 4, 2022 · aff0314 · aff0314
1 parent 384df28
commit aff0314
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/middleware.go b/middleware.go
@@ -178,12 +178,12 @@ func (ps *FiberPrometheus) RegisterAtWithOpts(app *fiber.App, opts RegisterOpts,
 	}
 	ps.defaultURL = opts.URL
 	app.Get(ps.defaultURL, func(ctx *fiber.Ctx) error {
-		reqIP := ctx.IP()
+		reqIP := strings.ToLower(ctx.IP())
 		if len(reqIP) == 0 {
-			reqIP = ctx.Context().RemoteIP().String()
+			reqIP = strings.ToLower(ctx.Context().RemoteIP().String())
 		}
 		for _, ip := range opts.WhitelistedIPs {
-			if strings.ToLower(ip) == strings.ToLower(ctx.IP()) {
+			if strings.ToLower(ip) == reqIP {
 				err := adaptor.HTTPHandler(promhttp.Handler())(ctx)
 				if err != nil {
 					return err