[fix] BasicAuth header and return

buaazp · Dec 15, 2016 · 05b0baa · 05b0baa
1 parent 7778060
commit 05b0baa
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/examples/auth/README.md b/examples/auth/README.md
@@ -58,10 +58,11 @@ func BasicAuth(h fasthttp.RequestHandler, requiredUser, requiredPassword string)
 		if hasAuth && user == requiredUser && password == requiredPassword {
 			// Delegate request to the given handle
 			h(ctx)
+			return
 		}
 		// Request Basic Authentication otherwise
-		ctx.Response.Header.Set("WWW-Authenticate", "Basic realm=Restricted")
 		ctx.Error(fasthttp.StatusMessage(fasthttp.StatusUnauthorized), fasthttp.StatusUnauthorized)
+		ctx.Response.Header.Set("WWW-Authenticate", "Basic realm=Restricted")
 	})
 }
 

diff --git a/examples/auth/auth.go b/examples/auth/auth.go
@@ -49,10 +49,11 @@ func BasicAuth(h fasthttp.RequestHandler, requiredUser, requiredPassword string)
 		if hasAuth && user == requiredUser && password == requiredPassword {
 			// Delegate request to the given handle
 			h(ctx)
+			return
 		}
 		// Request Basic Authentication otherwise
-		ctx.Response.Header.Set("WWW-Authenticate", "Basic realm=Restricted")
 		ctx.Error(fasthttp.StatusMessage(fasthttp.StatusUnauthorized), fasthttp.StatusUnauthorized)
+		ctx.Response.Header.Set("WWW-Authenticate", "Basic realm=Restricted")
 	})
 }