fix: nil ref error

dgate-io · May 12, 2024 · 38f3df2 · 38f3df2
1 parent 2fbe229
commit 38f3df2
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/pkg/dgclient/dgclient.go b/pkg/dgclient/dgclient.go
@@ -117,11 +117,12 @@ func WithBasicAuth(username, password string) Options {
 func WithFollowRedirect(follow bool) Options {
 	return func(dc DGateClient) {
 		if d, ok := dc.(*dgateClient); ok {
-			d.client.CheckRedirect = func(req *http.Request, via []*http.Request) error {
-				if follow {
-					return nil
+			if follow {
+				d.client.CheckRedirect = nil
+			} else {
+				d.client.CheckRedirect = func(req *http.Request, via []*http.Request) error {
+					return http.ErrUseLastResponse
 				}
-				return http.ErrUseLastResponse
 			}
 		}
 	}