golang不想http自動處理重定向的解決方案

目錄

• 前言
• 解決方案
• 結論

前言

    有時候發送http請求不想讓庫自動幫忙處理重定向，庫里面默認的是會把所有重定向都完成一遍，結果就是最后一個沒有重定向的請求的結果。因此需要一種方案直接獲取首次訪問的結果，不走重定向。go的http庫里面是使用如下代碼檢查重定向的，以前我傻傻的修改源碼讓下面這段代碼直接返回，這樣需要重新編譯go自帶的庫，后來發現更簡單的方案。

if err == ErrUseLastResponse {
    return resp, nil // 這里是攔截重定向，如果不攔截則走下面的重定向判斷
}

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]))
    }

    {   // 另一種不要重定向的方法
        req, err := http.NewRequest(http.MethodPost, mUrl, nil)
        if err != nil {
            log.Fatal(err)
        }
        client := &http.Client{
            CheckRedirect: func(req *http.Request, via []*http.Request) error {
                return http.ErrUseLastResponse /* 不進入重定向 */
            },
        }
        resp, err := client.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]))
    }
}

// 下面開啟一個服務，重定向到百度
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 | 百度一下，你就知道
2020/10/14 13:11:56 302 | 2020-10-14 13:11:56.6559382 +0800 CST m=+1.429170501
2020/10/14 13:11:56 302 | 2020-10-14 13:11:56.6559382 +0800 CST m=+1.429170501