golang不想http自動處理重定向的解決方案
阿新 • • 發佈:2020-10-14
目錄
前言
有時候傳送http請求不想讓庫自動幫忙處理重定向,庫裡面預設的是會把所有重定向都完成一遍,結果就是最後一個沒有重定向的請求的結果。因此需要一種方案直接獲取首次訪問的結果,不走重定向。go的http庫裡面是使用如下程式碼檢查重定向的,以前我傻傻的修改原始碼讓下面這段程式碼直接返回,這樣需要重新編譯go自帶的庫,後來發現更簡單的方案。
var shouldRedirect bool redirectMethod, shouldRedirect, includeBody = redirectBehavior(req.Method, resp, reqs[0]) if !shouldRedirect { return resp, nil }
解決方案
下面程式碼可以驗證自動處理重定向,以及不走重定向的方案。
package main import ( "io/ioutil" "log" "net/http" "time" ) func main() { go server() time.Sleep(time.Second) mUrl := "http://127.0.0.1:12345/post" { // 常規方法 req, err := http.NewRequest(http.MethodPost, mUrl, nil) if err != nil { log.Fatal(err) } resp, err := http.DefaultClient.Do(req) if resp != nil { defer resp.Body.Close() } if err != nil { log.Fatal(err) } byt, err := ioutil.ReadAll(resp.Body) if err != nil { log.Fatal(err) } log.Println(resp.StatusCode, "|", string(byt[:128])) } { // 去掉自動處理重定向 req, err := http.NewRequest(http.MethodPost, mUrl, nil) if err != nil { log.Fatal(err) } resp, err := http.DefaultTransport.RoundTrip(req) if resp != nil { defer resp.Body.Close() } if err != nil { log.Fatal(err) } byt, err := ioutil.ReadAll(resp.Body) if err != nil { log.Fatal(err) } log.Println(resp.StatusCode, "|", string(byt[:128])) } } // 下面開啟一個服務,重定向到百度 func server() { http.HandleFunc("/post", mPost) http.ListenAndServe(":12345", nil) } func mPost(w http.ResponseWriter, r *http.Request) { http.Redirect(w, r, "http://www.baidu.com", http.StatusFound) w.Write([]byte(time.Now().String())) }
結論
如下的第一個請求是直接返回百度的網頁,及重定向以後的內容。第二個請求直接返回第一個302重定向的內容。
2020/10/14 13:11:56
200 | <!DOCTYPE html><!--STATUS OK-->
2020/10/14 13:11:56302 | 2020-10-14 13:11:56.6559382 +0800 CST m=+1.429170501