测量 Go 中的反向代理响应时间

Question

I have a web reverse proxy based on http/net. I'm using NewSingleHostReverseProxy function. How can I measure response times of target web server?

Answer 1

You could do something like this. Wrap the proxy.Transport with another http.Roundtripper implementation that times how long it takes RoundTrip to respond.

I think this will generally show you "response time". Not "request time", since the body can still be read from the response, so the request could still be ongoing, but this should show you roughly how long it takes the server to respond to the request.

package main

import (
    "fmt"
    "net/http"
    "net/http/httputil"
    "net/url"
    "time"
)

func main() {
    url, _ := url.Parse("https://stackoverflow.com")
    proxy := httputil.NewSingleHostReverseProxy(url)
    proxy.Transport = NewTimingRoundtripper(http.DefaultTransport)
}

type TimingRoundtripper struct {
    transport http.RoundTripper
}

func NewTimingRoundtripper(transport http.RoundTripper) http.RoundTripper {
    return TimingRoundtripper{transport: transport}
}

func (rt TimingRoundtripper) RoundTrip(r *http.Request) (resp *http.Response, err error) {
    start := time.Now()
    resp, err = rt.transport.RoundTrip(r)
    fmt.Println("request", r.URL, time.Since(start))
    return resp, err
}