golanghttpclient.go

博百科 • 2023年11月20日 pm7:58 • 百科知识 • 阅读 8

Posted 2021-05-24

篇首语：本文由小常识网小编为大家整理，主要介绍了golang httpclient.go相关的知识，希望对你有一定的参考价值。

package httptimeout import ( "io" "io/ioutil" "net" "net/http" "net/http/httptest" "testing" "time" ) func TestHttpTimeout(t *testing.T) { http.HandleFunc("/normal", func(w http.ResponseWriter, req *http.Request) { // Empirically, timeouts less than these seem to be flaky time.Sleep(100 * time.Millisecond) io.WriteString(w, "ok") }) http.HandleFunc("/timeout", func(w http.ResponseWriter, req *http.Request) { time.Sleep(250 * time.Millisecond) io.WriteString(w, "ok") }) ts := httptest.NewServer(http.DefaultServeMux) defer ts.Close() numDials := 0 client := &http.Client{ Transport: &TimeoutTransport{ Transport: http.Transport{ Dial: func(netw, addr string) (net.Conn, error) { t.Logf("dial to %s://%s", netw, addr) numDials++ // For testing only. return net.Dial(netw, addr) // Regular ass dial. }, }, RoundTripTimeout: time.Millisecond * 200, }, } addr := ts.URL SendTestRequest(t, client, "1st", addr, "normal") if numDials != 1 { t.Fatalf("Should only have 1 dial at this point.") } SendTestRequest(t, client, "2st", addr, "normal") if numDials != 1 { t.Fatalf("Should only have 1 dial at this point.") } SendTestRequest(t, client, "3st", addr, "timeout") if numDials != 1 { t.Fatalf("Should only have 1 dial at this point.") } SendTestRequest(t, client, "4st", addr, "normal") if numDials != 2 { t.Fatalf("Should have our 2nd dial.") } time.Sleep(time.Millisecond * 700) SendTestRequest(t, client, "5st", addr, "normal") if numDials != 2 { t.Fatalf("Should still only have 2 dials.") } } func SendTestRequest(t *testing.T, client *http.Client, id, addr, path string) { req, err := http.NewRequest("GET", addr+"/"+path, nil) if err != nil { t.Fatalf("new request failed - %s", err) } req.Header.Add("Connection", "keep-alive") switch path { case "normal": if resp, err := client.Do(req); err != nil { t.Fatalf("%s request failed - %s", id, err) } else { result, err2 := ioutil.ReadAll(resp.Body) if err2 != nil { t.Fatalf("%s response read failed - %s", id, err2) } resp.Body.Close() t.Logf("%s request - %s", id, result) } case "timeout": if _, err := client.Do(req); err == nil { t.Fatalf("%s request not timeout", id) } else { t.Logf("%s request - %s", id, err) } } }

package httptimeout import ( "net/http" "time" "fmt" ) type TimeoutTransport struct { http.Transport RoundTripTimeout time.Duration } type respAndErr struct { resp *http.Response err error } type netTimeoutError struct { error } func (ne netTimeoutError) Timeout() bool { return true } // If you don't set RoundTrip on TimeoutTransport, this will always timeout at 0 func (t *TimeoutTransport) RoundTrip(req *http.Request) (*http.Response, error) { timeout := time.After(t.RoundTripTimeout) resp := make(chan respAndErr, 1) go func() { r, e := t.Transport.RoundTrip(req) resp <- respAndErr{ resp: r, err: e, } }() select { case <-timeout:// A round trip timeout has occurred. t.Transport.CancelRequest(req) return nil, netTimeoutError{ error: fmt.Errorf("timed out after %s", t.RoundTripTimeout), } case r := <-resp: // Success! return r.resp, r.err } }

以上是关于golang httpclient.go的主要内容，如果未能解决你的问题，请参考以下文章

版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人，因此内容不代表本站观点、本站不对文章中的任何观点负责，内容版权归原作者所有、内容只用于提供信息阅读，无任何商业用途。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站（文章、内容、图片、音频、视频）有涉嫌抄袭侵权/违法违规的内容，请发送邮件至2811500808@qq.com举报，一经查实，本站将立刻删除、维护您的正当权益!

golanghttpclient.go

相关推荐