七叶笔记 » golang编程 » golang不想http自动处理重定向的解决方案

golang不想http自动处理重定向的解决方案

前言

    有时候发送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 := "#34;
{ // 常规方法
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, "#34;, http.StatusFound)
w.Write([]byte(time.Now().String()))
}  

结论

    如下的第一个请求是直接返回百度的网页,及重定向以后的内容。第二个请求直接返回第一个302重定向的内容。

相关文章