Move header user copy & remove only when ip filter not match

alexandreh2ag · Oct 26, 2022 · 1b72403 · 1b72403
1 parent ca26f8e
commit 1b72403
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/middleware.go b/middleware.go
@@ -99,14 +99,15 @@ func (m *Middleware) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
 			fmt.Println("IP is not authorize and basic auth is not valid")
 			return
 		}
-	}
-	req.URL.User = url.User(user)
-	if m.headerField != "" {
-		req.Header[m.headerField] = []string{user}
-	}
 
-	if m.removeHeader {
-		req.Header.Del(authorizationHeader)
+		req.URL.User = url.User(user)
+		if m.headerField != "" {
+			req.Header[m.headerField] = []string{user}
+		}
+
+		if m.removeHeader {
+			req.Header.Del(authorizationHeader)
+		}
 	}
 
 	fmt.Println("Request authorized")